From 9eb4e77c75ec6a1f5b3601bb2d3bd91a5a66baf8 Mon Sep 17 00:00:00 2001 From: Dev Khant Date: Wed, 18 Jun 2025 01:37:45 +0530 Subject: [PATCH] Fix pinecone for async memory (#2975) --- mem0/memory/main.py | 4 +++- mem0/vector_stores/pinecone.py | 3 +-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/mem0/memory/main.py b/mem0/memory/main.py index 56814895..a4e082ef 100644 --- a/mem0/memory/main.py +++ b/mem0/memory/main.py @@ -1415,7 +1415,9 @@ class AsyncMemory(MemoryBase): async def _get_all_from_vector_store(self, filters, limit): memories_result = await asyncio.to_thread(self.vector_store.list, filters=filters, limit=limit) 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 = [ diff --git a/mem0/vector_stores/pinecone.py b/mem0/vector_stores/pinecone.py index ff350bb4..44ff27ac 100644 --- a/mem0/vector_stores/pinecone.py +++ b/mem0/vector_stores/pinecone.py @@ -5,8 +5,7 @@ from typing import Any, Dict, List, Optional, Union from pydantic import BaseModel try: - from pinecone import Pinecone, PodSpec, ServerlessSpec - from pinecone.data.dataclasses.vector import Vector + from pinecone import Pinecone, PodSpec, ServerlessSpec, Vector except ImportError: raise ImportError( "Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`"