fix: add haiku price table

aws-samples · Mar 18, 2024 · 63f8bdb · 63f8bdb
1 parent 643fee3
commit 63f8bdb
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 8 deletions.
diff --git a/backend/app/bedrock.py b/backend/app/bedrock.py
@@ -65,14 +65,16 @@ def compose_args_for_anthropic_client(
 def calculate_price(
     model: str, input_tokens: int, output_tokens: int, region: str = BEDROCK_REGION
 ) -> float:
-    if region in ("us-east-1", "us-west-2", "ap-northeast-1"):
-        input_price = ANTHROPIC_PRICING[region][model]["input"]
-        output_price = ANTHROPIC_PRICING[region][model]["output"]
-
-    else:
-        logger.warning(f"Unsupported region: {region}. Using default pricing.")
-        input_price = ANTHROPIC_PRICING["default"][model]["input"]
-        output_price = ANTHROPIC_PRICING["default"][model]["output"]
+    input_price = (
+        ANTHROPIC_PRICING.get(region, {})
+        .get(model, {})
+        .get("input", ANTHROPIC_PRICING["default"][model]["input"])
+    )
+    output_price = (
+        ANTHROPIC_PRICING.get(region, {})
+        .get(model, {})
+        .get("output", ANTHROPIC_PRICING["default"][model]["output"])
+    )
 
     return input_price * input_tokens / 1000.0 + output_price * output_tokens / 1000.0
 

diff --git a/backend/app/config.py b/backend/app/config.py
@@ -36,6 +36,7 @@
             "input": 0.00080,
             "output": 0.00240,
         },
+        "claude-v3-haiku": {"input": 0.00025, "output": 0.00125},
         "claude-v3-sonnet": {"input": 0.00300, "output": 0.01500},
     },
     "us-west-2": {
@@ -68,6 +69,7 @@
             "input": 0.00080,
             "output": 0.00240,
         },
+        "claude-v3-haiku": {"input": 0.00025, "output": 0.00125},
         "claude-v3-sonnet": {"input": 0.00300, "output": 0.01500},
     },
 }