Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -98,11 +98,11 @@ tools = [
|
|
| 98 |
]
|
| 99 |
|
| 100 |
def build_graph():
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 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 |
-
|
| 115 |
-
|
| 116 |
-
|
| 117 |
-
|
| 118 |
-
|
| 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"""
|