fixed bug in check_model_availability

This commit is contained in:
Paul Gauthier 2023-06-06 07:28:12 -07:00
parent 1ceeecddca
commit de0df5e7b1

View file

@ -36,7 +36,7 @@ class Coder:
def check_model_availability(self, main_model):
available_models = openai.Model.list()
model_ids = [model.id for model in available_models["data"]]
return main_model in model_ids
return main_model.value in model_ids
def __init__(
self,