Эх сурвалжийг харах

去掉含有敏感信息的AI模块测试代码

luoshi 1 жил өмнө
parent
commit
b677959bbb

+ 0 - 14
src/main/java/com/qmth/ops/biz/ai/client/aliyun/llm/AliyunChatClient.java

@@ -3,7 +3,6 @@ package com.qmth.ops.biz.ai.client.aliyun.llm;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.qmth.boot.core.ai.model.llm.ChatRequest;
 import com.qmth.boot.core.ai.model.llm.ChatResult;
-import com.qmth.boot.core.ai.model.llm.ChatRole;
 import com.qmth.boot.core.exception.ReentrantException;
 import com.qmth.boot.core.exception.StatusException;
 import com.qmth.ops.biz.ai.client.ChatApiClient;
@@ -58,17 +57,4 @@ public class AliyunChatClient extends ChatApiClient {
         }
     }
 
-    public static void main(String[] args) throws Exception {
-        ChatApiConfig config = new ChatApiConfig();
-        config.setSupplier("aliyun");
-        config.setUrl("https://dashscope.aliyuncs.com/api/v1/services/aigc/text-generation/generation");
-        config.setSecret("sk-94599ccd34e240de8818310387386d91");
-        config.setModel("qwen1.5-7b-chat");
-        config.setQpm(0);
-        AliyunChatClient client = new AliyunChatClient(config);
-        ChatRequest request = new ChatRequest();
-        request.addMessage(ChatRole.user, "who are you");
-        System.out.println(new ObjectMapper().writeValueAsString(client.call(request)));
-    }
-
 }

+ 0 - 13
src/main/java/com/qmth/ops/biz/ai/client/aliyun/ocr/AliyunOcrClient.java

@@ -6,7 +6,6 @@ import com.qmth.boot.core.exception.ParameterException;
 import com.qmth.boot.core.exception.ReentrantException;
 import com.qmth.boot.core.exception.StatusException;
 import com.qmth.boot.core.exception.UnauthorizedException;
-import com.qmth.boot.tools.models.ByteArray;
 import com.qmth.boot.tools.uuid.FastUUID;
 import com.qmth.ops.biz.ai.client.OcrApiClient;
 import com.qmth.ops.biz.ai.client.OcrApiConfig;
@@ -18,7 +17,6 @@ import okhttp3.RequestBody;
 import javax.crypto.Mac;
 import javax.crypto.SecretKey;
 import javax.crypto.spec.SecretKeySpec;
-import java.io.File;
 import java.io.IOException;
 import java.io.UnsupportedEncodingException;
 import java.net.URL;
@@ -176,15 +174,4 @@ public class AliyunOcrClient extends OcrApiClient {
                 .post(RequestBody.create(MediaType.parse("application/octet-stream"), image)).build();
     }
 
-    public static void main(String[] args) throws Exception {
-        OcrApiConfig config = new OcrApiConfig();
-        config.setUrl("https://ocr-api.cn-hangzhou.aliyuncs.com/");
-        config.setKey("LTAI5t6D5p62tDjYgwSz2mTR");
-        config.setSecret("twrXT7Dp1kG1bV5HZn6vgpoypu9PnZ");
-        config.setQps(0);
-        AliyunOcrClient client = new AliyunOcrClient(config);
-        System.out.println(client.forImage(OcrType.GENERAL,
-                ByteArray.fromFile(new File("/Users/luoshi/Downloads/cache/1-1.jpg")).value()));
-    }
-
 }

+ 0 - 15
src/main/java/com/qmth/ops/biz/ai/client/azure/llm/AzureChatClient.java

@@ -3,7 +3,6 @@ package com.qmth.ops.biz.ai.client.azure.llm;
 import com.fasterxml.jackson.databind.ObjectMapper;
 import com.qmth.boot.core.ai.model.llm.ChatRequest;
 import com.qmth.boot.core.ai.model.llm.ChatResult;
-import com.qmth.boot.core.ai.model.llm.ChatRole;
 import com.qmth.boot.core.exception.NotFoundException;
 import com.qmth.boot.core.exception.StatusException;
 import com.qmth.ops.biz.ai.client.ChatApiClient;
@@ -52,18 +51,4 @@ public class AzureChatClient extends ChatApiClient {
             throw new StatusException(message != null ? message : "chat model error");
         }
     }
-
-    public static void main(String[] args) throws Exception {
-        ChatApiConfig config = new ChatApiConfig();
-        config.setSupplier("Azure");
-        config.setUrl(
-                "https://qmth-ai.openai.azure.com/openai/deployments/{model}/chat/completions?api-version=2024-02-01");
-        config.setSecret("ed11ce6c22984fc085db7acdd023828b");
-        config.setModel("gpt35");
-        config.setQpm(0);
-        AzureChatClient client = new AzureChatClient(config);
-        ChatRequest request = new ChatRequest();
-        request.addMessage(ChatRole.user, "who are you");
-        System.out.println(new ObjectMapper().writeValueAsString(client.call(request)));
-    }
 }