init wip

codeinterpreterapi/config.py

@@ -0,0 +1,18 @@
+from pydantic import BaseSettings
+from dotenv import load_dotenv
+
+# .env file
+load_dotenv()
+
+
+class CodeInterpreterAPISettings(BaseSettings):
+    """
+    CodeInterpreter API Config
+    """
+    CODEBOX_API_KEY: str = None
+    CODEBOX_VERBOSE: bool = False
+    
+    OPENAI_API_KEY: str = ""
+    
+
+settings = CodeInterpreterAPISettings()

codeinterpreterapi/schemas.py

@@ -0,0 +1,13 @@
+from promptkit import AssistantResponse
+from promptkit.schema import UserMessage
+
+
+class UserRequest(UserMessage):
+    text: str = ""
+    files: list[dict[str, bytes]] = []
+
+
+class CodeInterpreterResponse(AssistantResponse):
+    files: list[dict[str, bytes]] = []
+    final_code: str = ""
+    final_output: str = ""

codeinterpreterapi/session.py

@@ -0,0 +1,46 @@
+from codeboxapi import CodeBox
+from promptkit import ChatSession
+
+
+class CodeInterpreterSession():
+    
+    def __init__(self):
+        self.chatgpt = ChatSession()
+        self.codebox = CodeBox()
+    
+    async def _init(self):
+        await self.codebox.start()
+        
+    async def _close(self):
+        await self.codebox.stop()
+    
+    async def code_decision(self, user_request: str):
+        # check if the user wants something that requires python code execution
+        # if yes, return "code"
+        # if no, return "default"
+        pass
+    
+    async def generate_response(self, text: str, files: list[dict[str, bytes]]):  # list of "file_name" x "file_content"
+        """ Generate a Code Interpreter response based on the user's input."""
+        if self.code_decision() == "code":
+            pass
+            # plan what code to write (potentially multiple steps)
+            # code = chatgpt.run(code generation template)
+            # codebox.run(code)
+                # on error
+                    # check if package is required
+                        # if yes, install package
+                    # ask for analysis if the error can be fixed
+                        # if yes, continue code generation
+                        # if no, return AssistantResponse
+                # on success
+                    # check if to output files to the user
+                        # if yes, return AssistantResponse with files
+            # write a response based on the code execution
+            # return AssistantResponse
+        else:
+            pass
+            # return AssistantResponse
+        pass
+
+