Added thinking capabilities to anthropic models
Browse files- 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(
|
|
|
|
|
|
|
|
|
|
|
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(
|
|
|
|
|
|
|
203 |
elif model_provider.lower() == "anthropic":
|
204 |
-
llm = Anthropic(
|
|
|
|
|
|
|
|
|
|
|
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 |
|