pre-commit all files

shroominic · Aug 7, 2023 · cf641fd · cf641fd
1 parent abc5993
commit cf641fd
Show file tree

Hide file tree

Showing 6 changed files with 46 additions and 29 deletions.
diff --git a/.env.example b/.env.example
@@ -8,4 +8,4 @@ VERBOSE=False
 # OPENAI_API_TYPE=azure
 # OPENAI_API_VERSION=2023-07-01-preview
 # OPENAI_API_BASE=
-# DEPLOYMENT_NAME=
+# DEPLOYMENT_NAME=
diff --git a/README.md b/README.md
@@ -31,18 +31,18 @@ To configure OpenAI and Azure OpenAI, ensure that you set the appropriate enviro
 
 For OpenAI, set the OPENAI_API_KEY environment variable:
 ```
-export OPENAI_API_KEY=your_openai_api_key  
+export OPENAI_API_KEY=your_openai_api_key
 ```
- 
+
 For Azure OpenAI, set the following environment variables:
 ```
-export OPENAI_API_TYPE=azure  
-export OPENAI_API_VERSION=your_api_version  
-export OPENAI_API_BASE=your_api_base  
-export OPENAI_API_KEY=your_azure_openai_api_key  
-export DEPLOYMENT_NAME=your_deployment_name  
+export OPENAI_API_TYPE=azure
+export OPENAI_API_VERSION=your_api_version
+export OPENAI_API_BASE=your_api_base
+export OPENAI_API_KEY=your_azure_openai_api_key
+export DEPLOYMENT_NAME=your_deployment_name
 ```
- 
+
 Remember to replace the placeholders with your actual API keys and other required information.
 
 ```python

diff --git a/codeinterpreterapi/agents/functions_agent.py b/codeinterpreterapi/agents/functions_agent.py
@@ -16,13 +16,22 @@
 from langchain.callbacks.base import BaseCallbackManager
 from langchain.callbacks.manager import Callbacks
 from langchain.chat_models.openai import ChatOpenAI
-from langchain.prompts.chat import (BaseMessagePromptTemplate,
-                                    ChatPromptTemplate,
-                                    HumanMessagePromptTemplate,
-                                    MessagesPlaceholder)
-from langchain.schema import (AgentAction, AgentFinish, AIMessage, BaseMessage,
-                              BasePromptTemplate, FunctionMessage,
-                              OutputParserException, SystemMessage)
+from langchain.prompts.chat import (
+    BaseMessagePromptTemplate,
+    ChatPromptTemplate,
+    HumanMessagePromptTemplate,
+    MessagesPlaceholder,
+)
+from langchain.schema import (
+    AgentAction,
+    AgentFinish,
+    AIMessage,
+    BaseMessage,
+    BasePromptTemplate,
+    FunctionMessage,
+    OutputParserException,
+    SystemMessage,
+)
 from langchain.tools import BaseTool
 from langchain.tools.convert_to_openai import format_tool_to_openai_function
 from pydantic import root_validator

diff --git a/codeinterpreterapi/prompts/remove_dl_link.py b/codeinterpreterapi/prompts/remove_dl_link.py
@@ -1,5 +1,4 @@
-from langchain.prompts.chat import (ChatPromptTemplate,
-                                    HumanMessagePromptTemplate)
+from langchain.prompts.chat import ChatPromptTemplate, HumanMessagePromptTemplate
 from langchain.schema import AIMessage, HumanMessage, SystemMessage
 
 remove_dl_link_prompt = ChatPromptTemplate(

diff --git a/codeinterpreterapi/session.py b/codeinterpreterapi/session.py
@@ -22,15 +22,20 @@
 from langchain.tools import BaseTool, StructuredTool
 
 from codeinterpreterapi.agents import OpenAIFunctionsAgent
-from codeinterpreterapi.chains import (get_file_modifications,
-                                       remove_download_link)
+from codeinterpreterapi.chains import get_file_modifications, remove_download_link
 from codeinterpreterapi.config import settings
 from codeinterpreterapi.prompts import code_interpreter_system_message
-from codeinterpreterapi.schema import (CodeInput, CodeInterpreterResponse,
-                                       File, UserRequest)
-from codeinterpreterapi.utils import (CodeAgentOutputParser,
-                                      CodeCallbackHandler,
-                                      CodeChatAgentOutputParser)
+from codeinterpreterapi.schema import (
+    CodeInput,
+    CodeInterpreterResponse,
+    File,
+    UserRequest,
+)
+from codeinterpreterapi.utils import (
+    CodeAgentOutputParser,
+    CodeCallbackHandler,
+    CodeChatAgentOutputParser,
+)
 
 
 class CodeInterpreterSession:
@@ -86,15 +91,20 @@ def _choose_llm(
             openai_api_base = getenv("OPENAI_API_BASE")
             deployment_name = getenv("DEPLOYMENT_NAME")
             openapi_type = getenv("OPENAI_API_TYPE")
-            if openapi_type == "azure" and openai_api_version and openai_api_base and deployment_name:
+            if (
+                openapi_type == "azure"
+                and openai_api_version
+                and openai_api_base
+                and deployment_name
+            ):
                 return AzureChatOpenAI(
                     temperature=0.03,
                     openai_api_base=openai_api_base,
                     openai_api_version=openai_api_version,
                     deployment_name=deployment_name,
                     openai_api_key=openai_api_key,
                     max_retries=3,
-                    request_timeout=60*3
+                    request_timeout=60 * 3,
                 )
             else:
                 return ChatOpenAI(

diff --git a/codeinterpreterapi/utils/parser.py b/codeinterpreterapi/utils/parser.py
@@ -39,8 +39,7 @@ def _type(self) -> str:
 
 class CodeChatAgentOutputParser(AgentOutputParser):
     def get_format_instructions(self) -> str:
-        from langchain.agents.conversational_chat.prompt import \
-            FORMAT_INSTRUCTIONS
+        from langchain.agents.conversational_chat.prompt import FORMAT_INSTRUCTIONS
 
         return FORMAT_INSTRUCTIONS