improvements

Author: shroominic

codeinterpreterapi/config.py

@@ -24,9 +24,6 @@ class CodeInterpreterAPISettings(BaseSettings):
     AZURE_API_VERSION: Optional[str] = None
     AZURE_DEPLOYMENT_NAME: Optional[str] = None
     ANTHROPIC_API_KEY: Optional[str] = None
-    OPENROUTER_API_KEY: Optional[str] = None
-    OPENROUTER_API_BASE: Optional[str] = "https://openrouter.ai/api/v1"
-    OPENROUTER_DEFAULT_CHAT_MODEL = "openai/gpt-4"
 
     # LLM Settings
     MODEL: str = "gpt-3.5-turbo"
@@ -44,7 +41,7 @@ class CodeInterpreterAPISettings(BaseSettings):
 
     # CodeBox
     CODEBOX_API_KEY: Optional[str] = None
-    CUSTOM_PACKAGES: list[str] = []  # implement custom logic for codebox
+    CUSTOM_PACKAGES: list[str] = []
 
     # deprecated
     VERBOSE: bool = DEBUG

codeinterpreterapi/session.py

@@ -152,13 +152,6 @@ def _choose_llm(self) -> BaseChatModel:
                 temperature=settings.TEMPERATURE,
                 anthropic_api_key=settings.ANTHROPIC_API_KEY,
             )
-        elif settings.OPENROUTER_API_KEY:
-            return ChatOpenAI(
-                model=settings.OPENROUTER_DEFAULT_CHAT_MODEL,
-                temperature=settings.TEMPERATURE,
-                openai_api_key=settings.OPENROUTER_API_KEY,
-                openai_api_base=settings.OPENROUTER_API_BASE,
-            )
         else:
             raise ValueError("Please set the API key for the LLM you want to use.")