diff --git a/agent/__init__.py b/agent/__init__.py
index 8bce473..291d81e 100644
--- a/agent/__init__.py
+++ b/agent/__init__.py
@@ -31,8 +31,8 @@
 OPENAI_API_BASE = os.getenv("OPENAI_API_BASE")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 OPENAI_MODEL_NAME = os.getenv("OPENAI_MODEL_NAME")
-OPENAI_TEMPERATURE = os.getenv("OPENAI_TEMPERATURE", 0.3)
-OPENAI_MODEL_SEED = os.getenv("OPENAI_MODEL_SEED", 11111111)
+OPENAI_TEMPERATURE = float(os.getenv("OPENAI_TEMPERATURE", 0.3))
+OPENAI_MODEL_SEED = int(os.getenv("OPENAI_MODEL_SEED", 11111111))
 
 
 class Purl(BaseModel):
@@ -166,7 +166,7 @@ class CWEFromSummaryParser(BaseParser):
     def __init__(self):
         super().__init__(PROMPT_CWE_FROM_SUMMARY, CWEList)
 
-    def get_cwes(self, summary: str) -> List[CWEList]:
+    def get_cwes(self, summary: str) -> list[str]:
         result = self.run_agent(f"**Vulnerability Description:**\n{summary}")
         return [cwe.string for cwe in result.output.cwes]