prozorov commited on
Commit
4a4ae8f
·
verified ·
1 Parent(s): d3763c5

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +13 -13
agent.py CHANGED
@@ -98,11 +98,11 @@ tools = [
98
  ]
99
 
100
  def build_graph():
101
- #llm = ChatHuggingFace(
102
- # llm=HuggingFaceEndpoint(
103
- # repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
104
- # ),
105
- #)
106
 
107
  #llm = YandexGPT(
108
  # api_key=os.environ["YANDEX_API_KEY"],
@@ -110,15 +110,15 @@ def build_graph():
110
  # model_uri=os.environ["YANDEX_MODEL_URI"],
111
  #)
112
 
113
- llm = ChatDeepSeek(
114
- model="deepseek-chat",
115
- temperature=0,
116
- max_tokens=None,
117
- timeout=None,
118
- max_retries=2,
119
- )
120
 
121
- llm_with_tools = llm.bind_tools(tools)
122
 
123
  def assistant(state: MessagesState):
124
  """Assistant node"""
 
98
  ]
99
 
100
  def build_graph():
101
+ llm = ChatHuggingFace(
102
+ llm=HuggingFaceEndpoint(
103
+ repo_id = "Qwen/Qwen2.5-Coder-32B-Instruct"
104
+ ),
105
+ )
106
 
107
  #llm = YandexGPT(
108
  # api_key=os.environ["YANDEX_API_KEY"],
 
110
  # model_uri=os.environ["YANDEX_MODEL_URI"],
111
  #)
112
 
113
+ #llm = ChatDeepSeek(
114
+ # model="deepseek-chat",
115
+ # temperature=0,
116
+ # max_tokens=None,
117
+ # timeout=None,
118
+ # max_retries=2,
119
+ #)
120
 
121
+ #llm_with_tools = llm.bind_tools(tools)
122
 
123
  def assistant(state: MessagesState):
124
  """Assistant node"""