tools fix and formatting (#2441)
This commit is contained in:
@@ -8,7 +8,9 @@ try:
|
||||
from pinecone import Pinecone, PodSpec, ServerlessSpec
|
||||
from pinecone.data.dataclasses.vector import Vector
|
||||
except ImportError:
|
||||
raise ImportError("Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`") from None
|
||||
raise ImportError(
|
||||
"Pinecone requires extra dependencies. Install with `pip install pinecone pinecone-text`"
|
||||
) from None
|
||||
|
||||
from mem0.vector_stores.base import VectorStoreBase
|
||||
|
||||
@@ -34,7 +36,7 @@ class PineconeDB(VectorStoreBase):
|
||||
hybrid_search: bool,
|
||||
metric: str,
|
||||
batch_size: int,
|
||||
extra_params: Optional[Dict[str, Any]]
|
||||
extra_params: Optional[Dict[str, Any]],
|
||||
):
|
||||
"""
|
||||
Initialize the Pinecone vector store.
|
||||
@@ -199,7 +201,9 @@ class PineconeDB(VectorStoreBase):
|
||||
|
||||
return pinecone_filter
|
||||
|
||||
def search(self, query: str, vectors: List[float], limit: int = 5, filters: Optional[Dict] = None) -> List[OutputData]:
|
||||
def search(
|
||||
self, query: str, vectors: List[float], limit: int = 5, filters: Optional[Dict] = None
|
||||
) -> List[OutputData]:
|
||||
"""
|
||||
Search for similar vectors.
|
||||
|
||||
|
||||
Reference in New Issue
Block a user