CTPC commited on
Commit
594cb55
·
verified ·
1 Parent(s): 94a1031

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -44,9 +44,9 @@ class BasicAgent:
44
  read_file,
45
  transcribe_audio
46
  ]
47
- self.google_llm = ChatGoogleGenerativeAI(model='gemini-2.5-flash')
48
- self.llm_with_tools = self.google_llm.bind_tools(self.tools)
49
- self.graph = self.create_graph()
50
 
51
  def __call__(self, task_id: str, question: str, file_name: str = None, local_file_name: str = None) -> str:
52
  print(f"Agent received question (first 50 chars): {question[:50]}...")
 
44
  read_file,
45
  transcribe_audio
46
  ]
47
+ self.llm = ChatOpenAI(model="gpt-4o-mini")
48
+ self.llm_with_tools = self.llm.bind_tools(self.tools)
49
+ self.google_llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-lite')
50
 
51
  def __call__(self, task_id: str, question: str, file_name: str = None, local_file_name: str = None) -> str:
52
  print(f"Agent received question (first 50 chars): {question[:50]}...")