Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
from langgraph.prebuilt import create_react_agent
|
| 2 |
from langchain_openai import ChatOpenAI
|
|
|
|
| 3 |
from langchain_core.messages import HumanMessage
|
| 4 |
import os
|
| 5 |
import dotenv
|
|
@@ -10,9 +11,9 @@ from tqdm import tqdm
|
|
| 10 |
|
| 11 |
dotenv.load_dotenv()
|
| 12 |
|
| 13 |
-
llm =
|
| 14 |
-
|
| 15 |
-
|
| 16 |
)
|
| 17 |
|
| 18 |
tools = get_tools()
|
|
|
|
| 1 |
from langgraph.prebuilt import create_react_agent
|
| 2 |
from langchain_openai import ChatOpenAI
|
| 3 |
+
from langchain.chat_models import ChatGroq
|
| 4 |
from langchain_core.messages import HumanMessage
|
| 5 |
import os
|
| 6 |
import dotenv
|
|
|
|
| 11 |
|
| 12 |
dotenv.load_dotenv()
|
| 13 |
|
| 14 |
+
llm = ChatGroq(
|
| 15 |
+
model="qwen/qwen3-32b",
|
| 16 |
+
api_key=os.getenv("GROQ_API_KEY")
|
| 17 |
)
|
| 18 |
|
| 19 |
tools = get_tools()
|