drAbreu commited on
Commit
e3970e6
·
1 Parent(s): 4848b7e

Added thinking capabilities to anthropic models

Browse files
Files changed (1) hide show
  1. agents/llama_index_agent.py +16 -3
agents/llama_index_agent.py CHANGED
@@ -88,7 +88,12 @@ class GaiaAgent(ReActAgent):
88
  return OpenAI(model=model_name, api_key=api_key or os.getenv("OPENAI_API_KEY"))
89
 
90
  elif model_provider == "anthropic":
91
- return Anthropic(model=model_name, api_key=api_key or os.getenv("ANTHROPIC_API_KEY"))
 
 
 
 
 
92
 
93
  else:
94
  raise ValueError(f"Unsupported model provider: {model_provider}. "
@@ -199,9 +204,17 @@ def create_writer_agent(model_config: Dict[str, Any]) -> ReActAgent:
199
  api_key = model_config.get("api_key")
200
 
201
  if model_provider.lower() == "openai":
202
- llm = OpenAI(model=model_name, api_key=api_key or os.getenv("OPENAI_API_KEY"))
 
 
 
203
  elif model_provider.lower() == "anthropic":
204
- llm = Anthropic(model=model_name, api_key=api_key or os.getenv("ANTHROPIC_API_KEY"))
 
 
 
 
 
205
  else:
206
  raise ValueError(f"Unsupported model provider for writer agent: {model_provider}")
207
 
 
88
  return OpenAI(model=model_name, api_key=api_key or os.getenv("OPENAI_API_KEY"))
89
 
90
  elif model_provider == "anthropic":
91
+ return Anthropic(
92
+ model=model_name,
93
+ api_key=api_key or os.getenv("ANTHROPIC_API_KEY"),
94
+ temperature=0.5,
95
+ thinking_dict={"type": "enabled", "budget_tokens": 1600}
96
+ )
97
 
98
  else:
99
  raise ValueError(f"Unsupported model provider: {model_provider}. "
 
204
  api_key = model_config.get("api_key")
205
 
206
  if model_provider.lower() == "openai":
207
+ llm = OpenAI(
208
+ model=model_name,
209
+ api_key=api_key or os.getenv("OPENAI_API_KEY"),
210
+ )
211
  elif model_provider.lower() == "anthropic":
212
+ llm = Anthropic(
213
+ model=model_name,
214
+ api_key=api_key or os.getenv("ANTHROPIC_API_KEY"),
215
+ temperature=0.5,
216
+ thinking_dict={"type": "enabled", "budget_tokens": 1600},
217
+ )
218
  else:
219
  raise ValueError(f"Unsupported model provider for writer agent: {model_provider}")
220