feat: support moonshot again (#442)

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),
+    ]
+);