Upload 2 files

agents/agent.py

@@ -56,8 +56,8 @@ class SimpleGeminiAgent(BasicAgent):
         response = self.client.models.generate_content(model=self.model, 
                                                        contents=question,
                                                        config=GenerateContentConfig(system_instruction=SYSTEM_PROMPT_GAIA))
-        self.tokens_this_minute += response.total_token_count
-        self.token_count += response.total_token_count
+        self.tokens_this_minute += response.usage_metadata.total_token_count
+        self.token_count += response.usage_metadata.total_token_count
         logger.info(f"AdvancedAgent received question (first 50 chars): {question[:50]}...")
         logger.info(f"AdvancedAgent returning answer: {response.text}")
         return response.text