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