Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,9 +44,9 @@ class BasicAgent:
|
|
| 44 |
read_file,
|
| 45 |
transcribe_audio
|
| 46 |
]
|
| 47 |
-
self.
|
| 48 |
-
self.llm_with_tools = self.
|
| 49 |
-
self.
|
| 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]}...")
|