Files
leocrm/app/plugins/builtins/agent_memory/schemas.py
T
Agent Zero 000c969b13
Check Cross-Plugin Imports / check (push) Has been cancelled
Phase 5.5-5.9: Plugin-Marketplace, Agent Memory, GraphRAG, Subagents, External Agent API
5.5 Plugin-Marketplace:
- New plugin: marketplace/ (models, routes, services, schemas, config)
- MarketplaceListing model (global, no tenant_id)
- Ed25519 signature verification via PluginSignature
- Endpoints: list, detail, install, verify, categories
- Config: MARKETPLACE_SERVER_URL setting

5.6 Agent Memory (persistent):
- New plugin: agent_memory/ (models, routes, services, schemas)
- AgentMemory model with embedding vector(768) + HNSW index
- store_memory() with auto-embedding
- retrieve_relevant_memories() with pgvector cosine similarity
- Semantic search endpoint

5.7 GraphRAG:
- New plugin: graph_rag/ (models, routes, services, provider, schemas)
- EntityRelationship model (source/target type+id, relationship_type, metadata)
- BFS graph traversal (bidirectional, configurable depth)
- GraphRAGSearchProvider registered in unified_search

5.8 Subagents / Multi-Agent:
- AgentCoordinator class (create_subtask, wait_for_subtask, aggregate, cancel)
- AgentSubtask model + migration 0002_agent_subtasks.sql
- 6 new API endpoints for subtask management
- Tools registered in AI tool registry

5.9 External Agent API:
- external_api.py: POST /run, GET /status, POST /stream (SSE)
- Bearer API token authentication
- Rate limiting: 10 req/min per token
- ExternalAgentRequest/Response schemas

3 new plugins registered in main.py and __init__.py
All files py_compile clean
2026-08-04 15:06:23 +02:00

35 lines
935 B
Python

"""Pydantic schemas for the Agent Memory plugin."""
from __future__ import annotations
from datetime import datetime
from pydantic import BaseModel, Field
class AgentMemoryCreate(BaseModel):
agent_id: str = Field(..., description="UUID of the agent")
memory_type: str = Field("fact", max_length=50, description="Type of memory (fact, context, pattern, instruction)")
content: str = Field(..., min_length=1, description="Memory content text")
class AgentMemoryUpdate(BaseModel):
content: str | None = Field(None, min_length=1)
memory_type: str | None = Field(None, max_length=50)
class AgentMemoryRead(BaseModel):
id: str
agent_id: str
memory_type: str
content: str
owner_id: str | None = None
created_at: datetime | None = None
updated_at: datetime | None = None
score: float = 0.0
class AgentMemoryListResponse(BaseModel):
items: list[AgentMemoryRead]
total: int