forked from phoenix-oss/llama-stack-mirror
# What does this PR do? It's a dict that may contain different types, as per resolver:instantiate_provider implementation. (AFAIU it also never contains ProviderSpecs, but *instances* of provider implementations.) [//]: # (If resolving an issue, uncomment and update the line below) [//]: # (Closes #[issue-number]) ## Test Plan mypy passing if enabled checks for these modules. (See #1543) [//]: # (## Documentation) Signed-off-by: Ihar Hrachyshka <ihar.hrachyshka@gmail.com>
17 lines
458 B
Python
17 lines
458 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
from typing import Any, Dict
|
|
|
|
from .config import VLLMConfig
|
|
|
|
|
|
async def get_provider_impl(config: VLLMConfig, _deps: Dict[str, Any]):
|
|
from .vllm import VLLMInferenceImpl
|
|
|
|
impl = VLLMInferenceImpl(config)
|
|
await impl.initialize()
|
|
return impl
|