mirror of
https://github.com/BerriAI/litellm.git
synced 2025-04-24 18:24:20 +00:00
fix InfinityEmbeddingConfig
This commit is contained in:
parent
8c385a94aa
commit
cf09a6770a
1 changed files with 16 additions and 8 deletions
|
@ -20,7 +20,15 @@ class InfinityEmbeddingConfig(BaseEmbeddingConfig):
|
|||
def __init__(self) -> None:
|
||||
pass
|
||||
|
||||
def get_complete_url(self, api_base: str | None, api_key: str | None, model: str, optional_params: dict, litellm_params: dict, stream: bool | None = None) -> str:
|
||||
def get_complete_url(
|
||||
self,
|
||||
api_base: Optional[str],
|
||||
api_key: Optional[str],
|
||||
model: str,
|
||||
optional_params: dict,
|
||||
litellm_params: dict,
|
||||
stream: Optional[bool] = None,
|
||||
) -> str:
|
||||
if api_base is None:
|
||||
raise ValueError("api_base is required for Infinity embeddings")
|
||||
# Remove trailing slashes and ensure clean base URL
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue