Fix type issues

plasma-umass · Mar 5, 2024 · f53f257 · f53f257
1 parent 9e3c34b
commit f53f257
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/llm_utils/utils.py b/src/llm_utils/utils.py
@@ -40,7 +40,7 @@ def calculate_cost(
     """
     # Latest pricing info from OpenAI (https://openai.com/pricing and
     # https://platform.openai.com/docs/deprecations/), as of March 5, 2024.
-    PRICING_PER_MILLION = {
+    PRICING_PER_MILLION: Dict[str, Dict[str, float]] = {
         # gpt-4-turbo.
         "gpt-4-0125-preview": {"input": 10, "output": 30},
         "gpt-4-turbo-preview": {"input": 10, "output": 30},