Skip to content

Commit 9eb4e77

Browse files
authored
Fix pinecone for async memory (#2975)
1 parent c700d79 commit 9eb4e77

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

mem0/memory/main.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1415,7 +1415,9 @@ async def get_all(
14151415
async def _get_all_from_vector_store(self, filters, limit):
14161416
memories_result = await asyncio.to_thread(self.vector_store.list, filters=filters, limit=limit)
14171417
actual_memories = (
1418-
memories_result[0] if isinstance(memories_result, tuple) and len(memories_result) > 0 else memories_result
1418+
memories_result[0]
1419+
if isinstance(memories_result, (tuple, list)) and len(memories_result) > 0
1420+
else memories_result
14191421
)
14201422

14211423
promoted_payload_keys = [

mem0/vector_stores/pinecone.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,7 @@
55
from pydantic import BaseModel
66

77
try:
8-
from pinecone import Pinecone, PodSpec, ServerlessSpec
9-
from pinecone.data.dataclasses.vector import Vector
8+
from pinecone import Pinecone, PodSpec, ServerlessSpec, Vector
109
except ImportError:
1110
raise ImportError(
1211
"Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`"

0 commit comments

Comments
 (0)