(fix) acompletion kwargs type hints

This commit is contained in:
ishaan-jaff 2024-01-11 14:22:37 +05:30
parent 40054f89b5
commit f89385eed8

View file

@ -202,6 +202,7 @@ async def acompletion(
- If `stream` is True, the function returns an async generator that yields completion lines.
"""
loop = asyncio.get_event_loop()
custom_llm_provider = None
# Adjusted to use explicit arguments instead of *args and **kwargs
completion_kwargs = {
"model": model,
@ -241,7 +242,7 @@ async def acompletion(
func_with_context = partial(ctx.run, func)
_, custom_llm_provider, _, _ = get_llm_provider(
model=model, api_base=completion_kwargs.get("base_url", None)
model=model, api_base=kwargs.get("api_base", None)
)
if (