✨cleanup examples

Author: shroominic

examples/analyze_dataset.py

@@ -1,7 +1,7 @@
 from codeinterpreterapi import CodeInterpreterSession, File
 
 
-async def main():
+async def main() -> None:
     # context manager for start/stop of the session
     async with CodeInterpreterSession() as session:
         # define the user request

examples/chat_history_backend.py

@@ -6,7 +6,7 @@
 from codeinterpreterapi import CodeInterpreterSession  # noqa: E402
 
 
-def main():
+def main() -> None:
     session_id = None
 
     session = CodeInterpreterSession()

frontend/__init__.py examples/frontend/__init__.pyfrontend/__init__.py renamed to examples/frontend/__init__.py

@@ -1,11 +1,12 @@
-import chainlit as cl
+import chainlit as cl  # type: ignore
 from codeinterpreterapi import CodeInterpreterSession
 from codeinterpreterapi import File as CIFile
 
-UPLOADED_FILES = []
+UPLOADED_FILES: list[CIFile] = []
+
 
 @cl.action_callback("upload_file")
-async def on_action(action):
+async def on_action(action: cl.Action) -> None:
     files = None
 
     # Wait for the user to upload a file
@@ -25,8 +26,9 @@ async def on_action(action):
     ).send()
     await action.remove()
 
+
 @cl.on_chat_start
-async def start_chat():
+async def start_chat() -> None:
     actions = [
         cl.Action(name="upload_file", value="example_value", description="Upload file")
     ]
@@ -35,14 +37,14 @@ async def start_chat():
         content="Hello, How can I assist you today", actions=actions
     ).send()
 
+
 @cl.on_message
-async def run_conversation(user_message: str):
+async def run_conversation(user_message: str) -> None:
     session = CodeInterpreterSession()
     await session.astart()
 
     files = [CIFile(name=it.name, content=it.content) for it in UPLOADED_FILES]
 
-
     response = await session.agenerate_response(user_message, files=files)
     elements = [
         cl.Image(
@@ -61,4 +63,4 @@ async def run_conversation(user_message: str):
         actions=actions,
     ).send()
 
-    await session.astop()
+    await session.astop()

frontend/app.py examples/frontend/app.pyfrontend/app.py renamed to examples/frontend/app.py

@@ -16,7 +16,7 @@ def create_temp_folder() -> str:
     return temp_folder
 
 
-async def get_images(prompt: str, files: Optional[list] = None):
+async def get_images(prompt: str, files: Optional[list] = None) -> list:
     if files is None:
         files = []
     with st.chat_message("user"):  # type: ignore
@@ -53,3 +53,4 @@ async def get_images(prompt: str, files: Optional[list] = None):
                             f,
                             file_name="archive.zip",
                         )
+    return response.files

frontend/chainlitui.py examples/frontend/chainlitui.pyfrontend/chainlitui.py renamed to examples/frontend/chainlitui.py

@@ -1,7 +1,7 @@
 from codeinterpreterapi import CodeInterpreterSession
 
 
-async def main():
+async def main() -> None:
     async with CodeInterpreterSession() as session:
         response = await session.agenerate_response(
             "Plot a sin wave and show it to me."

frontend/utils.py examples/frontend/utils.pyfrontend/utils.py renamed to examples/frontend/utils.py

@@ -3,7 +3,7 @@
 from codeinterpreterapi import CodeInterpreterSession
 
 
-def main():
+def main() -> None:
     with CodeInterpreterSession(local=True) as session:
         currentdate = datetime.now().strftime("%Y-%m-%d")
 

examples/plot_sin_wave.py

@@ -17,10 +17,10 @@ class ExampleKnowledgeBaseTool(BaseTool):
     name: str = "salary_database"
     description: str = "Use to get salary data of company employees"
 
-    def _run(self, *args, **kwargs):
+    def _run(self, *args: Any, **kwargs: Any) -> Any:
         raise NotImplementedError()
 
-    async def _arun(self, *args, **kwargs: Any) -> Any:
+    async def _arun(self, *args: Any, **kwargs: Any) -> Any:
         f = io.StringIO()
         writer = csv.writer(f)
         writer.writerow(["month", "employee", "salary"])
@@ -31,7 +31,7 @@ async def _arun(self, *args, **kwargs: Any) -> Any:
         return f.getvalue()
 
 
-async def main():
+async def main() -> None:
     async with CodeInterpreterSession(
         additional_tools=[ExampleKnowledgeBaseTool()]
     ) as session: