Update ShrewdAgent.py
Browse files- ShrewdAgent.py +2 -2
ShrewdAgent.py
CHANGED
|
@@ -82,7 +82,7 @@ class ShrewdAgent:
|
|
| 82 |
|
| 83 |
print("Binding tools to LLM...")
|
| 84 |
self.llm = ChatOpenAI(
|
| 85 |
-
model="gpt-
|
| 86 |
temperature=0,
|
| 87 |
api_key=SecretStr(os.environ['OPENAI_API_KEY'])
|
| 88 |
).bind_tools(self.tools)
|
|
@@ -121,7 +121,7 @@ class ShrewdAgent:
|
|
| 121 |
logger.error(f"GraphRecursionError: {e}")
|
| 122 |
|
| 123 |
final_answer = "I couldn't find the answer"
|
| 124 |
-
if accumulated_response[-1].get("assistant"):
|
| 125 |
final_answer = accumulated_response[-1]["assistant"]['messages'][-1].content
|
| 126 |
logger.info(f"Agent returning answer: {final_answer}")
|
| 127 |
return final_answer
|
|
|
|
| 82 |
|
| 83 |
print("Binding tools to LLM...")
|
| 84 |
self.llm = ChatOpenAI(
|
| 85 |
+
model="gpt-4o",
|
| 86 |
temperature=0,
|
| 87 |
api_key=SecretStr(os.environ['OPENAI_API_KEY'])
|
| 88 |
).bind_tools(self.tools)
|
|
|
|
| 121 |
logger.error(f"GraphRecursionError: {e}")
|
| 122 |
|
| 123 |
final_answer = "I couldn't find the answer"
|
| 124 |
+
if accumulated_response and accumulated_response[-1].get("assistant"):
|
| 125 |
final_answer = accumulated_response[-1]["assistant"]['messages'][-1].content
|
| 126 |
logger.info(f"Agent returning answer: {final_answer}")
|
| 127 |
return final_answer
|