Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -47,7 +47,8 @@ class BasicAgent:
|
|
| 47 |
]
|
| 48 |
self.llm = ChatOpenAI(model="gpt-4o-mini")
|
| 49 |
self.llm_with_tools = self.llm.bind_tools(self.tools)
|
| 50 |
-
self.google_llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-lite')
|
|
|
|
| 51 |
|
| 52 |
def __call__(self, task_id: str, question: str, file_name: str = None, local_file_name: str = None) -> str:
|
| 53 |
print(f"Agent received question (first 50 chars): {question[:50]}...")
|
|
|
|
| 47 |
]
|
| 48 |
self.llm = ChatOpenAI(model="gpt-4o-mini")
|
| 49 |
self.llm_with_tools = self.llm.bind_tools(self.tools)
|
| 50 |
+
self.google_llm = ChatGoogleGenerativeAI(model='gemini-2.0-flash-lite')
|
| 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]}...")
|