HOTFIX: rename all ssambanova to sambanova

This commit is contained in:
seyeong-han 2024-11-29 17:26:58 -06:00
parent a82eb97bdf
commit c69dc293ae
11 changed files with 51 additions and 51 deletions

View file

@ -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

View file

@ -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",
)

View file

@ -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,