Przeglądaj źródła

Fix linting: ruff import sorting and black formatting

0xReLogic 2 dni temu
rodzic
commit
28e91a13cc
4 zmienionych plików z 10 dodań i 13 usunięć
  1. 5 10
      src/database.py
  2. 3 1
      src/embeddings.py
  3. 1 1
      src/engram.py
  4. 1 1
      src/memory.py

+ 5 - 10
src/database.py

@@ -151,9 +151,7 @@ class Database:
                 )
                 )
                 """
                 """
             )
             )
-            cursor.execute(
-                "CREATE INDEX IF NOT EXISTS idx_engram_bucket ON engram_index(bucket)"
-            )
+            cursor.execute("CREATE INDEX IF NOT EXISTS idx_engram_bucket ON engram_index(bucket)")
             cursor.execute(
             cursor.execute(
                 "CREATE INDEX IF NOT EXISTS idx_engram_memory ON engram_index(memory_id)"
                 "CREATE INDEX IF NOT EXISTS idx_engram_memory ON engram_index(memory_id)"
             )
             )
@@ -262,9 +260,7 @@ class Database:
             "SELECT e.memory_id AS id, SUM(e.hits) AS hits "
             "SELECT e.memory_id AS id, SUM(e.hits) AS hits "
             "FROM engram_index e "
             "FROM engram_index e "
             "JOIN memories m ON m.id = e.memory_id "
             "JOIN memories m ON m.id = e.memory_id "
-            "WHERE m.archived = 0 AND e.bucket IN ("
-            + placeholders
-            + ")"
+            "WHERE m.archived = 0 AND e.bucket IN (" + placeholders + ")"
         )
         )
         params: list[Any] = [*buckets]
         params: list[Any] = [*buckets]
         if project:
         if project:
@@ -318,9 +314,7 @@ class Database:
         if self.conn is None:
         if self.conn is None:
             raise RuntimeError(_DB_NOT_CONNECTED_ERROR)
             raise RuntimeError(_DB_NOT_CONNECTED_ERROR)
         if include_none:
         if include_none:
-            cursor = self.execute(
-                "SELECT DISTINCT project FROM memories WHERE archived = 0"
-            )
+            cursor = self.execute("SELECT DISTINCT project FROM memories WHERE archived = 0")
         else:
         else:
             cursor = self.execute(
             cursor = self.execute(
                 "SELECT DISTINCT project FROM memories WHERE archived = 0 AND project IS NOT NULL"
                 "SELECT DISTINCT project FROM memories WHERE archived = 0 AND project IS NOT NULL"
@@ -476,11 +470,12 @@ class Database:
             )
             )
             return False
             return False
 
 
-        import numpy as np
         import os
         import os
         import pickle
         import pickle
         import tempfile
         import tempfile
 
 
+        import numpy as np
+
         index_path = self._leann_index_path(project)
         index_path = self._leann_index_path(project)
         index_path.parent.mkdir(parents=True, exist_ok=True)
         index_path.parent.mkdir(parents=True, exist_ok=True)
 
 

+ 3 - 1
src/embeddings.py

@@ -73,7 +73,9 @@ class EmbeddingService:
         """Load the sentence-transformer model."""
         """Load the sentence-transformer model."""
         if self.model is None:
         if self.model is None:
             logger.info(f"Loading embedding model: {self.model_name}")
             logger.info(f"Loading embedding model: {self.model_name}")
-            self.model = SentenceTransformer(self.model_name, device=self.device, trust_remote_code=True)
+            self.model = SentenceTransformer(
+                self.model_name, device=self.device, trust_remote_code=True
+            )
             self.embedding_dim = self.model.get_sentence_embedding_dimension()
             self.embedding_dim = self.model.get_sentence_embedding_dimension()
             logger.info(f"Model loaded. Embedding dimension: {self.embedding_dim}")
             logger.info(f"Model loaded. Embedding dimension: {self.embedding_dim}")
 
 

+ 1 - 1
src/engram.py

@@ -7,7 +7,7 @@ import logging
 import re
 import re
 import unicodedata
 import unicodedata
 from collections import Counter
 from collections import Counter
-from typing import Iterable
+from collections.abc import Iterable
 
 
 from .config import settings
 from .config import settings
 
 

+ 1 - 1
src/memory.py

@@ -439,7 +439,7 @@ class MemoryService:
             # Additional safety check for empty arrays
             # Additional safety check for empty arrays
             if not selected:
             if not selected:
                 return []
                 return []
-                
+
             emb_dim = embedding_service.embedding_dim
             emb_dim = embedding_service.embedding_dim
             cand_mems = [
             cand_mems = [
                 m
                 m