diff --git a/litellm/main.py b/litellm/main.py index a7ed8f2a5b..f97659a320 100644 --- a/litellm/main.py +++ b/litellm/main.py @@ -99,17 +99,18 @@ class Chat(): def __init__(self, params): self.params = params self.completions = Completions(self.params) - + class Completions(): def __init__(self, params): self.params = params - def create(self, model, messages, **kwargs): + def create(self, messages, model=None, **kwargs): for k, v in kwargs.items(): self.params[k] = v + model = model or self.params.get('model') response = completion(model=model, messages=messages, **self.params) - return response + return response @client async def acompletion(*args, **kwargs):