Update app.py

app.py

@@ -1,6 +1,7 @@
 import yaml
 import os
-from smolagents import GradioUI, CodeAgent
+from smolagents import GradioUI, CodeAgent, LiteLLMModel
+from litellm import token_counter  # fallback when provider doesn't return usage
 
 # Get current directory path
 CURRENT_DIR = os.path.dirname(os.path.abspath(__file__))
@@ -46,8 +47,8 @@ class CountedLiteLLMModel(LiteLLMModel):
         return out
 
 model = CountedLiteLLMModel(
-model_id='claude-3-5-sonnet-latest',
-api_key=os.getenv('ANTHROPIC_API_KEY'),
+    model_id='claude-3-5-sonnet-latest',
+    api_key=os.getenv('ANTHROPIC_API_KEY'),
 )
 
 web_search = WebSearch()