apunkt/SurfSense
1
0
Fork 0
mirror of https://github.com/MODSetter/SurfSense.git synced 2026-05-03 21:02:40 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

error

Browse source
This commit is contained in:
Manoj Aggarwal 2026-01-20 15:34:01 -08:00
parent 48fb38bafc
commit 92aa3f4eab
3 changed files with 72 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

11
surfsense_backend/app/db.py
View file

@ -475,10 +475,11 @@ class ChatCommentMention(BaseModel, TimestampMixin):
class MemoryCategory(str, Enum):
"""Categories for user memories."""
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")
CONTEXT = "context" # Contextual information (e.g., "working on project X")
# 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")
context = "context" # Contextual information (e.g., "working on project X")
class UserMemory(BaseModel, TimestampMixin):
@ -510,7 +511,7 @@ class UserMemory(BaseModel, TimestampMixin):
category = Column(
SQLAlchemyEnum(MemoryCategory),
nullable=False,
default=MemoryCategory.FACT,
default=MemoryCategory.fact,
)
# Vector embedding for semantic search
embedding = Column(Vector(config.embedding_model_instance.dimension))