diff --git a/galacticview_bot/agents.py b/galacticview_bot/agents.py index fa569b7..0397f3d 100644 --- a/galacticview_bot/agents.py +++ b/galacticview_bot/agents.py @@ -14,7 +14,7 @@ tools = [tavily_search_tool] -llm_with_tools = llm.bind_tools(tools) +llm_with_tools = llm.bind_tools(tools, tool_choice="auto") class AgentState(TypedDict): messages: Annotated[List[BaseMessage], add_messages] diff --git a/galacticview_bot/search.py b/galacticview_bot/search.py index 9bf44df..142ef27 100644 --- a/galacticview_bot/search.py +++ b/galacticview_bot/search.py @@ -8,17 +8,17 @@ load_dotenv() class TavilyInput(BaseModel): - query: str = Field(description="The search query") + query: str = Field(description="The search query to find information on the internet.") if os.getenv("TAVILY_API_KEY"): logger.info("TAVILY_API_KEY found. Initializing Tavily search tool.") tavily_search_tool = TavilySearch( max_results=3, - topic="general", include_answer=True, include_raw_content=False, args_schema=TavilyInput, - description="Search the internet for space-related information and return relevant results.", + name="tavily_search", + description="Search the internet for real-time information. Input should be a simple search query string." ) else: logger.warning("TAVILY_API_KEY not set. Tavily search tool will be disabled.") \ No newline at end of file