dmfelder commited on
Commit
54db84e
·
verified ·
1 Parent(s): 3cd929c

Update ShrewdAgent.py

Browse files
Files changed (1) hide show
  1. 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-4.1",
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