jyapayne/dify
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Azure openai init (#1929)

Browse source 
Co-authored-by: luowei <glpat-EjySCyNjWiLqAED-YmwM>
Co-authored-by: crazywoola <427733928@qq.com>
Co-authored-by: crazywoola <100913391+crazywoola@users.noreply.github.com>
This commit is contained in:
Charlie.Wei 2024-01-09 19:17:47 +08:00 • committed by GitHub
commit 5b24d7129e
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 151 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

8
api/core/entities/provider_entities.py
View file

@ -21,6 +21,12 @@ class SystemConfigurationStatus(Enum):
UNSUPPORTED = 'unsupported'
class RestrictModel(BaseModel):
model: str
base_model_name: Optional[str] = None
model_type: ModelType
class QuotaConfiguration(BaseModel):
"""
Model class for provider quota configuration.
@ -30,7 +36,7 @@ class QuotaConfiguration(BaseModel):
quota_limit: int
quota_used: int
is_valid: bool
restrict_llms: list[str] = []
restrict_models: list[RestrictModel] = []
class SystemConfiguration(BaseModel):