From f53f2575a75426b46b4d801f4fbee231ca4a78a2 Mon Sep 17 00:00:00 2001
From: Nicolas van Kempen <nvankemp@gmail.com>
Date: Tue, 5 Mar 2024 18:25:25 +0000
Subject: [PATCH] Fix type issues

---
 src/llm_utils/utils.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/llm_utils/utils.py b/src/llm_utils/utils.py
index 6330a24..a83aa8b 100644
--- 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},