Use ruff instead of black for performance boost

.github/workflows/ruff.yml

@@ -0,0 +1,8 @@
+name: Ruff
+on: [push, pull_request]
+jobs:
+  ruff:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions/checkout@v4
+      - uses: chartboost/ruff-action@v1

.pre-commit-config.yaml

@@ -1,11 +1,10 @@
 repos:
-  # Using this mirror lets us use mypyc-compiled black, which is about 2x faster
-  - repo: https://github.com/psf/black-pre-commit-mirror
-    rev: 24.3.0
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    # Ruff version.
+    rev: v0.3.7
     hooks:
-      - id: black
-        # It is recommended to specify the latest version of Python
-        # supported by your project here, or alternatively use
-        # pre-commit's default_language_version, see
-        # https://pre-commit.com/#top_level-default_language_version
-        language_version: python3.10
+      # Run the linter.
+      - id: ruff
+        args: [ --fix ]
+      # Run the formatter.
+      - id: ruff-format

app.py

@@ -1078,14 +1078,10 @@ async def delete_conversation():
             raise Exception("CosmosDB is not configured or not working")
 
         ## delete the conversation messages from cosmos first
-        deleted_messages = await cosmos_conversation_client.delete_messages(
-            conversation_id, user_id
-        )
+        await cosmos_conversation_client.delete_messages(conversation_id, user_id)
 
         ## Now delete the conversation
-        deleted_conversation = await cosmos_conversation_client.delete_conversation(
-            user_id, conversation_id
-        )
+        await cosmos_conversation_client.delete_conversation(user_id, conversation_id)
 
         await cosmos_conversation_client.cosmosdb_client.close()
 
@@ -1246,12 +1242,12 @@ async def delete_all_conversations():
         # delete each conversation
         for conversation in conversations:
             ## delete the conversation messages from cosmos first
-            deleted_messages = await cosmos_conversation_client.delete_messages(
+            await cosmos_conversation_client.delete_messages(
                 conversation["id"], user_id
             )
 
             ## Now delete the conversation
-            deleted_conversation = await cosmos_conversation_client.delete_conversation(
+            await cosmos_conversation_client.delete_conversation(
                 user_id, conversation["id"]
             )
         await cosmos_conversation_client.cosmosdb_client.close()
@@ -1289,9 +1285,7 @@ async def clear_messages():
             raise Exception("CosmosDB is not configured or not working")
 
         ## delete the conversation messages from cosmos
-        deleted_messages = await cosmos_conversation_client.delete_messages(
-            conversation_id, user_id
-        )
+        await cosmos_conversation_client.delete_messages(conversation_id, user_id)
 
         return (
             jsonify(
@@ -1367,7 +1361,7 @@ async def generate_title(conversation_messages):
 
         title = json.loads(response.choices[0].message.content)["title"]
         return title
-    except Exception as e:
+    except Exception:
         return messages[-2]["content"]
 
 

backend/history/cosmosdbservice.py

@@ -51,16 +51,16 @@ async def ensure(self):
             return False, "CosmosDB client not initialized correctly"
 
         try:
-            database_info = await self.database_client.read()
-        except:
+            await self.database_client.read()
+        except Exception:
             return (
                 False,
                 f"CosmosDB database {self.database_name} on account {self.cosmosdb_endpoint} not found",
             )
 
         try:
-            container_info = await self.container_client.read()
-        except:
+            await self.container_client.read()
+        except Exception:
             return False, f"CosmosDB container {self.container_name} not found"
 
         return True, "CosmosDB client initialized successfully"
@@ -131,7 +131,7 @@ async def get_conversation(self, user_id, conversation_id):
             {"name": "@conversationId", "value": conversation_id},
             {"name": "@userId", "value": user_id},
         ]
-        query = f"SELECT * FROM c where c.id = @conversationId and c.type='conversation' and c.userId = @userId"
+        query = "SELECT * FROM c where c.id = @conversationId and c.type='conversation' and c.userId = @userId"
         conversations = []
         async for item in self.container_client.query_items(
             query=query, parameters=parameters
@@ -187,7 +187,7 @@ async def get_messages(self, user_id, conversation_id):
             {"name": "@conversationId", "value": conversation_id},
             {"name": "@userId", "value": user_id},
         ]
-        query = f"SELECT * FROM c WHERE c.conversationId = @conversationId AND c.type='message' AND c.userId = @userId ORDER BY c.timestamp ASC"
+        query = "SELECT * FROM c WHERE c.conversationId = @conversationId AND c.type='message' AND c.userId = @userId ORDER BY c.timestamp ASC"
         messages = []
         async for item in self.container_client.query_items(
             query=query, parameters=parameters

pyproject.toml

@@ -31,8 +31,12 @@ pytest = "7.4.0"
 pytest-asyncio = "0.23.2"
 chardet = "5.2.0"
 pre-commit = "^3.7.0"
+ruff = "^0.3.7"
 
 
 [build-system]
 requires = ["poetry-core"]
 build-backend = "poetry.core.masonry.api"
+
+[tool.ruff]
+include = ["app.py", "backend/*.py", "backend/**/*.py"]