📠 fix typing for python3.9

codeinterpreterapi/chains/modifications_check.py

@@ -1,5 +1,5 @@
 import json
-from typing import List
+from typing import List, Optional
 
 from langchain.base_language import BaseLanguageModel
 from langchain.chat_models.openai import ChatOpenAI
@@ -12,7 +12,7 @@ async def get_file_modifications(
     code: str,
     llm: BaseLanguageModel,
     retry: int = 2,
-) -> List[str] | None:
+) -> Optional[List[str]]:
     if retry < 1:
         return None
     messages = determine_modifications_prompt.format_prompt(code=code).to_messages()

codeinterpreterapi/config.py

@@ -1,5 +1,6 @@
 from pydantic import BaseSettings
 from dotenv import load_dotenv
+from typing import Optional
 
 # .env file
 load_dotenv()
@@ -12,8 +13,8 @@ class CodeInterpreterAPISettings(BaseSettings):
 
     VERBOSE: bool = False
 
-    CODEBOX_API_KEY: str | None = None
-    OPENAI_API_KEY: str | None = None
+    CODEBOX_API_KEY: Optional[str] = None
+    OPENAI_API_KEY: Optional[str] = None
 
 
 settings = CodeInterpreterAPISettings()

codeinterpreterapi/session.py

@@ -1,5 +1,6 @@
 import uuid, base64, re
 from io import BytesIO
+from typing import Optional
 from codeboxapi import CodeBox  # type: ignore
 from codeboxapi.schema import CodeBoxOutput  # type: ignore
 from langchain.tools import StructuredTool
@@ -45,7 +46,7 @@ class CodeInterpreterSession:
             ),
         ]
 
-    def _llm(self, model: str | None, openai_api_key: str | None) -> BaseChatModel:
+    def _llm(self, model: Optional[str] = None, openai_api_key: Optional[str] = None) -> BaseChatModel:
         if model is None:
             model = "gpt-4"