summaryrefslogtreecommitdiffstats
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/client/mod.rs1
-rw-r--r--src/client/moonshot.rs11
2 files changed, 12 insertions, 0 deletions
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),
+ ]
+);