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

feat: support chatglm_turbo model #1443 (#1460)

Browse Source
This commit is contained in:
takatost 2023-11-06 18:33:05 +08:00 committed by GitHub
parent a9ee18300e
commit 4dfbcd0b4e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/core/model_providers/providers/zhipuai_provider.py
View File

@ -26,6 +26,11 @@ class ZhipuAIProvider(BaseModelProvider):
def _get_fixed_model_list(self, model_type: ModelType) -> list[dict]: def _get_fixed_model_list(self, model_type: ModelType) -> list[dict]:
if model_type == ModelType.TEXT_GENERATION: if model_type == ModelType.TEXT_GENERATION:
return [ return [
{
'id': 'chatglm_turbo',
'name': 'chatglm_turbo',
'mode': ModelMode.CHAT.value,
},
{ {
'id': 'chatglm_pro', 'id': 'chatglm_pro',
'name': 'chatglm_pro', 'name': 'chatglm_pro',

6
api/core/model_providers/rules/zhipuai.json
View File

@ -11,6 +11,12 @@
}, },
"model_flexibility": "fixed", "model_flexibility": "fixed",
"price_config": { "price_config": {
"chatglm_turbo": {
"prompt": "0.005",
"completion": "0.005",
"unit": "0.001",
"currency": "RMB"
},
"chatglm_pro": { "chatglm_pro": {
"prompt": "0.01", "prompt": "0.01",
"completion": "0.01", "completion": "0.01",

2
api/core/third_party/langchain/llms/zhipuai_llm.py vendored
View File

@ -96,7 +96,7 @@ class ZhipuAIChatLLM(BaseChatModel):
return True return True
client: Any = None #: :meta private: client: Any = None #: :meta private:
model: str = "chatglm_lite" model: str = "chatglm_turbo"
"""Model name to use.""" """Model name to use."""
temperature: float = 0.95 temperature: float = 0.95
"""A non-negative float that tunes the degree of randomness in generation.""" """A non-negative float that tunes the degree of randomness in generation."""