ai-world-sdk 1.0.3 → 1.0.6

package/dist/index.js

@@ -10,7 +10,6 @@ exports.sdkConfig = exports.VideoGenerationClient = exports.ImageGenerationClien
 exports.createChatModel = createChatModel;
 const openai_1 = require("./chat_models/openai");
 const google_1 = require("./chat_models/google");
-const anthropic_1 = require("./chat_models/anthropic");
 const doubao_image_generation_1 = require("./doubao-image-generation");
 Object.defineProperty(exports, "DoubaoImageGenerationClient", { enumerable: true, get: function () { return doubao_image_generation_1.DoubaoImageGenerationClient; } });
 Object.defineProperty(exports, "ImageGenerationClient", { enumerable: true, get: function () { return doubao_image_generation_1.DoubaoImageGenerationClient; } });
@@ -32,8 +31,8 @@ var openai_2 = require("./chat_models/openai");
 Object.defineProperty(exports, "ChatOpenAI", { enumerable: true, get: function () { return openai_2.ChatOpenAI; } });
 var google_2 = require("./chat_models/google");
 Object.defineProperty(exports, "ChatGoogleGenerativeAI", { enumerable: true, get: function () { return google_2.ChatGoogleGenerativeAI; } });
-var anthropic_2 = require("./chat_models/anthropic");
-Object.defineProperty(exports, "ChatAnthropic", { enumerable: true, get: function () { return anthropic_2.ChatAnthropic; } });
+var anthropic_1 = require("./chat_models/anthropic");
+Object.defineProperty(exports, "ChatAnthropic", { enumerable: true, get: function () { return anthropic_1.ChatAnthropic; } });
 /**
  * Create a chat model instance based on model name
  */
@@ -49,33 +48,19 @@ function createChatModel(model, config) {
             ...config.headers,
         },
     };
-    const modelLower = model.toLowerCase();
-    if (modelLower.startsWith("gpt") || modelLower.startsWith("o1")) {
-        return new openai_1.ChatOpenAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("doubao-")) {
-        // Doubao 与 OpenAI 兼容，使用 ChatOpenAI
-        return new openai_1.ChatOpenAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("gemini")) {
-        return new google_1.ChatGoogleGenerativeAI({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else if (modelLower.startsWith("claude")) {
-        return new anthropic_1.ChatAnthropic({
-            modelName: model,
-            ...finalConfig,
-        });
-    }
-    else {
-        throw new Error(`Unsupported model: ${model}. Supported models: gpt-*, o1-*, doubao-*, gemini-*, claude-*`);
+    switch (config.provider) {
+        case "doubao":
+        case "aihubmix":
+            return new openai_1.ChatOpenAI({
+                modelName: model,
+                ...finalConfig,
+            });
+        case "gemini":
+            return new google_1.ChatGoogleGenerativeAI({
+                modelName: model,
+                ...finalConfig,
+            });
+        default:
+            throw new Error(`Unsupported provider: ${config.provider}`);
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-world-sdk",
-  "version": "1.0.3",
+  "version": "1.0.6",
   "description": "TypeScript SDK for AI World Platform - Chat Models, Image Generation, and Video Generation",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
@@ -9,11 +9,12 @@
     "prepublishOnly": "npm run build",
     "test": "jest",
     "test:watch": "jest --watch",
-    "test:non-stream": "jest -t 'ChatGoogleGenerativeAI - 非流式调用'",
-    "test:stream": "jest -t 'ChatGoogleGenerativeAI - 流式调用'",
+    "test:genimi-non-stream": "jest -t 'ChatGoogleGenerativeAI - 非流式调用'",
+    "test:genimi-stream": "jest -t 'ChatGoogleGenerativeAI - 流式调用'",
     "test:factory": "jest -t '工厂函数'",
     "test:bind": "jest -t 'bind 方法'",
     "test:batch": "jest -t 'batch 方法'",
+    "test:openai": "jest -t 'ChatOpenAI - 基础测试'",
     "test:aihubmix-no-stream": "jest -t 'aihubmix.com no stream'",
     "test:aihubmix-stream": "jest -t 'aihubmix.com stream'",
     "test:aihubmix-image": "jest -t 'aihubmix.com 图像生成'",