Fix indentation and linting issues
- Fixed tab/space inconsistencies in client.py and __init__.py - Renamed ambiguous variable 'l' to 'line' - All ruff checks now passing
This commit is contained in:
@@ -56,17 +56,7 @@ class LocalMem0Client:
|
||||
rerank: bool = False,
|
||||
top_k: int = 10,
|
||||
) -> List[Dict]:
|
||||
"""Search memories by semantic similarity.
|
||||
|
||||
Args:
|
||||
query: Search query string
|
||||
filters: Filters dict (e.g., {"user_id": "hermes-user"})
|
||||
rerank: Enable reranking for higher precision
|
||||
top_k: Maximum results to return
|
||||
|
||||
Returns:
|
||||
List of memory dicts with "text", "score", etc.
|
||||
"""
|
||||
"""Search memories by semantic similarity."""
|
||||
payload = {
|
||||
"query": query,
|
||||
"user_id": filters.get("user_id"),
|
||||
@@ -78,35 +68,19 @@ class LocalMem0Client:
|
||||
result = self._request("POST", "/search", json=payload)
|
||||
return self._unwrap_results(result)
|
||||
|
||||
def get_all(self, filters: Dict[str, Any]) -> List[Dict]:
|
||||
"""Get all memories matching filters.
|
||||
|
||||
Args:
|
||||
filters: Filters dict (e.g., {"user_id": "hermes-user"})
|
||||
|
||||
Returns:
|
||||
List of all matching memory dicts.
|
||||
"""
|
||||
def get_all(self, filters: Dict[str, Any]) -> List[Dict]:
|
||||
"""Get all memories matching filters."""
|
||||
params = filters
|
||||
result = self._request("GET", "/memories", params=params)
|
||||
return self._unwrap_results(result)
|
||||
|
||||
def add(
|
||||
def add(
|
||||
self,
|
||||
messages: List[Dict[str, str]],
|
||||
filters: Dict[str, Any],
|
||||
infer: bool = True,
|
||||
) -> Dict:
|
||||
"""Add conversation messages for fact extraction.
|
||||
|
||||
Args:
|
||||
messages: List of {"role": "user|assistant", "content": "..."}
|
||||
filters: Filters dict for scoping (user_id, agent_id)
|
||||
infer: Whether to extract facts via LLM (True) or store verbatim (False)
|
||||
|
||||
Returns:
|
||||
Response dict with added memory IDs.
|
||||
"""
|
||||
"""Add conversation messages for fact extraction."""
|
||||
payload = {
|
||||
"messages": messages,
|
||||
"user_id": filters.get("user_id"),
|
||||
@@ -118,13 +92,8 @@ class LocalMem0Client:
|
||||
|
||||
@staticmethod
|
||||
def _unwrap_results(response: Any) -> List[Dict]:
|
||||
"""Normalize Mem0 API response.
|
||||
|
||||
OSS server returns {"memories": [...]} or {"results": [...]}
|
||||
Cloud API returns {"results": [...]}
|
||||
"""
|
||||
"""Normalize Mem0 API response."""
|
||||
if isinstance(response, dict):
|
||||
# Try "memories" first (OSS server), then "results" (cloud/API v2)
|
||||
return response.get("memories", response.get("results", []))
|
||||
if isinstance(response, list):
|
||||
return response
|
||||
|
||||
Reference in New Issue
Block a user