scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

Fix black

Browse source
This commit is contained in:
Sameer Kankute 2026-04-23 18:32:24 +05:30
parent e3440baa0c
commit 2e3a4bb27a
No known key found for this signature in database
2 changed files with 12 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
litellm/llms/dashscope/image_generation/__init__.py
View file

@ -1,4 +1,6 @@
from litellm.llms.base_llm.image_generation.transformation import BaseImageGenerationConfig
from litellm.llms.base_llm.image_generation.transformation import (
BaseImageGenerationConfig,
)
from .transformation import DashScopeImageGenerationConfig

17
litellm/llms/dashscope/image_generation/transformation.py
View file

@ -27,9 +27,14 @@ from typing import TYPE_CHECKING, Any, List, Optional
import httpx
from litellm.llms.base_llm.image_generation.transformation import BaseImageGenerationConfig
from litellm.llms.base_llm.image_generation.transformation import (
BaseImageGenerationConfig,
)
from litellm.secret_managers.main import get_secret_str
from litellm.types.llms.openai import AllMessageValues, OpenAIImageGenerationOptionalParams
from litellm.types.llms.openai import (
AllMessageValues,
OpenAIImageGenerationOptionalParams,
)
from litellm.types.utils import ImageObject, ImageResponse
if TYPE_CHECKING:
@ -93,9 +98,7 @@ class DashScopeImageGenerationConfig(BaseImageGenerationConfig):
stream: Optional[bool] = None,
) -> str:
return (
api_base
or get_secret_str("DASHSCOPE_API_BASE_IMAGE")
or DEFAULT_API_BASE
api_base or get_secret_str("DASHSCOPE_API_BASE_IMAGE") or DEFAULT_API_BASE
)
def validate_environment(
@ -176,9 +179,7 @@ class DashScopeImageGenerationConfig(BaseImageGenerationConfig):
choices = response_data.get("output", {}).get("choices", [])
for choice in choices:
content_list = (
choice.get("message", {}).get("content", [])
)
content_list = choice.get("message", {}).get("content", [])
for content_item in content_list:
image_url = content_item.get("image")
if image_url: