Update main.py
This commit is contained in:
+12
-4
@@ -854,7 +854,7 @@ class Memory(MemoryBase):
|
|||||||
agent_id: Optional[str] = None,
|
agent_id: Optional[str] = None,
|
||||||
run_id: Optional[str] = None,
|
run_id: Optional[str] = None,
|
||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
limit: int = 100,
|
top_k: int = 100,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
List all memories.
|
List all memories.
|
||||||
@@ -873,6 +873,8 @@ class Memory(MemoryBase):
|
|||||||
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", ...}]}`
|
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", ...}]}`
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
limit = top_k
|
||||||
|
|
||||||
_, effective_filters = _build_filters_and_metadata(
|
_, effective_filters = _build_filters_and_metadata(
|
||||||
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
||||||
)
|
)
|
||||||
@@ -943,7 +945,7 @@ class Memory(MemoryBase):
|
|||||||
user_id: Optional[str] = None,
|
user_id: Optional[str] = None,
|
||||||
agent_id: Optional[str] = None,
|
agent_id: Optional[str] = None,
|
||||||
run_id: Optional[str] = None,
|
run_id: Optional[str] = None,
|
||||||
limit: int = 100,
|
top_k: int = 100,
|
||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
threshold: float = 0.1,
|
threshold: float = 0.1,
|
||||||
rerank: bool = False,
|
rerank: bool = False,
|
||||||
@@ -979,6 +981,8 @@ class Memory(MemoryBase):
|
|||||||
dict: A dictionary containing the search results under a "results" key.
|
dict: A dictionary containing the search results under a "results" key.
|
||||||
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", "score": 0.8, ...}]}`
|
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", "score": 0.8, ...}]}`
|
||||||
"""
|
"""
|
||||||
|
limit = top_k
|
||||||
|
|
||||||
_, effective_filters = _build_filters_and_metadata(
|
_, effective_filters = _build_filters_and_metadata(
|
||||||
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
||||||
)
|
)
|
||||||
@@ -2093,7 +2097,7 @@ class AsyncMemory(MemoryBase):
|
|||||||
agent_id: Optional[str] = None,
|
agent_id: Optional[str] = None,
|
||||||
run_id: Optional[str] = None,
|
run_id: Optional[str] = None,
|
||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
limit: int = 100,
|
top_k: int = 100,
|
||||||
):
|
):
|
||||||
"""
|
"""
|
||||||
List all memories.
|
List all memories.
|
||||||
@@ -2112,6 +2116,8 @@ class AsyncMemory(MemoryBase):
|
|||||||
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", ...}]}`
|
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", ...}]}`
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
limit = top_k
|
||||||
|
|
||||||
_, effective_filters = _build_filters_and_metadata(
|
_, effective_filters = _build_filters_and_metadata(
|
||||||
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
||||||
)
|
)
|
||||||
@@ -2185,7 +2191,7 @@ class AsyncMemory(MemoryBase):
|
|||||||
user_id: Optional[str] = None,
|
user_id: Optional[str] = None,
|
||||||
agent_id: Optional[str] = None,
|
agent_id: Optional[str] = None,
|
||||||
run_id: Optional[str] = None,
|
run_id: Optional[str] = None,
|
||||||
limit: int = 100,
|
top_k: int = 100,
|
||||||
filters: Optional[Dict[str, Any]] = None,
|
filters: Optional[Dict[str, Any]] = None,
|
||||||
threshold: float = 0.1,
|
threshold: float = 0.1,
|
||||||
metadata_filters: Optional[Dict[str, Any]] = None,
|
metadata_filters: Optional[Dict[str, Any]] = None,
|
||||||
@@ -2223,6 +2229,8 @@ class AsyncMemory(MemoryBase):
|
|||||||
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", "score": 0.8, ...}]}`
|
Example for v1.1+: `{"results": [{"id": "...", "memory": "...", "score": 0.8, ...}]}`
|
||||||
"""
|
"""
|
||||||
|
|
||||||
|
limit = top_k
|
||||||
|
|
||||||
_, effective_filters = _build_filters_and_metadata(
|
_, effective_filters = _build_filters_and_metadata(
|
||||||
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
user_id=user_id, agent_id=agent_id, run_id=run_id, input_filters=filters
|
||||||
)
|
)
|
||||||
|
|||||||
Reference in New Issue
Block a user