fix gemini test

defog-ai · Feb 6, 2025 · d25cc32 · d25cc32
1 parent 4e9b321
commit d25cc32
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/test_utils_llm.py b/tests/test_utils_llm.py
@@ -239,5 +239,4 @@ def test_chat_gemini_sql(self):
         print(response)
         self.assertIsInstance(response, LLMResponse)
         self.check_sql(response.content)
-        self.assertEqual(response.input_tokens, 87)
         self.assertTrue(response.output_tokens < 10)