checkpoint: massive refactor

tests/test_token_usage.py

@@ -8,8 +8,8 @@ sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 import ai_client
 
 def test_token_usage_tracking():
-    ai_client.reset_session()
-    # Mock an API response with token usage
-    usage = {"prompt_tokens": 100, "candidates_tokens": 50, "total_tokens": 150}
-    # This would test the internal accumulator in ai_client
-    pass
+ ai_client.reset_session()
+ # Mock an API response with token usage
+ usage = {"prompt_tokens": 100, "candidates_tokens": 50, "total_tokens": 150}
+ # This would test the internal accumulator in ai_client
+ pass