klas/t6_mem0
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Proper error msg for API Key validation (#2220)

Browse Source
This commit is contained in:
Dev Khant
2025-02-18 23:59:59 +05:30
committed by GitHub
parent b052a86424
commit cbee71a63e
1 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
mem0/client/main.py
View File

@@ -96,16 +96,21 @@ class MemoryClient:
try: try:
params = self._prepare_params() params = self._prepare_params()
response = self.client.get("/v1/ping/", params=params) response = self.client.get("/v1/ping/", params=params)
data = response.json()
response.raise_for_status() response.raise_for_status()
if response.status_code == 200: if data.get("org_id") and data.get("project_id"):
data = response.json() self.org_id = data.get("org_id")
if data.get("org_id") and data.get("project_id"): self.project_id = data.get("project_id")
self.org_id = data.get("org_id")
self.project_id = data.get("project_id")
except httpx.HTTPStatusError: except httpx.HTTPStatusError as e:
raise ValueError("Invalid API Key. Please get a valid API Key from https://app.mem0.ai") try:
error_data = e.response.json()
error_message = error_data.get("detail", str(e))
except:
error_message = str(e)
raise ValueError(f"Error: {error_message}")
@api_error_handler @api_error_handler
def add(self, messages: Union[str, List[Dict[str, str]]], **kwargs) -> Dict[str, Any]: def add(self, messages: Union[str, List[Dict[str, str]]], **kwargs) -> Dict[str, Any]: