log token usage

Element84 · Feb 14, 2025 · 87e7a4f · 87e7a4f
1 parent 91b15d7
commit 87e7a4f
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/e84_geoai_common/llm/models/claude.py b/src/e84_geoai_common/llm/models/claude.py
@@ -274,6 +274,8 @@ def invoke_model_with_request(self, request: ClaudeInvokeLLMRequest) -> ClaudeRe
             log.exception("Request body: %s", request.model_dump_json())
             raise
         response_body = response["body"].read().decode("UTF-8")
-        return ClaudeResponse.model_validate_json(response_body)
+        claude_response = ClaudeResponse.model_validate_json(response_body)
+        log.info("Token usage: %s", claude_response.usage)
+        return claude_response
 
     # FUTURE implement tool use
diff --git a/src/e84_geoai_common/llm/models/nova.py b/src/e84_geoai_common/llm/models/nova.py
@@ -240,7 +240,9 @@ def invoke_model_with_request(self, request: NovaInvokeLLMRequest) -> NovaRespon
             log.exception("Request body: %s", request.model_dump_json())
             raise
         response_body = response["body"].read().decode("UTF-8")
-        return NovaResponse.model_validate_json(response_body)
+        nova_response = NovaResponse.model_validate_json(response_body)
+        log.info("Token usage: %s", nova_response.usage)
+        return nova_response
 
 
 #########################