admin/basicmachines-co-basic-memory
1
0
Fork 0
mirror of https://github.com/basicmachines-co/basic-memory synced 2026-06-21 13:47:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: replace deprecated json_encoders with Pydantic V2 field serializers (#295)

Browse Source 
Co-authored-by: claude[bot] <209825114+claude[bot]@users.noreply.github.com>
Co-authored-by: Paul Hernandez <phernandez@users.noreply.github.com>
This commit is contained in:
Paul Hernandez
2025-09-14 22:19:16 -05:00
committed by GitHub
parent 0499319ded
commit b00e4ff5a1
2 changed files with 48 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/basic_memory/schemas/memory.py
+17 -9
View File
@@ -4,7 +4,7 @@ from datetime import datetime
from typing import List, Optional, Annotated, Sequence, Literal, Union
from annotated_types import MinLen, MaxLen
from pydantic import BaseModel, Field, BeforeValidator, TypeAdapter, ConfigDict
from pydantic import BaseModel, Field, BeforeValidator, TypeAdapter, ConfigDict, field_serializer
from basic_memory.schemas.search import SearchItemType
@@ -118,8 +118,6 @@ def memory_url_path(url: memory_url) -> str: # pyright: ignore
class EntitySummary(BaseModel):
"""Simplified entity representation."""
model_config = ConfigDict(json_encoders={datetime: lambda dt: dt.isoformat()})
type: Literal["entity"] = "entity"
permalink: Optional[str]
title: str
@@ -127,12 +125,14 @@ class EntitySummary(BaseModel):
file_path: str
created_at: datetime
@field_serializer('created_at')
def serialize_created_at(self, dt: datetime) -> str:
return dt.isoformat()
class RelationSummary(BaseModel):
"""Simplified relation representation."""
model_config = ConfigDict(json_encoders={datetime: lambda dt: dt.isoformat()})
type: Literal["relation"] = "relation"
title: str
file_path: str
@@ -142,12 +142,14 @@ class RelationSummary(BaseModel):
to_entity: Optional[str] = None
created_at: datetime
@field_serializer('created_at')
def serialize_created_at(self, dt: datetime) -> str:
return dt.isoformat()
class ObservationSummary(BaseModel):
"""Simplified observation representation."""
model_config = ConfigDict(json_encoders={datetime: lambda dt: dt.isoformat()})
type: Literal["observation"] = "observation"
title: str
file_path: str
@@ -156,12 +158,14 @@ class ObservationSummary(BaseModel):
content: str
created_at: datetime
@field_serializer('created_at')
def serialize_created_at(self, dt: datetime) -> str:
return dt.isoformat()
class MemoryMetadata(BaseModel):
"""Simplified response metadata."""
model_config = ConfigDict(json_encoders={datetime: lambda dt: dt.isoformat()})
uri: Optional[str] = None
types: Optional[List[SearchItemType]] = None
depth: int
@@ -173,6 +177,10 @@ class MemoryMetadata(BaseModel):
total_relations: Optional[int] = None
total_observations: Optional[int] = None
@field_serializer('generated_at')
def serialize_generated_at(self, dt: datetime) -> str:
return dt.isoformat()
class ContextResult(BaseModel):
"""Context result containing a primary item with its observations and related items."""