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

Fix pinecone for async memory (#2975)

Browse Source
This commit is contained in:
Dev Khant
2025-06-18 01:37:45 +05:30
committed by GitHub
parent c700d790db
commit 9eb4e77c75
2 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
mem0/memory/main.py
View File

@@ -1415,7 +1415,9 @@ class AsyncMemory(MemoryBase):
async def _get_all_from_vector_store(self, filters, limit): async def _get_all_from_vector_store(self, filters, limit):
memories_result = await asyncio.to_thread(self.vector_store.list, filters=filters, limit=limit) memories_result = await asyncio.to_thread(self.vector_store.list, filters=filters, limit=limit)
actual_memories = ( actual_memories = (
memories_result[0] if isinstance(memories_result, tuple) and len(memories_result) > 0 else memories_result memories_result[0]
if isinstance(memories_result, (tuple, list)) and len(memories_result) > 0
else memories_result
) )
promoted_payload_keys = [ promoted_payload_keys = [

3
mem0/vector_stores/pinecone.py
View File

@@ -5,8 +5,7 @@ from typing import Any, Dict, List, Optional, Union
from pydantic import BaseModel from pydantic import BaseModel
try: try:
from pinecone import Pinecone, PodSpec, ServerlessSpec from pinecone import Pinecone, PodSpec, ServerlessSpec, Vector
from pinecone.data.dataclasses.vector import Vector
except ImportError: except ImportError:
raise ImportError( raise ImportError(
"Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`" "Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`"