Switch cost estimation to genai-prices with disclaimer

Author: saksham-jain177

pyproject.toml

@@ -35,6 +35,7 @@ dependencies = [
     "tomli==2.2.1",
     "claude-agent-sdk>=0.1.0",
     "tiktoken==0.12.0",
+    "genai-prices==0.0.51",
 ]
 classifiers = [
     "Development Status :: 5 - Production/Stable",

python_gpt_po/main.py

@@ -231,6 +231,7 @@ def main():
                 cost_str = f"${data['cost']:.4f}" if data['cost'] is not None else "unavailable"
                 print(f"  - {lang:5}: {data['tokens']:8,} tokens | {cost_str}")
 
+            print("\nNote: Cost estimates are approximate and may not reflect current provider pricing.")
             print(f"{'=' * 40}\n")
 
             if estimation['total_tokens'] == 0:

python_gpt_po/tests/unit/test_cost_estimator.py

@@ -37,7 +37,7 @@ def test_minimal_token_math(self):
         self.assertEqual(t3, t1 * 3)
 
     def test_pricing_lookup(self):
-        """Verify static pricing lookup."""
+        """Verify dynamic pricing lookup via genai-prices."""
         po_path = os.path.join(self.test_dir, "test.po")
         po = polib.POFile()
         po.append(polib.POEntry(msgid="Test", msgstr=""))

python_gpt_po/utils/cost_estimator.py

@@ -11,20 +11,8 @@
 
 
 class CostEstimator:
-    """Estimates token usage and costs for translation tasks."""
-
-    # Pricing per 1,000 tokens (Input price, Output price)
-    # Data sourced from provider websites (Jan 2026)
-    PRICING = {
-        "gpt-4o": (0.0025, 0.010),
-        "gpt-4o-mini": (0.00015, 0.0006),
-        "claude-3-5-sonnet": (0.003, 0.015),
-        "claude-3-5-sonnet-20241022": (0.003, 0.015),
-        "deepseek-chat": (0.00014, 0.00028),
-        "deepseek-v3": (0.00014, 0.00028),
-    }
-
-    OUTPUT_MULTIPLIER = 1.3  # Conservative estimate for translation expansion
+    # Conservative estimate for translation expansion
+    OUTPUT_MULTIPLIER = 1.3
 
     @classmethod
     def estimate_cost(
@@ -124,5 +112,15 @@ def _get_token_count(text: str, model: str) -> int:
 
     @classmethod
     def _get_pricing(cls, model: str) -> Optional[Tuple[float, float]]:
-        """Lookup pricing for the active model name."""
-        return cls.PRICING.get(model)
+        """Lookup pricing for the active model name using genai-prices."""
+        try:
+            import genai_prices
+            from genai_prices.types import Usage
+
+            # Exact match only. genai-prices raises LookupError if not found.
+            # We use a unit usage of 1000 tokens to get the price per 1k tokens.
+            usage = Usage(input_tokens=1000, output_tokens=1000)
+            price_detail = genai_prices.calc_price(usage, model)
+            return float(price_detail.input_price), float(price_detail.output_price)
+        except (ImportError, LookupError, Exception):
+            return None

requirements.txt

@@ -11,3 +11,4 @@ responses==0.25.6
 isort==6.0.1
 tomli==2.2.1
 tiktoken==0.12.0
+genai-prices==0.0.51