Yongkang ZOU
commited on
Commit
·
664b67c
1
Parent(s):
fc07371
update agent
Browse files
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 |
-
|
|
|
|
|
|
|
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(
|