requirements update for llm compat

app.py

@@ -462,9 +462,8 @@ class BasicAgent:
             logger.info("\n=== Tool Execution ===")
             if isinstance(last_message, AIMessage):
                 # Extract tool calls from the message
-                tool_calls = last_message.tool_calls
-                if tool_calls:
-                    for tool_call in tool_calls:
+                if hasattr(last_message, 'tool_calls') and last_message.tool_calls:
+                    for tool_call in last_message.tool_calls:
                         try:
                             tool_name = tool_call.name
                             tool_args = tool_call.args
@@ -479,11 +478,25 @@ class BasicAgent:
                             logger.error(f"Error executing tool {tool_name}: {str(e)}")
                             messages.append(AIMessage(content=f"Tool error: {str(e)}"))
                 else:
-                    logger.info("No tool calls found in AI message")
+                    # If no tool calls found, check if we need to prompt for a tool call
+                    content = last_message.content.strip().lower()
+                    if any(phrase in content for phrase in ["let me", "i'll", "i will", "sure", "okay", "alright"]):
+                        logger.info("No tool calls found, prompting for search")
+                        messages.append(AIMessage(content="Please use the google_search tool to find the information."))
+                        messages.append(HumanMessage(content="Please search for the information using the google_search tool."))
+                    else:
+                        logger.info("No tool calls found in AI message")
+                        # If the message looks like a final answer, return it
+                        if content.isdigit() or (content.startswith('[') and content.endswith(']')):
+                            return {"messages": [last_message], "next": END}
+                        else:
+                            # Otherwise, continue the conversation
+                            return {"messages": messages, "next": "agent"}
             
             return {"messages": messages, "next": "agent"}
         except Exception as e:
             logger.error(f"Error in call_tools: {str(e)}")
+            # If there's an error, try to continue the conversation
             return {"messages": messages, "next": "agent"}
     
     async def __call__(self, question: str, task_id: str) -> str: