CTPC commited on
Commit
fde125c
·
verified ·
1 Parent(s): 98fa13f

Update app.py

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