ORromu commited on
Commit
766670a
·
verified ·
1 Parent(s): f944c3f

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
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",