add new tg ai llm check

This commit is contained in:
ishaan-jaff 2023-09-15 09:29:46 -07:00
parent 6f42b5559a
commit ee9e2af570

View file

@ -609,7 +609,7 @@ def completion(
)
return response
response = model_response
elif custom_llm_provider == "together_ai" or ("togethercomputer" in model):
elif custom_llm_provider == "together_ai" or ("togethercomputer" in model) or (model in litellm.together_ai_models):
custom_llm_provider = "together_ai"
together_ai_key = (
api_key