fix interface

jaytoday · Jul 14, 2023 · 2b0a7db · 2b0a7db
1 parent 18fc95f
commit 2b0a7db
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/codeinterpreterapi/session.py b/codeinterpreterapi/session.py
@@ -152,11 +152,12 @@ async def output_handler(self, final_response: str) -> CodeInterpreterResponse:
 
     async def generate_response(
         self, 
-        user_request: UserRequest, 
-        files: list[File], # list of "file_name" x "file_content"
+        user_msg: str, 
+        files: list[File] = [],
         detailed_error: bool = False,
     ) -> CodeInterpreterResponse:
         """ Generate a Code Interpreter response based on the user's input."""
+        user_request = UserRequest(content=user_msg, files=files)
         try:
             await self.input_handler(user_request)
             response = await self.agent_executor.arun(input=user_request.content)