fix - use csv list for batch completions

This commit is contained in:
Ishaan Jaff 2024-05-11 14:24:48 -07:00
parent 62276fc221
commit d4288b134b
2 changed files with 4 additions and 6 deletions

View file

@ -3673,8 +3673,9 @@ async def chat_completion(
# skip router if user passed their key
if "api_key" in data:
tasks.append(litellm.acompletion(**data))
elif isinstance(data["model"], list) and llm_router is not None:
_models = data.pop("model")
elif "," in data["model"] and llm_router is not None:
_models_csv_string = data.pop("model")
_models = _models_csv_string.split(",")
tasks.append(llm_router.abatch_completion(models=_models, **data))
elif "user_config" in data:
# initialize a new router instance. make request using this Router

View file

@ -424,10 +424,7 @@ async def test_batch_chat_completions():
response = await chat_completion(
session=session,
key="sk-1234",
model=[
"gpt-3.5-turbo",
"fake-openai-endpoint",
],
model="gpt-3.5-turbo,fake-openai-endpoint",
)
print(f"response: {response}")