Skip to content

Commit

Permalink
increase timeout and version bump (#2205)
Browse files Browse the repository at this point in the history
  • Loading branch information
Dev-Khant authored Feb 14, 2025
1 parent 0bb6137 commit cd31c58
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 5 deletions.
8 changes: 4 additions & 4 deletions mem0/client/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,7 +100,7 @@ def __init__(
self.client = httpx.Client(
base_url=self.host,
headers={"Authorization": f"Token {self.api_key}", "Mem0-User-ID": self.user_id},
timeout=60,
timeout=300,
)
self._validate_api_key()
capture_client_event("client.init", self)
Expand Down Expand Up @@ -169,7 +169,7 @@ def get_all(self, version: str = "v1", **kwargs) -> List[Dict[str, Any]]:
Args:
version: The API version to use for the search endpoint.
**kwargs: Optional parameters for filtering (user_id, agent_id, app_id, limit).
**kwargs: Optional parameters for filtering (user_id, agent_id, app_id, top_k).
Returns:
A list of dictionaries containing memories.
Expand Down Expand Up @@ -203,7 +203,7 @@ def search(self, query: str, version: str = "v1", **kwargs) -> List[Dict[str, An
Args:
query: The search query string.
version: The API version to use for the search endpoint.
**kwargs: Additional parameters such as user_id, agent_id, app_id, limit, filters.
**kwargs: Additional parameters such as user_id, agent_id, app_id, top_k, filters.
Returns:
A list of dictionaries containing search results.
Expand Down Expand Up @@ -612,7 +612,7 @@ def __init__(
self.async_client = httpx.AsyncClient(
base_url=self.sync_client.host,
headers=self.sync_client.client.headers,
timeout=60,
timeout=300,
)

async def __aenter__(self):
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
[tool.poetry]
name = "mem0ai"
version = "0.1.48"
version = "0.1.49"
description = "Long-term memory for AI Agents"
authors = ["Mem0 <[email protected]>"]
exclude = [
Expand Down

0 comments on commit cd31c58

Please sign in to comment.