CTPC commited on
Commit
55b0611
·
verified ·
1 Parent(s): b651132

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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]}...")