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(
|