AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-12 06:29:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: adjust anthropic (#1387)

Browse Source
This commit is contained in:
takatost 2023-10-20 15:27:46 +08:00 committed by GitHub
parent 288705fefd
commit 4c63cbf5b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
5 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/model_providers/providers/anthropic_provider.py
View File

@ -172,7 +172,7 @@ class AnthropicProvider(BaseModelProvider):
def should_deduct_quota(self): def should_deduct_quota(self):
if hosted_model_providers.anthropic and \ if hosted_model_providers.anthropic and \
hosted_model_providers.anthropic.quota_limit and hosted_model_providers.anthropic.quota_limit > 0: hosted_model_providers.anthropic.quota_limit and hosted_model_providers.anthropic.quota_limit > -1:
return True return True
return False return False

2
api/core/model_providers/providers/azure_openai_provider.py
View File

@ -329,7 +329,7 @@ class AzureOpenAIProvider(BaseModelProvider):
def should_deduct_quota(self): def should_deduct_quota(self):
if hosted_model_providers.azure_openai \ if hosted_model_providers.azure_openai \
and hosted_model_providers.azure_openai.quota_limit and hosted_model_providers.azure_openai.quota_limit > 0: and hosted_model_providers.azure_openai.quota_limit and hosted_model_providers.azure_openai.quota_limit > -1:
return True return True
return False return False

6
api/core/model_providers/providers/hosted.py
View File

@ -11,7 +11,7 @@ class HostedOpenAI(BaseModel):
api_organization: str = None api_organization: str = None
api_key: str api_key: str
quota_limit: int = 0 quota_limit: int = 0
"""Quota limit for the openai hosted model. 0 means unlimited.""" """Quota limit for the openai hosted model. -1 means unlimited."""
paid_enabled: bool = False paid_enabled: bool = False
paid_stripe_price_id: str = None paid_stripe_price_id: str = None
paid_increase_quota: int = 1 paid_increase_quota: int = 1
@ -21,14 +21,14 @@ class HostedAzureOpenAI(BaseModel):
api_base: str api_base: str
api_key: str api_key: str
quota_limit: int = 0 quota_limit: int = 0
"""Quota limit for the azure openai hosted model. 0 means unlimited.""" """Quota limit for the azure openai hosted model. -1 means unlimited."""
class HostedAnthropic(BaseModel): class HostedAnthropic(BaseModel):
api_base: str = None api_base: str = None
api_key: str api_key: str
quota_limit: int = 0 quota_limit: int = 0
"""Quota limit for the anthropic hosted model. 0 means unlimited.""" """Quota limit for the anthropic hosted model. -1 means unlimited."""
paid_enabled: bool = False paid_enabled: bool = False
paid_stripe_price_id: str = None paid_stripe_price_id: str = None
paid_increase_quota: int = 1000000 paid_increase_quota: int = 1000000

2
api/core/model_providers/providers/openai_provider.py
View File

@ -250,7 +250,7 @@ class OpenAIProvider(BaseModelProvider):
def should_deduct_quota(self): def should_deduct_quota(self):
if hosted_model_providers.openai \ if hosted_model_providers.openai \
and hosted_model_providers.openai.quota_limit and hosted_model_providers.openai.quota_limit > 0: and hosted_model_providers.openai.quota_limit and hosted_model_providers.openai.quota_limit > -1:
return True return True
return False return False

2
api/core/model_providers/rules/anthropic.json
View File

@ -9,7 +9,7 @@
"trial" "trial"
], ],
"quota_unit": "tokens", "quota_unit": "tokens",
"quota_limit": 600000 "quota_limit": 0
}, },
"model_flexibility": "fixed", "model_flexibility": "fixed",
"price_config": { "price_config": {