Bläddra i källkod

fix: replicate text generation model validate (#923)

tags/0.3.15
takatost 2 år sedan
förälder
incheckning
95b179fb39
Inget konto är kopplat till bidragsgivarens mejladress
1 ändrade filer med 2 tillägg och 1 borttagningar
  1. 2
    1
      api/core/model_providers/providers/replicate_provider.py

+ 2
- 1
api/core/model_providers/providers/replicate_provider.py Visa fil

@@ -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:

Laddar…
Avbryt
Spara