Yerma commited on
Commit
1745c9e
·
1 Parent(s): 5220444

replaced model to HG

Browse files
Files changed (1) hide show
  1. agent.py +9 -2
agent.py CHANGED
@@ -16,6 +16,7 @@ from langchain_openai import ChatOpenAI
16
  from langchain_core.messages import AnyMessage, HumanMessage, SystemMessage
17
  from langchain_core.tools import tool
18
  from pydantic import BaseModel, Field
 
19
 
20
  from dotenv import load_dotenv
21
  load_dotenv()
@@ -271,9 +272,15 @@ def transcribe_audio(file_path: str) -> str:
271
  return f"Transcription failed: {e}"
272
 
273
 
274
- llm_fast = ChatOpenAI(model="gpt-4o-mini", temperature=0) # основной агент
275
- llm_strong = ChatOpenAI(model="gpt-4o", temperature=0)
276
 
 
 
 
 
 
 
277
 
278
  tools = [
279
  web_search,
 
16
  from langchain_core.messages import AnyMessage, HumanMessage, SystemMessage
17
  from langchain_core.tools import tool
18
  from pydantic import BaseModel, Field
19
+ from langchain_huggingface import HuggingFaceEndpoint, ChatHuggingFace
20
 
21
  from dotenv import load_dotenv
22
  load_dotenv()
 
272
  return f"Transcription failed: {e}"
273
 
274
 
275
+ # llm_fast = #ChatOpenAI(model="gpt-4o-mini", temperature=0) # основной агент
276
+ # llm_strong = ChatOpenAI(model="gpt-4o", temperature=0)
277
 
278
+ llm = HuggingFaceEndpoint(
279
+ repo_id="Qwen/Qwen2.5-72B-Instruct",
280
+ huggingfacehub_api_token= os.environ["HF_TOKEN"]
281
+ )
282
+
283
+ llm_fast = ChatHuggingFace(llm=llm, verbose=True)
284
 
285
  tools = [
286
  web_search,