wt002 commited on
Commit
2131451
Β·
verified Β·
1 Parent(s): c648e7c

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +10 -1
agent.py CHANGED
@@ -458,16 +458,25 @@ wikiq_tool = wikidata_query
458
  # Step 7: Create the Planner-Agent Logic
459
  # -------------------------------
460
  # Define the agent tool set
 
 
 
 
461
  tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, video_tool, analyze_tool, wikiq_tool]
462
 
 
 
 
463
  # Create an agent using the planner, task classifier, and decision logic
464
  agent = initialize_agent(
465
  tools=tools,
466
- llm=llm, # βœ… Required argument
467
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
468
  verbose=True
469
  )
470
 
 
 
471
  # -------------------------------
472
  # Step 8: Use the Planner, Classifier, and Decision Logic
473
  # -------------------------------
 
458
  # Step 7: Create the Planner-Agent Logic
459
  # -------------------------------
460
  # Define the agent tool set
461
+ from langchain.chat_models import ChatOpenAI
462
+ from langchain.agents import initialize_agent, AgentType
463
+
464
+ # Define the tools (as you've already done)
465
  tools = [wiki_tool, calc_tool, file_tool, web_tool, arvix_tool, youtube_tool, video_tool, analyze_tool, wikiq_tool]
466
 
467
+ # Define the LLM before using it
468
+ llm = ChatOpenAI(temperature=0, model="gpt-4") # or "gpt-3.5-turbo"
469
+
470
  # Create an agent using the planner, task classifier, and decision logic
471
  agent = initialize_agent(
472
  tools=tools,
473
+ llm=llm,
474
  agent_type=AgentType.ZERO_SHOT_REACT_DESCRIPTION,
475
  verbose=True
476
  )
477
 
478
+
479
+
480
  # -------------------------------
481
  # Step 8: Use the Planner, Classifier, and Decision Logic
482
  # -------------------------------