|
@@ -116,7 +116,8 @@ class ReplicateProvider(BaseModelProvider):
|
|
|
and 'Embedding' not in rst.openapi_schema['components']['schemas']:
|
|
|
raise CredentialsValidateFailedError(f"Model {model_name}:{version} is not a Embedding model.")
|
|
|
elif model_type == ModelType.TEXT_GENERATION \
|
|
|
- and ('type' not in rst.openapi_schema['components']['schemas']['Output']['items']
|
|
|
+ and ('items' not in rst.openapi_schema['components']['schemas']['Output']
|
|
|
+ or 'type' not in rst.openapi_schema['components']['schemas']['Output']['items']
|
|
|
or rst.openapi_schema['components']['schemas']['Output']['items']['type'] != 'string'):
|
|
|
raise CredentialsValidateFailedError(f"Model {model_name}:{version} is not a Text Generation model.")
|
|
|
except ReplicateError as e:
|