fix: add default param to pass pydantic error

This commit is contained in:
Young Han 2025-10-31 11:30:22 -07:00
parent d8849f0b95
commit 8f0f4a26a3
2 changed files with 9 additions and 2 deletions

View file

@ -23,8 +23,8 @@ class MongoDBVectorIOConfig(BaseModel):
""" """
# MongoDB Atlas connection details # MongoDB Atlas connection details
connection_string: str = Field( connection_string: str | None = Field(
default="", default=None,
description="MongoDB Atlas connection string (e.g., mongodb+srv://user:pass@cluster.mongodb.net/)", description="MongoDB Atlas connection string (e.g., mongodb+srv://user:pass@cluster.mongodb.net/)",
) )
database_name: str = Field(default="llama_stack", description="Database name to use for vector collections") database_name: str = Field(default="llama_stack", description="Database name to use for vector collections")

View file

@ -419,6 +419,13 @@ class MongoDBVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocol
# Initialize KV store for metadata # Initialize KV store for metadata
self.kvstore = await kvstore_impl(self.config.kvstore) self.kvstore = await kvstore_impl(self.config.kvstore)
# Validate connection string
if not self.config.connection_string:
raise ValueError(
"MongoDB connection_string is required but not provided. "
"Please set MONGODB_CONNECTION_STRING environment variable or provide it in config."
)
# Connect to MongoDB with optimized settings for RAG # Connect to MongoDB with optimized settings for RAG
self.client = MongoClient( self.client = MongoClient(
self.config.connection_string, self.config.connection_string,