sarath2803 commited on
Commit
06d87ae
·
verified ·
1 Parent(s): 7fc1950

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import inspect
7
  import pandas as pd
8
  #model requirement
9
  from smolagents import DuckDuckGoSearchTool, load_tool, tool, CodeAgent,InferenceClientModel
10
- from typing import TypedDict, List, Dict, Any, Optional
11
  from langgraph.graph import StateGraph, END
12
  from langchain_openai import ChatOpenAI
13
  from langchain_core.messages import HumanMessage
@@ -101,7 +101,7 @@ def web_search(query: str) -> str:
101
  return f"Search results for '{query}' : {result}."
102
 
103
  @tool
104
- def wikipedia_tool(query: str, llm: Optional[callable[[str], str]] =openrouter_inference) -> str:
105
  """
106
  Searches Wikipedia for the given query and uses the LLM to extract a direct answer.
107
  Args:
 
7
  import pandas as pd
8
  #model requirement
9
  from smolagents import DuckDuckGoSearchTool, load_tool, tool, CodeAgent,InferenceClientModel
10
+ from typing import TypedDict, List, Dict, Any, Optional,Callable
11
  from langgraph.graph import StateGraph, END
12
  from langchain_openai import ChatOpenAI
13
  from langchain_core.messages import HumanMessage
 
101
  return f"Search results for '{query}' : {result}."
102
 
103
  @tool
104
+ def wikipedia_tool(query: str, llm: Optional[Callable[[str], str]] =openrouter_inference) -> str:
105
  """
106
  Searches Wikipedia for the given query and uses the LLM to extract a direct answer.
107
  Args: