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

feat: add models_api for consistency

Browse source
This commit is contained in:
Young Han 2025-10-31 10:55:38 -07:00
parent 8a6794a605
commit 8acc4ef7f8
2 changed files with 5 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/providers/remote/vector_io/mongodb/__init__.py
View file

@ -14,6 +14,7 @@ async def get_adapter_impl(config: MongoDBVectorIOConfig, deps: dict[Api, Provid
# Handle the deps resolution - if files API exists, pass it, otherwise None # Handle the deps resolution - if files API exists, pass it, otherwise None
files_api = deps.get(Api.files) files_api = deps.get(Api.files)
impl = MongoDBVectorIOAdapter(config, deps[Api.inference], files_api) models_api = deps.get(Api.models)
impl = MongoDBVectorIOAdapter(config, deps[Api.inference], files_api, models_api)
await impl.initialize() await impl.initialize()
return impl return impl

8
llama_stack/providers/remote/vector_io/mongodb/mongodb.py
View file

@ -18,11 +18,7 @@ from pymongo.server_api import ServerApi
from llama_stack.apis.common.errors import VectorStoreNotFoundError from llama_stack.apis.common.errors import VectorStoreNotFoundError
from llama_stack.apis.inference import InterleavedContent from llama_stack.apis.inference import InterleavedContent
from llama_stack.apis.vector_dbs import VectorDB from llama_stack.apis.vector_dbs import VectorDB
from llama_stack.apis.vector_io import ( from llama_stack.apis.vector_io import Chunk, QueryChunksResponse, VectorIO
Chunk,
QueryChunksResponse,
VectorIO,
)
from llama_stack.log import get_logger from llama_stack.log import get_logger
from llama_stack.providers.datatypes import ( from llama_stack.providers.datatypes import (
HealthResponse, HealthResponse,
@ -402,12 +398,14 @@ class MongoDBVectorIOAdapter(OpenAIVectorStoreMixin, VectorIO, VectorDBsProtocol
config: MongoDBVectorIOConfig, config: MongoDBVectorIOConfig,
inference_api, inference_api,
files_api=None, files_api=None,
models_api=None,
) -> None: ) -> None:
# Handle the case where files_api might be a ProviderSpec that needs resolution # Handle the case where files_api might be a ProviderSpec that needs resolution
resolved_files_api = files_api resolved_files_api = files_api
super().__init__(files_api=resolved_files_api, kvstore=None) super().__init__(files_api=resolved_files_api, kvstore=None)
self.config = config self.config = config
self.inference_api = inference_api self.inference_api = inference_api
self.models_api = models_api
self.client: MongoClient | None = None self.client: MongoClient | None = None
self.database: Database | None = None self.database: Database | None = None
self.cache: dict[str, VectorDBWithIndex] = {} self.cache: dict[str, VectorDBWithIndex] = {}