mirror of
https://github.com/pydantic/pydantic-ai-harness.git
synced 2026-07-21 02:45:34 +00:00
818808d6fe
* feat: add Memory capability * feat(memory): add SQLite and PostgreSQL stores * fix(memory): address CI pyright and CodeRabbit review * style(memory): ASCII dashes, labeled README fence (review follow-up) * fix(memory): SQLite in-memory rejection and case-sensitive listing * fix(memory): use qmark placeholders in SQLite prefix listing * Make memory safe for durable shared use * Harden memory trust and recovery boundaries --------- Co-authored-by: David SF <64162682+dsfaccini@users.noreply.github.com>
50 lines
1.2 KiB
Python
50 lines
1.2 KiB
Python
"""Memory capability: a persistent, injected notebook plus on-demand memory files."""
|
|
|
|
from pydantic_ai_harness.memory._capability import Memory
|
|
from pydantic_ai_harness.memory._postgres import PostgresConnection, PostgresMemoryStore, PostgresPool
|
|
from pydantic_ai_harness.memory._store import (
|
|
FileStore,
|
|
InMemoryStore,
|
|
MemoryConflictError,
|
|
MemoryFile,
|
|
MemoryMutation,
|
|
MemoryOperation,
|
|
MemoryOperationConflictError,
|
|
MemorySearchMatch,
|
|
MemorySearchResult,
|
|
MemoryStore,
|
|
SearchableMemoryStore,
|
|
SqliteMemoryStore,
|
|
)
|
|
from pydantic_ai_harness.memory._toolset import (
|
|
MemoryDeleteResult,
|
|
MemorySearchMatchResult,
|
|
MemorySearchResponse,
|
|
MemoryToolset,
|
|
MemoryWriteResult,
|
|
)
|
|
|
|
__all__ = [
|
|
'FileStore',
|
|
'InMemoryStore',
|
|
'Memory',
|
|
'MemoryConflictError',
|
|
'MemoryDeleteResult',
|
|
'MemoryFile',
|
|
'MemoryMutation',
|
|
'MemoryOperation',
|
|
'MemoryOperationConflictError',
|
|
'MemorySearchMatch',
|
|
'MemorySearchMatchResult',
|
|
'MemorySearchResponse',
|
|
'MemorySearchResult',
|
|
'MemoryStore',
|
|
'MemoryToolset',
|
|
'MemoryWriteResult',
|
|
'PostgresConnection',
|
|
'PostgresMemoryStore',
|
|
'PostgresPool',
|
|
'SearchableMemoryStore',
|
|
'SqliteMemoryStore',
|
|
]
|