Skip to content

Commit 8a06b27

Browse files
authored
Merge pull request #100 from fcenedes/tools_customisation
feat(client): Allow customizing memory tool descriptions with ToolSchema wrapper classes
2 parents 454e94b + 723f722 commit 8a06b27

File tree

5 files changed

+1016
-321
lines changed

5 files changed

+1016
-321
lines changed

agent-memory-client/agent_memory_client/__init__.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,12 +18,16 @@
1818
# Re-export essential models for convenience
1919
ModelNameLiteral,
2020
)
21+
from .tool_schema import ToolSchema, ToolSchemaCollection
2122

2223
__all__ = [
2324
# Client classes
2425
"MemoryAPIClient",
2526
"MemoryClientConfig",
2627
"create_memory_client",
28+
# Tool schema classes
29+
"ToolSchema",
30+
"ToolSchemaCollection",
2731
# Exceptions
2832
"MemoryClientError",
2933
"MemoryValidationError",

0 commit comments

Comments
 (0)