Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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[
|
| 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:
|