Skip to content

Commit e883e13

Browse files
authored
add Model.GPT4O into the list of known models
1 parent 9d39ebd commit e883e13

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

‎common/llm_client.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -55,9 +55,9 @@ class Model(Enum):
5555
# GROQ can be used as a provider with the following models: MIXTRAL, GEMMA.
5656
def is_valid_provider_model_combination(provider: Provider, model: Model) -> bool:
5757
if provider == Provider.OPENAI:
58-
return model in [Model.GPT3, Model.GPT4]
58+
return model in (Model.GPT3, Model.GPT4, Model.GPT4O)
5959
elif provider == Provider.GROQ:
60-
return model in [Model.MIXTRAL, Model.GEMMA, Model.LLAMA3_8B, Model.LLAMA3_70B]
60+
return model in (Model.MIXTRAL, Model.GEMMA, Model.LLAMA3_8B, Model.LLAMA3_70B)
6161
elif provider == Provider.PERPLEXITY:
6262
return model == Model.LLAMA3_70B_INSTRUCT
6363

0 commit comments

Comments
 (0)