ruff

core/source/clients/base_device.py

@@ -4,20 +4,14 @@
 import os
 import asyncio
 import threading
-import os
 import pyaudio
-from starlette.websockets import WebSocket
-from queue import Queue
 from pynput import keyboard
 import json
 import traceback
 import websockets
 import queue
-import pydub
-import ast
 from pydub import AudioSegment
 from pydub.playback import play
-import io
 import time
 import wave
 import tempfile

core/source/server/conftest.py

@@ -1,8 +1,5 @@
-import os
-import sys
 import pytest
 from source.server.i import configure_interpreter
-from unittest.mock import Mock
 from interpreter import OpenInterpreter
 from fastapi.testclient import TestClient
 from .server import app

core/source/server/i.py

@@ -1,11 +1,9 @@
 from dotenv import load_dotenv
 load_dotenv()  # take environment variables from .env.
 
-import os
 import glob
 import time
 import json
-from pathlib import Path
 from interpreter import OpenInterpreter
 import shutil
 
@@ -295,7 +293,6 @@ def configure_interpreter(interpreter: OpenInterpreter):
     #     if chunk.get("format") != "active_line":
     #         print(chunk.get("content"))
 
-    import os
 
     from platformdirs import user_data_dir
 

core/source/server/server.py

@@ -3,7 +3,6 @@
 
 import traceback
 from platformdirs import user_data_dir
-import ast
 import json
 import queue
 import os
@@ -13,9 +12,7 @@
 from fastapi import FastAPI, Request
 from fastapi.responses import PlainTextResponse
 from starlette.websockets import WebSocket, WebSocketDisconnect
-from pathlib import Path
 import asyncio
-import urllib.parse
 from .utils.kernel import put_kernel_messages_into_queue
 from .i import configure_interpreter
 from interpreter import interpreter
@@ -362,7 +359,7 @@ def stream_tts(sentence):
 async def startup_event():
     server_url = f"{HOST}:{PORT}"
     print("")
-    print_markdown(f"\n*Ready.*\n")
+    print_markdown("\n*Ready.*\n")
     print("")
 
 @app.on_event("shutdown")

core/source/server/services/stt/local-whisper/stt.py

@@ -10,9 +10,6 @@
 import ffmpeg
 import subprocess
 
-import os
-import subprocess
-import platform
 import urllib.request
 
 

core/source/server/services/tts/openai/tts.py

@@ -2,8 +2,6 @@
 import tempfile
 from openai import OpenAI
 import os
-import subprocess
-import tempfile
 
 from source.server.utils.logs import logger
 from source.server.utils.logs import setup_logging
@@ -12,7 +10,7 @@
 # If this TTS service is used, the OPENAI_API_KEY environment variable must be set
 if not os.getenv('OPENAI_API_KEY'):
     logger.error("")
-    logger.error(f"OpenAI API key not found. Please set the OPENAI_API_KEY environment variable, or run 01 with the --local option.")
+    logger.error("OpenAI API key not found. Please set the OPENAI_API_KEY environment variable, or run 01 with the --local option.")
     logger.error("Aborting...")
     logger.error("")
     os._exit(1)

core/source/server/skills/schedule.py

@@ -3,7 +3,6 @@
 from pytimeparse import parse
 from crontab import CronTab
 from uuid import uuid4
-from datetime import datetime
 from platformdirs import user_data_dir
 
 def schedule(message="", start=None, interval=None) -> None:

core/source/server/tests/test_run.py

@@ -1,10 +1,5 @@
 # test_main.py
-import subprocess
-import uuid
 import pytest
-from source.server.i import configure_interpreter
-from unittest.mock import Mock
-from fastapi.testclient import TestClient
 
 
 

core/source/server/tunnel.py

@@ -1,12 +1,10 @@
-import os
 import subprocess
 import re
-import shutil
 import time
 from ..utils.print_markdown import print_markdown
 
 def create_tunnel(tunnel_method='ngrok', server_host='localhost', server_port=10001):
-    print_markdown(f"Exposing server to the internet...")
+    print_markdown("Exposing server to the internet...")
 
     if tunnel_method == "bore":
         try:

core/source/server/utils/get_system_info.py

@@ -1,4 +1,3 @@
-import os
 import platform
 
 def get_system_info():

core/source/server/utils/local_mode.py

@@ -1,6 +1,4 @@
 import sys
-import os
-import platform
 import subprocess
 import time
 import inquirer
@@ -67,7 +65,7 @@ def select_local_model():
             if not names:
                 time.sleep(1)
 
-                interpreter.display_message(f"\nYou don't have any Ollama models downloaded. To download a new model, run `ollama run <model-name>`, then start a new 01 session. \n\n For a full list of downloadable models, check out [https://ollama.com/library](https://ollama.com/library) \n")
+                interpreter.display_message("\nYou don't have any Ollama models downloaded. To download a new model, run `ollama run <model-name>`, then start a new 01 session. \n\n For a full list of downloadable models, check out [https://ollama.com/library](https://ollama.com/library) \n")
 
                 print("Please download a model then try again\n")
                 time.sleep(2)
@@ -91,10 +89,10 @@ def select_local_model():
                 time.sleep(1)
 
         # If Ollama is not installed or not recognized as a command, prompt the user to download Ollama and try again
-        except (subprocess.CalledProcessError, FileNotFoundError) as e:
+        except (subprocess.CalledProcessError, FileNotFoundError):
             print("Ollama is not installed or not recognized as a command.")
             time.sleep(1)
-            interpreter.display_message(f"\nPlease visit [https://ollama.com/](https://ollama.com/) to download Ollama and try again\n")
+            interpreter.display_message("\nPlease visit [https://ollama.com/](https://ollama.com/) to download Ollama and try again\n")
             time.sleep(2)
             sys.exit(1)
 

core/source/server/utils/process_utils.py

@@ -25,4 +25,4 @@ def kill_process_tree():
     except psutil.NoSuchProcess:
         print(f"Process {pid} does not exist or is already terminated")
     except psutil.AccessDenied:
-        print(f"Permission denied to terminate some processes")
+        print("Permission denied to terminate some processes")