Spaces:
Sleeping
Sleeping
Update agent.py
Browse files
agent.py
CHANGED
|
@@ -16,6 +16,7 @@ from langchain_google_genai import ChatGoogleGenerativeAI
|
|
| 16 |
from langchain_community.llms import YandexGPT
|
| 17 |
from langchain_core.tools import tool
|
| 18 |
from supabase.client import Client, create_client
|
|
|
|
| 19 |
|
| 20 |
load_dotenv()
|
| 21 |
|
|
@@ -109,7 +110,13 @@ def build_graph():
|
|
| 109 |
# model_uri=os.environ["YANDEX_MODEL_URI"],
|
| 110 |
#)
|
| 111 |
|
| 112 |
-
llm =
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 113 |
|
| 114 |
llm_with_tools = llm.bind_tools(tools)
|
| 115 |
|
|
|
|
| 16 |
from langchain_community.llms import YandexGPT
|
| 17 |
from langchain_core.tools import tool
|
| 18 |
from supabase.client import Client, create_client
|
| 19 |
+
from langchain_deepseek import ChatDeepSeek
|
| 20 |
|
| 21 |
load_dotenv()
|
| 22 |
|
|
|
|
| 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 |
|