Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from typing import TypedDict, Annotated
|
| 2 |
from tool import add, substract, multiply, divide, DuckDuckGoSearchTool, WikipediaSearchTool, ArxivSearchTool, PubmedSearchTool
|
| 3 |
-
|
| 4 |
from langgraph.graph.message import add_messages
|
| 5 |
from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
|
| 6 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
|
@@ -15,6 +15,8 @@ with open("prompt.txt", "r", encoding="utf-8") as f:
|
|
| 15 |
# System message
|
| 16 |
sys_msg = SystemMessage(content=system_prompt)
|
| 17 |
|
|
|
|
|
|
|
| 18 |
# Making the agent
|
| 19 |
llm = HuggingFaceEndpoint(
|
| 20 |
repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|
|
|
|
| 1 |
from typing import TypedDict, Annotated
|
| 2 |
from tool import add, substract, multiply, divide, DuckDuckGoSearchTool, WikipediaSearchTool, ArxivSearchTool, PubmedSearchTool
|
| 3 |
+
from os import getenv
|
| 4 |
from langgraph.graph.message import add_messages
|
| 5 |
from langchain_core.messages import AnyMessage, SystemMessage, HumanMessage, AIMessage
|
| 6 |
from langgraph.graph import StateGraph, START, END, MessagesState
|
|
|
|
| 15 |
# System message
|
| 16 |
sys_msg = SystemMessage(content=system_prompt)
|
| 17 |
|
| 18 |
+
HUGGINGFACEHUB_API_TOKEN = getenv("HUGGINGFACEHUB_API_TOKEN")
|
| 19 |
+
|
| 20 |
# Making the agent
|
| 21 |
llm = HuggingFaceEndpoint(
|
| 22 |
repo_id="Qwen/Qwen2.5-Coder-32B-Instruct",
|