manasajanj commited on
Commit
cee86e1
·
verified ·
1 Parent(s): 5c78308

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +2 -2
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