diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-25 21:19:04 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-25 21:19:04 +0800 |
| commit | 503ec98bd14fd833c5687790234f550a24b01107 (patch) | |
| tree | f9508313553533e11d6423951a999583a72c3698 | |
| parent | 29a4ffd5146af917af3fba30c7d3b7739c7bad0c (diff) | |
| download | aichat-503ec98bd14fd833c5687790234f550a24b01107.tar.gz | |
feat: support moonshot again (#442)
| -rw-r--r-- | config.example.yaml | 4 | ||||
| -rw-r--r-- | src/client/mod.rs | 1 | ||||
| -rw-r--r-- | src/client/moonshot.rs | 11 |
3 files changed, 16 insertions, 0 deletions
diff --git a/config.example.yaml b/config.example.yaml index 10eeeaa..79e369a 100644 --- a/config.example.yaml +++ b/config.example.yaml @@ -119,3 +119,7 @@ clients: # See https://help.aliyun.com/zh/dashscope/ - type: qianwen api_key: sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx + + # See https://platform.moonshot.cn/docs/intro + - type: moonshot + api_key: sk-xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx diff --git a/src/client/mod.rs b/src/client/mod.rs index 9de07d3..ff7ea42 100644 --- a/src/client/mod.rs +++ b/src/client/mod.rs @@ -32,4 +32,5 @@ register_client!( (ollama, "ollama", OllamaConfig, OllamaClient), (ernie, "ernie", ErnieConfig, ErnieClient), (qianwen, "qianwen", QianwenConfig, QianwenClient), + (moonshot, "moonshot", MoonshotConfig, MoonshotClient), ); diff --git a/src/client/moonshot.rs b/src/client/moonshot.rs new file mode 100644 index 0000000..6befadb --- /dev/null +++ b/src/client/moonshot.rs @@ -0,0 +1,11 @@ +openai_compatible_module!( + MoonshotConfig, + MoonshotClient, + "https://api.moonshot.cn/v1", + [ + // https://platform.moonshot.cn/docs/intro#%E6%A8%A1%E5%9E%8B%E5%88%97%E8%A1%A8 + ("moonshot-v1-8k", "text", 8000), + ("moonshot-v1-32k", "text", 32000), + ("moonshot-v1-128k", "text", 128000), + ] +); |