Add memory_manager source files and tests
This commit is contained in:
@@ -1,24 +1,27 @@
|
|||||||
"""Pydantic schemas for API."""
|
"""Pydantic schemas for API request/response validation."""
|
||||||
|
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from typing import List, Optional
|
from typing import Any
|
||||||
|
|
||||||
from pydantic import BaseModel, Field
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
from memory_manager.db.models import MemoryCategory
|
||||||
|
|
||||||
|
|
||||||
class MemoryEntryCreate(BaseModel):
|
class MemoryEntryCreate(BaseModel):
|
||||||
title: str = Field(..., min_length=1, max_length=255)
|
title: str = Field(..., min_length=1, max_length=255)
|
||||||
content: str = Field(..., min_length=1)
|
content: str = Field(..., min_length=1)
|
||||||
category: str = Field(...)
|
category: MemoryCategory
|
||||||
tags: List[str] = Field(default_factory=list)
|
tags: list[str] = Field(default_factory=list)
|
||||||
agent_id: Optional[str] = Field(default="unknown")
|
agent_id: str | None = None
|
||||||
project_path: Optional[str] = Field(default=".")
|
project_path: str | None = None
|
||||||
parent_id: Optional[int] = None
|
|
||||||
|
|
||||||
|
|
||||||
class MemoryEntryUpdate(BaseModel):
|
class MemoryEntryUpdate(BaseModel):
|
||||||
title: Optional[str] = Field(None, min_length=1, max_length=255)
|
title: str | None = Field(None, min_length=1, max_length=255)
|
||||||
content: Optional[str] = Field(None, min_length=1)
|
content: str | None = Field(None, min_length=1)
|
||||||
category: Optional[str] = None
|
category: MemoryCategory | None = None
|
||||||
tags: Optional[List[str]] = None
|
tags: list[str] | None = None
|
||||||
|
|
||||||
|
|
||||||
class MemoryEntryResponse(BaseModel):
|
class MemoryEntryResponse(BaseModel):
|
||||||
@@ -26,21 +29,25 @@ class MemoryEntryResponse(BaseModel):
|
|||||||
title: str
|
title: str
|
||||||
content: str
|
content: str
|
||||||
category: str
|
category: str
|
||||||
tags: List[str]
|
tags: list[str]
|
||||||
agent_id: str
|
agent_id: str
|
||||||
project_path: str
|
project_path: str
|
||||||
created_at: datetime
|
created_at: datetime | None
|
||||||
updated_at: datetime
|
updated_at: datetime | None
|
||||||
parent_id: Optional[int] = None
|
|
||||||
|
|
||||||
class Config:
|
|
||||||
from_attributes = True
|
class SearchQuery(BaseModel):
|
||||||
|
q: str = Field(..., min_length=1)
|
||||||
|
category: MemoryCategory | None = None
|
||||||
|
agent_id: str | None = None
|
||||||
|
project_path: str | None = None
|
||||||
|
limit: int = Field(default=100, ge=1, le=1000)
|
||||||
|
|
||||||
|
|
||||||
class CommitCreate(BaseModel):
|
class CommitCreate(BaseModel):
|
||||||
message: str = Field(..., min_length=1)
|
message: str = Field(..., min_length=1)
|
||||||
agent_id: Optional[str] = Field(default="unknown")
|
agent_id: str | None = None
|
||||||
project_path: Optional[str] = Field(default=".")
|
project_path: str | None = None
|
||||||
|
|
||||||
|
|
||||||
class CommitResponse(BaseModel):
|
class CommitResponse(BaseModel):
|
||||||
@@ -49,21 +56,24 @@ class CommitResponse(BaseModel):
|
|||||||
message: str
|
message: str
|
||||||
agent_id: str
|
agent_id: str
|
||||||
project_path: str
|
project_path: str
|
||||||
created_at: datetime
|
snapshot: list[dict[str, Any]]
|
||||||
|
created_at: datetime | None
|
||||||
class Config:
|
|
||||||
from_attributes = True
|
|
||||||
|
|
||||||
|
|
||||||
class DiffResponse(BaseModel):
|
class DiffResponse(BaseModel):
|
||||||
hash1: str
|
commit1: dict[str, Any]
|
||||||
hash2: str
|
commit2: dict[str, Any]
|
||||||
added: List[dict]
|
added: list[dict[str, Any]]
|
||||||
removed: List[dict]
|
removed: list[dict[str, Any]]
|
||||||
modified: List[dict]
|
modified: list[dict[str, Any]]
|
||||||
|
|
||||||
|
|
||||||
|
class HealthResponse(BaseModel):
|
||||||
|
status: str
|
||||||
|
version: str
|
||||||
|
|
||||||
|
|
||||||
class StatsResponse(BaseModel):
|
class StatsResponse(BaseModel):
|
||||||
total_entries: int
|
total_entries: int
|
||||||
|
entries_by_category: dict[str, int]
|
||||||
total_commits: int
|
total_commits: int
|
||||||
category_counts: dict
|
|
||||||
|
|||||||
Reference in New Issue
Block a user