Yongkang ZOU commited on
Commit
664b67c
·
1 Parent(s): fc07371

update agent

Browse files
Files changed (1) hide show
  1. agent.py +7 -1
agent.py CHANGED
@@ -9,9 +9,12 @@ from langchain_community.tools.tavily_search import TavilySearchResults
9
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
10
  from langchain_core.messages import SystemMessage, HumanMessage
11
  from langchain_core.tools import tool
 
12
 
13
  load_dotenv()
14
 
 
 
15
  # ------------------- TOOL DEFINITIONS -------------------
16
 
17
  @tool
@@ -82,7 +85,10 @@ def build_graph(provider: str = "groq"):
82
  if provider == "google":
83
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
84
  elif provider == "groq":
85
- llm = ChatGroq(model="qwen-qwq-32b", temperature=0)
 
 
 
86
  elif provider == "huggingface":
87
  llm = ChatHuggingFace(
88
  llm=HuggingFaceEndpoint(
 
9
  from langchain_community.document_loaders import WikipediaLoader, ArxivLoader
10
  from langchain_core.messages import SystemMessage, HumanMessage
11
  from langchain_core.tools import tool
12
+ from langchain_groq import ChatGroq
13
 
14
  load_dotenv()
15
 
16
+
17
+
18
  # ------------------- TOOL DEFINITIONS -------------------
19
 
20
  @tool
 
85
  if provider == "google":
86
  llm = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
87
  elif provider == "groq":
88
+ groq_key = os.getenv("GROQ_API_KEY")
89
+ if not groq_key:
90
+ raise ValueError("GROQ_API_KEY is not set.")
91
+ llm = ChatGroq(model="qwen-qwq-32b", temperature=0, api_key=groq_key)
92
  elif provider == "huggingface":
93
  llm = ChatHuggingFace(
94
  llm=HuggingFaceEndpoint(