scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-11 22:51:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 124

fix - use router level client on proxy img gen

Browse source
This commit is contained in:
Ishaan Jaff 2024-06-19 15:27:46 -07:00
parent 8c07f3edf4
commit 931a72523e
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/main.py
View file

@ -3944,6 +3944,7 @@ def image_generation(
proxy_server_request = kwargs.get("proxy_server_request", None)
model_info = kwargs.get("model_info", None)
metadata = kwargs.get("metadata", {})
client = kwargs.get("client", None)
model_response = litellm.utils.ImageResponse()
if model is not None or custom_llm_provider is not None:
@ -4082,6 +4083,7 @@ def image_generation(
model_response=model_response,
api_version=api_version,
aimg_generation=aimg_generation,
client=client,
)
elif custom_llm_provider == "openai":
model_response = openai_chat_completions.image_generation(
@ -4094,6 +4096,7 @@ def image_generation(
optional_params=optional_params,
model_response=model_response,
aimg_generation=aimg_generation,
client=client,
)
elif custom_llm_provider == "bedrock":
if model is None: