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`"