add new tg ai llm check

This commit is contained in:
ishaan-jaff 2023-09-15 09:29:46 -07:00
parent 96dbbf7422
commit b2c4c3576c

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