phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

add ImageObject

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-20 10:45:37 -07:00
parent a4f906b464
commit 2519879e67
5 changed files with 111 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -3874,7 +3874,7 @@ def image_generation(
or optional_params.pop("vertex_ai_credentials", None)
or get_secret("VERTEXAI_CREDENTIALS")
)
model_response = vertex_chat_completion.aimage_generation( # type: ignore
model_response = vertex_chat_completion.image_generation(
model=model,
prompt=prompt,
timeout=timeout,
@ -3883,6 +3883,7 @@ def image_generation(
model_response=model_response,
vertex_project=vertex_ai_project,
vertex_location=vertex_ai_location,
aimg_generation=aimg_generation,
)
return model_response