fix ai/ml api

This commit is contained in:
Ishaan Jaffer 2025-11-26 18:55:32 -08:00
parent b09c64e4f3
commit 65f5cc29dd

View file

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