fix: support usage as part of generation (#192)

Author: k11kirky

CHANGELOG.md

@@ -1,3 +1,8 @@
+
+## 3.14.1 - 2025-02-18
+
+1. Add support for Bedrock Anthropic Usage
+
 ## 3.13.0 - 2025-02-12
 
 1. Automatically retry connection errors

posthog/ai/langchain/callbacks.py

@@ -597,6 +597,9 @@ def _parse_usage_model(
         # Bedrock: https://docs.aws.amazon.com/bedrock/latest/userguide/monitoring-cw.html#runtime-cloudwatch-metrics
         ("inputTokenCount", "input"),
         ("outputTokenCount", "output"),
+        # Bedrock Anthropic
+        ("prompt_tokens", "input"),
+        ("completion_tokens", "output"),
         # langchain-ibm https://pypi.org/project/langchain-ibm/
         ("input_token_count", "input"),
         ("generated_token_count", "output"),
@@ -627,6 +630,10 @@ def _parse_usage(response: LLMResult):
 
     if hasattr(response, "generations"):
         for generation in response.generations:
+            if "usage" in generation:
+                llm_usage = _parse_usage_model(generation["usage"])
+                break
+
             for generation_chunk in generation:
                 if generation_chunk.generation_info and ("usage_metadata" in generation_chunk.generation_info):
                     llm_usage = _parse_usage_model(generation_chunk.generation_info["usage_metadata"])

posthog/version.py

@@ -1,4 +1,4 @@
-VERSION = "3.14.0"
+VERSION = "3.14.1"
 
 if __name__ == "__main__":
     print(VERSION, end="")  # noqa: T201