|
@@ -187,18 +187,30 @@ class ProviderConfiguration(BaseModel):
|
|
|
:return:
|
|
|
"""
|
|
|
# get provider
|
|
|
- provider_record = (
|
|
|
- db.session.query(Provider)
|
|
|
- .filter(
|
|
|
- Provider.tenant_id == self.tenant_id,
|
|
|
- Provider.provider_type == ProviderType.CUSTOM.value,
|
|
|
- or_(
|
|
|
- Provider.provider_name == ModelProviderID(self.provider.provider).plugin_name,
|
|
|
+ model_provider_id = ModelProviderID(self.provider.provider)
|
|
|
+ if model_provider_id.is_langgenius():
|
|
|
+ provider_record = (
|
|
|
+ db.session.query(Provider)
|
|
|
+ .filter(
|
|
|
+ Provider.tenant_id == self.tenant_id,
|
|
|
+ Provider.provider_type == ProviderType.CUSTOM.value,
|
|
|
+ or_(
|
|
|
+ Provider.provider_name == model_provider_id.provider_name,
|
|
|
+ Provider.provider_name == self.provider.provider,
|
|
|
+ ),
|
|
|
+ )
|
|
|
+ .first()
|
|
|
+ )
|
|
|
+ else:
|
|
|
+ provider_record = (
|
|
|
+ db.session.query(Provider)
|
|
|
+ .filter(
|
|
|
+ Provider.tenant_id == self.tenant_id,
|
|
|
+ Provider.provider_type == ProviderType.CUSTOM.value,
|
|
|
Provider.provider_name == self.provider.provider,
|
|
|
- ),
|
|
|
+ )
|
|
|
+ .first()
|
|
|
)
|
|
|
- .first()
|
|
|
- )
|
|
|
|
|
|
# Get provider credential secret variables
|
|
|
provider_credential_secret_variables = self.extract_secret_variables(
|