mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-06 08:16:43 +00:00
fix ai/ml api
This commit is contained in:
parent
b09c64e4f3
commit
65f5cc29dd
1 changed files with 3 additions and 0 deletions
|
|
@ -97,6 +97,9 @@ class AimlImageGenerationConfig(BaseImageGenerationConfig):
|
|||
)
|
||||
|
||||
complete_url = complete_url.rstrip("/")
|
||||
# Strip /v1 suffix if present since IMAGE_GENERATION_ENDPOINT already includes v1
|
||||
if complete_url.endswith("/v1"):
|
||||
complete_url = complete_url[:-3]
|
||||
complete_url = f"{complete_url}/{self.IMAGE_GENERATION_ENDPOINT}"
|
||||
return complete_url
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue