From 6a8721bd176304f859778b21de30f7433735b725 Mon Sep 17 00:00:00 2001
From: Manoj Aggarwal <manoj6543@gmail.com>
Date: Wed, 21 Jan 2026 09:55:40 -0800
Subject: [PATCH] ruff format

---
 .../app/agents/new_chat/tools/user_memory.py  | 20 +++++++++++++++----
 surfsense_backend/app/db.py                   |  4 +++-
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/surfsense_backend/app/agents/new_chat/tools/user_memory.py b/surfsense_backend/app/agents/new_chat/tools/user_memory.py
index 30ca4503e..23a0b8666 100644
--- a/surfsense_backend/app/agents/new_chat/tools/user_memory.py
+++ b/surfsense_backend/app/agents/new_chat/tools/user_memory.py
@@ -98,7 +98,9 @@ def format_memories_for_context(memories: list[dict[str, Any]]) -> str:
         category = memory.get("category", "unknown")
         text = memory.get("memory_text", "")
         updated = memory.get("updated_at", "")
-        parts.append(f"  <memory category='{category}' updated='{updated}'>{text}</memory>")
+        parts.append(
+            f"  <memory category='{category}' updated='{updated}'>{text}</memory>"
+        )
     parts.append("</user_memories>")
 
     return "\n".join(parts)
@@ -187,7 +189,7 @@ def create_save_memory_tool(
                 category=MemoryCategory(category),  # Convert string to enum
                 embedding=embedding,  # Pass embedding directly (list or numpy array)
             )
-            
+
             db_session.add(new_memory)
             await db_session.commit()
             await db_session.refresh(new_memory)
@@ -279,7 +281,12 @@ def create_recall_memory_tool(
                 )
 
                 # Add category filter if specified
-                if category and category in ["preference", "fact", "instruction", "context"]:
+                if category and category in [
+                    "preference",
+                    "fact",
+                    "instruction",
+                    "context",
+                ]:
                     stmt = stmt.where(UserMemory.category == MemoryCategory(category))
 
                 # Order by vector similarity
@@ -299,7 +306,12 @@ def create_recall_memory_tool(
                 )
 
                 # Add category filter if specified
-                if category and category in ["preference", "fact", "instruction", "context"]:
+                if category and category in [
+                    "preference",
+                    "fact",
+                    "instruction",
+                    "context",
+                ]:
                     stmt = stmt.where(UserMemory.category == MemoryCategory(category))
 
                 stmt = stmt.order_by(UserMemory.updated_at.desc()).limit(top_k)
diff --git a/surfsense_backend/app/db.py b/surfsense_backend/app/db.py
index 550f0a704..c56d19355 100644
--- a/surfsense_backend/app/db.py
+++ b/surfsense_backend/app/db.py
@@ -478,7 +478,9 @@ class MemoryCategory(str, Enum):
     # Using lowercase keys to match PostgreSQL enum values
     preference = "preference"  # User preferences (e.g., "prefers dark mode")
     fact = "fact"  # Facts about the user (e.g., "is a Python developer")
-    instruction = "instruction"  # Standing instructions (e.g., "always respond in bullet points")
+    instruction = (
+        "instruction"  # Standing instructions (e.g., "always respond in bullet points")
+    )
     context = "context"  # Contextual information (e.g., "working on project X")