mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-18 14:39:48 +00:00
HOTFIX: rename all ssambanova to sambanova
This commit is contained in:
parent
a82eb97bdf
commit
c69dc293ae
11 changed files with 51 additions and 51 deletions
|
|
@ -4,14 +4,14 @@
|
|||
# This source code is licensed under the terms described in the LICENSE file in
|
||||
# the root directory of this source tree.
|
||||
|
||||
from .config import SsambanovaImplConfig
|
||||
from .ssambanova import SsambanovaInferenceAdapter
|
||||
from .config import SambanovaImplConfig
|
||||
from .sambanova import SambanovaInferenceAdapter
|
||||
|
||||
|
||||
async def get_adapter_impl(config: SsambanovaImplConfig, _deps):
|
||||
async def get_adapter_impl(config: SambanovaImplConfig, _deps):
|
||||
assert isinstance(
|
||||
config, SsambanovaImplConfig
|
||||
config, SambanovaImplConfig
|
||||
), f"Unexpected config type: {type(config)}"
|
||||
impl = SsambanovaInferenceAdapter(config)
|
||||
impl = SambanovaInferenceAdapter(config)
|
||||
await impl.initialize()
|
||||
return impl
|
||||
|
|
@ -10,12 +10,12 @@ from pydantic import BaseModel, Field
|
|||
|
||||
|
||||
@json_schema_type
|
||||
class SsambanovaImplConfig(BaseModel):
|
||||
class SambanovaImplConfig(BaseModel):
|
||||
url: str = Field(
|
||||
default="https://api.sambanova.ai/v1",
|
||||
description="The URL for the Ssambanova model serving endpoint",
|
||||
description="The URL for the Sambanova model serving endpoint",
|
||||
)
|
||||
api_token: str = Field(
|
||||
default=None,
|
||||
description="The Ssambanova API token",
|
||||
description="The Sambanova API token",
|
||||
)
|
||||
|
|
@ -30,7 +30,7 @@ from llama_stack.providers.utils.inference.prompt_adapter import (
|
|||
chat_completion_request_to_prompt,
|
||||
)
|
||||
|
||||
from .config import SsambanovaImplConfig
|
||||
from .config import SambanovaImplConfig
|
||||
|
||||
|
||||
model_aliases = [
|
||||
|
|
@ -57,8 +57,8 @@ model_aliases = [
|
|||
]
|
||||
|
||||
|
||||
class SsambanovaInferenceAdapter(ModelRegistryHelper, Inference):
|
||||
def __init__(self, config: SsambanovaImplConfig) -> None:
|
||||
class SambanovaInferenceAdapter(ModelRegistryHelper, Inference):
|
||||
def __init__(self, config: SambanovaImplConfig) -> None:
|
||||
ModelRegistryHelper.__init__(
|
||||
self,
|
||||
model_aliases=model_aliases,
|
||||
Loading…
Add table
Add a link
Reference in a new issue