Tamannathakur commited on
Commit
6d0c141
·
verified ·
1 Parent(s): 26c769a

Update supervisor_agent.py

Browse files
Files changed (1) hide show
  1. supervisor_agent.py +2 -2
supervisor_agent.py CHANGED
@@ -1,6 +1,6 @@
1
  from langchain.agents import create_agent
2
 
3
- from llm import groq_model
4
  from prompts import SUPERVISOR_AGENT_PROMPT
5
  from agents import (
6
  general_query_tool,
@@ -10,7 +10,7 @@ from agents import (
10
  )
11
 
12
  supervisor_agent = create_agent(
13
- groq_model,
14
  tools=[
15
  general_query_tool,
16
  recommendation_tool,
 
1
  from langchain.agents import create_agent
2
 
3
+ from llm import gemini_model
4
  from prompts import SUPERVISOR_AGENT_PROMPT
5
  from agents import (
6
  general_query_tool,
 
10
  )
11
 
12
  supervisor_agent = create_agent(
13
+ gemini_model,
14
  tools=[
15
  general_query_tool,
16
  recommendation_tool,