prozorov commited on
Commit
9968396
·
verified ·
1 Parent(s): 7010984

Update agent.py

Browse files
Files changed (1) hide show
  1. agent.py +8 -1
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 = ChatGoogleGenerativeAI(model="gemini-2.0-flash", temperature=0)
 
 
 
 
 
 
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