Update agent.py
Browse files
agent.py
CHANGED
|
@@ -8,7 +8,7 @@ from langgraph.prebuilt import ToolNode
|
|
| 8 |
|
| 9 |
from prompts import AGENT_SYSTEM_PROMPT, build_question_prompt
|
| 10 |
from state import AgentState
|
| 11 |
-
from tools import wikipedia_tool, arxiv_tool, get_current_year, ddg_search_tool
|
| 12 |
from langchain_groq import ChatGroq
|
| 13 |
|
| 14 |
from langgraph.graph import StateGraph
|
|
@@ -20,7 +20,7 @@ GROQ_API_KEY= os.getenv("GROQ_API_KEY")
|
|
| 20 |
RECURSION_LIMIT = 50
|
| 21 |
AGENT_CODE_URL = os.getenv("AGENT_CODE_URL", "https://huggingface.co/spaces/manasajanj/Final_Assignment_Template/tree/main")
|
| 22 |
llm = ChatGroq(model="openai/gpt-oss-120b")
|
| 23 |
-
tools_list = [wikipedia_tool, ddg_search_tool, arxiv_tool, get_current_year]
|
| 24 |
llm_with_tools = llm.bind_tools(tools_list)
|
| 25 |
|
| 26 |
|
|
|
|
| 8 |
|
| 9 |
from prompts import AGENT_SYSTEM_PROMPT, build_question_prompt
|
| 10 |
from state import AgentState
|
| 11 |
+
from tools import wikipedia_tool, arxiv_tool, get_current_year, ddg_search_tool,get_youtube_transcript, fetch_url_content,get_gaia_file
|
| 12 |
from langchain_groq import ChatGroq
|
| 13 |
|
| 14 |
from langgraph.graph import StateGraph
|
|
|
|
| 20 |
RECURSION_LIMIT = 50
|
| 21 |
AGENT_CODE_URL = os.getenv("AGENT_CODE_URL", "https://huggingface.co/spaces/manasajanj/Final_Assignment_Template/tree/main")
|
| 22 |
llm = ChatGroq(model="openai/gpt-oss-120b")
|
| 23 |
+
tools_list = [wikipedia_tool, ddg_search_tool, arxiv_tool, get_current_year,get_youtube_transcript,fetch_url_content,get_gaia_file]
|
| 24 |
llm_with_tools = llm.bind_tools(tools_list)
|
| 25 |
|
| 26 |
|