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

Add logger in Opensearch (#2957)

Browse Source
This commit is contained in:
Dev Khant
2025-06-14 21:55:22 +05:30
committed by GitHub
parent a0cd4065d9
commit 3e5f68ee90
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mem0/vector_stores/opensearch.py
View File

@@ -83,19 +83,19 @@ class OpenSearchDB(VectorStoreBase):
}, },
} }
logger.warning(f"Creating index {name}, it might take 1-2 minutes...")
if not self.client.indices.exists(index=name): if not self.client.indices.exists(index=name):
self.client.indices.create(index=name, body=index_settings) self.client.indices.create(index=name, body=index_settings)
logger.info(f"Created index {name}")
# Wait for index to be ready # Wait for index to be ready
max_retries = 60 # 60 seconds timeout max_retries = 180 # 3 minutes timeout
retry_count = 0 retry_count = 0
while retry_count < max_retries: while retry_count < max_retries:
try: try:
# Check if index is ready by attempting a simple search # Check if index is ready by attempting a simple search
self.client.search(index=name, body={"query": {"match_all": {}}}) self.client.search(index=name, body={"query": {"match_all": {}}})
logger.info(f"Index {name} is ready")
time.sleep(1) time.sleep(1)
logger.info(f"Index {name} is ready")
return return
except Exception: except Exception:
retry_count += 1 retry_count += 1