(seatunnel) branch dev updated: [Feature][Transform-V2] llm add deepseek (#8544)

[wanghailin]

This is an automated email from the ASF dual-hosted git repository.

wanghailin pushed a commit to branch dev
in repository https://gitbox.apache.org/repos/asf/seatunnel.git


The following commit(s) were added to refs/heads/dev by this push:
     new 6a39c9a883 [Feature][Transform-V2] llm add deepseek (#8544)
6a39c9a883 is described below

commit 6a39c9a883a505f667a32337ecb2233f78cdc754
Author: corgy-w <73771213+corg...@users.noreply.github.com>
AuthorDate: Fri Jan 17 15:57:03 2025 +0800

    [Feature][Transform-V2] llm add deepseek (#8544)
---
 docs/en/transform-v2/llm.md                                             | 2 +-
 docs/zh/transform-v2/llm.md                                             | 2 +-
 .../java/org/apache/seatunnel/transform/nlpmodel/ModelProvider.java     | 1 +
 .../java/org/apache/seatunnel/transform/nlpmodel/llm/LLMTransform.java  | 1 +
 4 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/docs/en/transform-v2/llm.md b/docs/en/transform-v2/llm.md
index c1c9798abe..680121cb4d 100644
--- a/docs/en/transform-v2/llm.md
+++ b/docs/en/transform-v2/llm.md
@@ -28,7 +28,7 @@ more.
 ### model_provider
 
 The model provider to use. The available options are:
-OPENAI, DOUBAO, KIMIAI, MICROSOFT, CUSTOM
+OPENAI, DOUBAO, DEEPSEEK, KIMIAI, MICROSOFT, CUSTOM
 
 > tips: If you use Microsoft, please make sure api_path cannot be empty
 
diff --git a/docs/zh/transform-v2/llm.md b/docs/zh/transform-v2/llm.md
index 7b505bde24..c1d05d59a3 100644
--- a/docs/zh/transform-v2/llm.md
+++ b/docs/zh/transform-v2/llm.md
@@ -26,7 +26,7 @@
 ### model_provider
 
 要使用的模型提供者。可用选项为:
-OPENAI、DOUBAO、KIMIAI、MICROSOFT, CUSTOM
+OPENAI,DOUBAO,DEEPSEEK,KIMIAI,MICROSOFT, CUSTOM
 
 > tips: 如果使用 Microsoft, 请确保 api_path 配置不能为空
 
diff --git 
a/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/ModelProvider.java
 
b/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/ModelProvider.java
index 3172137706..f18ffdfc8e 100644
--- 
a/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/ModelProvider.java
+++ 
b/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/ModelProvider.java
@@ -26,6 +26,7 @@ public enum ModelProvider {
             "https://ark.cn-beijing.volces.com/api/v3/embeddings";),
     QIANFAN("", 
"https://aip.baidubce.com/rpc/2.0/ai_custom/v1/wenxinworkshop/embeddings";),
     KIMIAI("https://api.moonshot.cn/v1/chat/completions";, ""),
+    DEEPSEEK("https://api.deepseek.com/chat/completions";, ""),
     MICROSOFT("", ""),
     CUSTOM("", ""),
     LOCAL("", "");
diff --git 
a/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/llm/LLMTransform.java
 
b/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/llm/LLMTransform.java
index c99b03776e..346fd68808 100644
--- 
a/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/llm/LLMTransform.java
+++ 
b/seatunnel-transforms-v2/src/main/java/org/apache/seatunnel/transform/nlpmodel/llm/LLMTransform.java
@@ -106,6 +106,7 @@ public class LLMTransform extends 
SingleFieldOutputTransform {
                                 config.get(LLMTransformConfig.API_KEY),
                                 
provider.usedLLMPath(config.get(LLMTransformConfig.API_PATH)));
                 break;
+            case DEEPSEEK:
             case OPENAI:
             case DOUBAO:
                 model =