Fixed s3 vectors memory initialization issue from configuration (#3481)

This commit is contained in:
Abdullah Irfan
2025-09-20 12:43:18 +05:30
committed by GitHub
parent ead210ffe4
commit 9df392b26a
4 changed files with 94 additions and 38 deletions
+3 -7
View File
@@ -5,17 +5,13 @@ from pydantic import BaseModel, ConfigDict, Field, model_validator
class S3VectorsConfig(BaseModel):
vector_bucket_name: str = Field(description="Name of the S3 Vector bucket")
index_name: str = Field("mem0", description="Name of the vector index")
embedding_model_dims: int = Field(
1536, description="Dimension of the embedding vector"
)
collection_name: str = Field("mem0", description="Name of the vector index")
embedding_model_dims: int = Field(1536, description="Dimension of the embedding vector")
distance_metric: str = Field(
"cosine",
description="Distance metric for similarity search. Options: 'cosine', 'euclidean'",
)
region_name: Optional[str] = Field(
None, description="AWS region for the S3 Vectors client"
)
region_name: Optional[str] = Field(None, description="AWS region for the S3 Vectors client")
@model_validator(mode="before")
@classmethod
+2 -2
View File
@@ -25,14 +25,14 @@ class S3Vectors(VectorStoreBase):
def __init__(
self,
vector_bucket_name: str,
index_name: str,
collection_name: str,
embedding_model_dims: int,
distance_metric: str = "cosine",
region_name: Optional[str] = None,
):
self.client = boto3.client("s3vectors", region_name=region_name)
self.vector_bucket_name = vector_bucket_name
self.collection_name = index_name
self.collection_name = collection_name
self.embedding_model_dims = embedding_model_dims
self.distance_metric = distance_metric