fix(utils.py): route together ai calls to openai client

together ai is now openai-compatible

n
This commit is contained in:
Krrish Dholakia 2024-02-03 19:22:48 -08:00
parent c2f674ebe0
commit c49c88c8e5
6 changed files with 23 additions and 4 deletions

View file

@ -440,8 +440,8 @@ class OpenAIChatCompletion(BaseLLM):
input=data["messages"],
api_key=api_key,
additional_args={
"headers": headers,
"api_base": api_base,
"headers": {"Authorization": f"Bearer {openai_client.api_key}"},
"api_base": openai_client._base_url._uri_reference,
"acompletion": False,
"complete_input_dict": data,
},