feat:Add application file

.env.example

@@ -1,3 +1,5 @@
-OPENAI_API_KEY= # your openai api key (required)
+OPENAI_API_KEY=  # your openai api key (required)
+OPENAI_PROXY=    # your openai proxy (optional)
+OPENAI_API_BASE= # your openai api base url (optional)
 CODEBOX_API_KEY= # your codebox api key (optional, required for production)
-VERBOSE=False # set to True to enable verbose logging
+VERBOSE=False    # set to True to enable verbose logging

frontend/app.py → app.py

@@ -12,12 +12,17 @@ st.title('Code Interpreter API 🚀')
 
 # This will create a sidebar
 st.sidebar.title("Code Interpreter API 🚀")
-st.sidebar.markdown("[Github Repo](https://github.com/shroominic/codeinterpreter-api)")
+
+st.sidebar.markdown("### 给勇哥哥打call 🚀")
+
+st.sidebar.markdown(
+    "![Code Interpreter](https://vercel.brzhang.club/_next/image?url=%2Fcoffer.jpg&w=640&q=75)")
 
 
 # This will create a textbox where you can input text
 input_text = st.text_area("Write your prompt")
-uploaded_files = st.file_uploader("Upload your files", accept_multiple_files=True)
+uploaded_files = st.file_uploader(
+    "Upload your files", accept_multiple_files=True)
 
 uploaded_files_list = []
 for uploaded_file in uploaded_files:
@@ -30,4 +35,4 @@ button_pressed = st.button('Run code interpreter', use_container_width=True)
 
 # This will display the images only when the button is pressed
 if button_pressed and input_text != "":
-    asyncio.run(get_images(input_text, files=uploaded_files_list))
+    asyncio.run(get_images(input_text, files=uploaded_files_list))

codeinterpreterapi/config.py

@@ -1,3 +1,4 @@
+from os import path
 from pydantic import BaseSettings
 from dotenv import load_dotenv
 from typing import Optional
@@ -15,6 +16,8 @@ class CodeInterpreterAPISettings(BaseSettings):
 
     CODEBOX_API_KEY: Optional[str] = None
     OPENAI_API_KEY: Optional[str] = None
+    OPENAI_API_BASE: Optional[str] = "https://api.openai.com/v1"
+    OPENAI_PROXY: Optional[str] = None
 
 
 settings = CodeInterpreterAPISettings()

codeinterpreterapi/session.py

@@ -1,4 +1,6 @@
-import uuid, base64, re
+import uuid
+import base64
+import re
 from io import BytesIO
 from typing import Optional
 from codeboxapi import CodeBox  # type: ignore
@@ -24,13 +26,16 @@ class CodeInterpreterSession:
         self,
         model=None,
         openai_api_key=settings.OPENAI_API_KEY,
+        openai_proxy=settings.OPENAI_PROXY,
+        openai_api_base=settings.OPENAI_API_BASE,
         verbose=settings.VERBOSE,
         tools: list[BaseTool] = None
     ) -> None:
         self.codebox = CodeBox()
         self.verbose = verbose
         self.tools: list[BaseTool] = self._tools(tools)
-        self.llm: BaseChatModel = self._llm(model, openai_api_key)
+        self.llm: BaseChatModel = self._llm(
+            model, openai_api_key, openai_proxy, openai_api_base)
         self.agent_executor: AgentExecutor = self._agent_executor()
         self.input_files: list[File] = []
         self.output_files: list[File] = []
@@ -43,8 +48,7 @@ class CodeInterpreterSession:
         return additional_tools + [
             StructuredTool(
                 name="python",
-                description=
-                # TODO: variables as context to the agent
+                description=  # TODO: variables as context to the agent
                 # TODO: current files as context to the agent
                 "Input a string of code to a python interpreter (jupyter kernel). "
                 "Variables are preserved between runs. ",
@@ -54,9 +58,14 @@ class CodeInterpreterSession:
             ),
         ]
 
-    def _llm(self, model: Optional[str] = None, openai_api_key: Optional[str] = None) -> BaseChatModel:
+    def _llm(self, model: Optional[str] = None, openai_api_key: Optional[str] = None, openai_proxy: Optional[str] = None, openai_api_base: Optional[str] = None) -> BaseChatModel:
+
+        print("OpenAI API Key:", openai_api_key)
+        print("OpenAI Proxy:", openai_proxy)
+        print("OpenAI API Base:", openai_api_base)
+        print("OpenAI Model:", model)
         if model is None:
-            model = "gpt-4"
+            model = "gpt-3.5-turbo"
 
         if openai_api_key is None:
             raise ValueError(
@@ -67,6 +76,8 @@ class CodeInterpreterSession:
             temperature=0.03,
             model=model,
             openai_api_key=openai_api_key,
+            # openai_proxy=openai_proxy,
+            openai_api_base=openai_api_base,
             max_retries=3,
             request_timeout=60 * 3,
         )  # type: ignore
@@ -76,7 +87,8 @@ class CodeInterpreterSession:
             llm=self.llm,
             tools=self.tools,
             system_message=code_interpreter_system_message,
-            extra_prompt_messages=[MessagesPlaceholder(variable_name="memory")],
+            extra_prompt_messages=[
+                MessagesPlaceholder(variable_name="memory")],
         )
 
     def _agent_executor(self) -> AgentExecutor:
@@ -86,7 +98,8 @@ class CodeInterpreterSession:
             max_iterations=9,
             tools=self.tools,
             verbose=self.verbose,
-            memory=ConversationBufferMemory(memory_key="memory", return_messages=True),
+            memory=ConversationBufferMemory(
+                memory_key="memory", return_messages=True),
         )
 
     async def show_code(self, code: str) -> None:
@@ -108,7 +121,8 @@ class CodeInterpreterSession:
             filename = f"image-{uuid.uuid4()}.png"
             file_buffer = BytesIO(base64.b64decode(output.content))
             file_buffer.name = filename
-            self.output_files.append(File(name=filename, content=file_buffer.read()))
+            self.output_files.append(
+                File(name=filename, content=file_buffer.read()))
             return f"Image {filename} got send to the user."
 
         elif output.type == "error":
@@ -118,7 +132,8 @@ class CodeInterpreterSession:
                 ):
                     await self.codebox.ainstall(package.group(1))
                     return f"{package.group(1)} was missing but got installed now. Please try again."
-            else: pass
+            else:
+                pass
                 # TODO: preanalyze error to optimize next code generation
             if self.verbose:
                 print("Error:", output.content)
@@ -157,7 +172,8 @@ class CodeInterpreterSession:
         for file in self.output_files:
             if str(file.name) in final_response:
                 # rm ![Any](file.name) from the response
-                final_response = re.sub(rf"\n\n!\[.*\]\(.*\)", "", final_response)
+                final_response = re.sub(
+                    rf"\n\n!\[.*\]\(.*\)", "", final_response)
 
         if self.output_files and re.search(rf"\n\[.*\]\(.*\)", final_response):
             final_response = await remove_download_link(final_response, self.llm)

requirements.txt

@@ -0,0 +1,9 @@
+python-dotenv>=1.0.0,<2.0.0
+openai>=0.27.8,<0.28.0
+langchain>=0.0.232,<0.1.0
+codeboxapi>=0.0.8,<0.1.0
+streamlit>=1.24.1,<1.25.0
+pytest>=7.4.0,<8.0.0
+mypy>=1.4.1,<2.0.0
+black>=23.7.0,<24.0.0
+Pillow>=8.0.0,<9.0.0