Skip to content

Commit b09d617

Browse files
committed
refactor: migrate some classes to dataclasses
1 parent d3336dc commit b09d617

1 file changed

Lines changed: 5 additions & 2 deletions

File tree

memstate/schemas.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import uuid
2+
from dataclasses import dataclass
23
from datetime import datetime, timezone
34
from typing import Any
45

@@ -68,7 +69,8 @@ class TxEntry(BaseModel):
6869
reason: str | None = None
6970

7071

71-
class SearchResult(BaseModel):
72+
@dataclass(slots=True)
73+
class SearchResult:
7274
"""
7375
Represents a search result with related metadata.
7476
@@ -84,7 +86,8 @@ class SearchResult(BaseModel):
8486
score: float
8587

8688

87-
class ScoredFact(BaseModel):
89+
@dataclass(slots=True)
90+
class ScoredFact:
8891
"""
8992
Represents a scored fact model.
9093

0 commit comments

Comments
 (0)