fix

agent.py

@@ -192,15 +192,14 @@ class GeminiAgent:
             max_print_outputs_length=2_000
             )
         
-        def check_token_safety(self, question):
-            messages = self.gemini_agent.memory.get_messages()
-            count = token_counter(model="gemma3", messages=messages)
-
-            print(f"📊 Current Context Size: {count} tokens")
-            return count
-
-        
         print(f"✅ Gemini agent initialized with model: {model_id}")
+        
+    def check_token_safety(self, question):
+        messages = self.gemini_agent.memory.get_messages()
+        count = token_counter(model="gemma3", messages=messages)
+
+        print(f"📊 Current Context Size: {count} tokens")
+        return count
                 
     def __call__(self, question: str, file_path: Optional[str] = None) -> str:
         prompt = f"{self.system_prompt}\n\nQuestion: {question}"