add: [llm] moonshot

luochen1990 · Oct 10, 2024 · 55bb79d · 55bb79d
1 parent 0037bf8
commit 55bb79d
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/src/ai_powered/llm/known_models.py b/src/ai_powered/llm/known_models.py
@@ -49,7 +49,8 @@ def equals(s: str) -> Callable[[str], bool]:
     ''' create a function to check if a string contains a substring '''
     return lambda text: s == text
 
-KNOWN_PLATFORMS : list[KnownPlatform] = [
+
+KNOWN_PLATFORMS: list[KnownPlatform] = [
     KnownPlatform(
         platform_name = "openai",
         match_platform_url = contains("openai"),
@@ -66,6 +67,13 @@ def equals(s: str) -> Callable[[str], bool]:
             KnownModel("deepseek-coder", {ModelFeature.tools}),
         ]
     ),
+    KnownPlatform(
+        platform_name = "moonshot",
+        match_platform_url = contains("moonshot"),
+        known_model_list = [
+            KnownModel("moonshot-v1-auto", {ModelFeature.tools}),
+        ]
+    ),
     KnownPlatform(
         platform_name = "localhost",
         match_platform_url = contains("localhost"),