phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-03 09:53:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: resolve unit test failures in starter distribution with MongoDB provider

Browse source
This commit is contained in:
Young Han 2025-10-10 11:30:08 -07:00
parent 885631d071
commit 572062de9d
5 changed files with 36 additions and 36 deletions
Download patch file Download diff file Expand all files Collapse all files

16
llama_stack/providers/remote/vector_io/mongodb/config.py
View file

@ -40,25 +40,25 @@ class MongoDBVectorIOConfig(BaseModel):
max_pool_size: int = Field(default=100, description="Maximum connection pool size")
timeout_ms: int = Field(default=30000, description="Connection timeout in milliseconds")
# KV store for metadata
# KV store configuration
kvstore: KVStoreConfig = Field(description="Config for KV store backend for metadata storage")
@classmethod
def sample_run_config(
cls,
__distro_dir__: str,
connection_string: str = "${env.MONGODB_CONNECTION_STRING}",
database_name: str = "llama_stack",
connection_string: str = "${env.MONGODB_CONNECTION_STRING:=}",
database_name: str = "${env.MONGODB_DATABASE_NAME:=llama_stack}",
**kwargs: Any,
) -> dict[str, Any]:
return {
"connection_string": connection_string,
"database_name": database_name,
"index_name": "vector_index",
"path_field": "embedding",
"similarity_metric": "cosine",
"max_pool_size": 100,
"timeout_ms": 30000,
"index_name": "${env.MONGODB_INDEX_NAME:=vector_index}",
"path_field": "${env.MONGODB_PATH_FIELD:=embedding}",
"similarity_metric": "${env.MONGODB_SIMILARITY_METRIC:=cosine}",
"max_pool_size": "${env.MONGODB_MAX_POOL_SIZE:=100}",
"timeout_ms": "${env.MONGODB_TIMEOUT_MS:=30000}",
"kvstore": SqliteKVStoreConfig.sample_run_config(
__distro_dir__=__distro_dir__,
db_name="mongodb_registry.db",