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

feat(proxy_server.py): add new images/generation endpoint

Browse source
This commit is contained in:
Krrish Dholakia 2023-12-21 15:39:09 +05:30
parent 8101ad6801
commit 14115d0d60
4 changed files with 112 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

12
litellm/tests/test_router.py
View file

@ -455,12 +455,12 @@ async def test_aimg_gen_on_router():
}
]
router = Router(model_list=model_list)
# response = await router.aimage_generation(
# model="dall-e-3",
# prompt="A cute baby sea otter"
# )
# print(response)
# assert len(response.data) > 0
response = await router.aimage_generation(
model="dall-e-3",
prompt="A cute baby sea otter"
)
print(response)
assert len(response.data) > 0
response = await router.aimage_generation(
model="dall-e-2",