EasyFlow-Tech-Dev/Easy-Agents
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

初始化

Browse Source
This commit is contained in:
陈子默
2026-02-22 18:55:40 +08:00
commit 8392cdd861
496 changed files with 45020 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
easy-agents-image/easy-agents-image-tencent/pom.xml Normal file
View File

@@ -0,0 +1,37 @@
<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>
<parent>
<groupId>com.easyagents</groupId>
<artifactId>easy-agents-image</artifactId>
<version>${revision}</version>
</parent>
<name>easy-agents-image-tencent</name>
<artifactId>easy-agents-image-tencent</artifactId>
<properties>
<maven.compiler.source>8</maven.compiler.source>
<maven.compiler.target>8</maven.compiler.target>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
</properties>
<dependencies>
<dependency>
<groupId>com.easyagents</groupId>
<artifactId>easy-agents-core</artifactId>
</dependency>
<dependency>
<groupId>com.tencentcloudapi</groupId>
<artifactId>tencentcloud-sdk-java-common</artifactId>
<version>3.1.1261</version>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<scope>test</scope>
</dependency>
</dependencies>
</project>

237
easy-agents-image/easy-agents-image-tencent/src/main/java/com/easyagents/image/tencent/TencentImageModel.java Normal file
View File

@@ -0,0 +1,237 @@
/*
* Copyright (c) 2023-2026, Easy-Agents (fuhai999@gmail.com).
* <p>
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.easyagents.image.tencent;
import com.easyagents.core.model.client.HttpClient;
import com.easyagents.core.model.image.*;
import com.easyagents.core.util.Maps;
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.tencentcloudapi.common.DatatypeConverter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.security.MessageDigest;
import java.text.SimpleDateFormat;
import java.util.*;
public class TencentImageModel implements ImageModel {
private static final Logger LOG = LoggerFactory.getLogger(TencentImageModel.class);
private final TencentImageModelConfig config;
private final HttpClient httpClient = new HttpClient();
public TencentImageModel(TencentImageModelConfig config) {
this.config = config;
}
@Override
public ImageResponse generate(GenerateImageRequest request) {
try {
String payload = promptToPayload(request);
Map<String, String> headers = createAuthorizationToken("SubmitHunyuanImageJob", payload);
String response = httpClient.post(config.getEndpoint(), headers, payload);
JSONObject jsonObject = JSON.parseObject(response);
JSONObject error = jsonObject.getJSONObject("Response").getJSONObject("Error");
if (error != null && !error.isEmpty()) {
return ImageResponse.error(error.getString("Message"));
}
Object jobId = jsonObject.getJSONObject("Response").get("JobId");
if (Objects.isNull(jobId)) {
return ImageResponse.error("response is no jobId");
}
String id = (String) jobId;
return getImage(id);
} catch (Exception e) {
return ImageResponse.error(e.getMessage());
}
}
@Override
public ImageResponse img2imggenerate(GenerateImageRequest request) {
return null;
}
@Override
public ImageResponse edit(EditImageRequest request) {
throw new IllegalStateException("TencentImageModel Can not support edit image.");
}
@Override
public ImageResponse vary(VaryImageRequest request) {
throw new IllegalStateException("TencentImageModel Can not support vary image.");
}
private static final Object LOCK = new Object();
private ImageResponse getImage(String jobId) {
ImageResponse imageResponse = null;
while (true) {
synchronized (LOCK) {
imageResponse = callService(jobId);
if (!Objects.isNull(imageResponse)) {
break;
}
// 等待一段时间再重试
try {
LOCK.wait(1000);
} catch (InterruptedException e) {
// 线程在等待时被中断
Thread.currentThread().interrupt();
imageResponse = ImageResponse.error(e.toString());
break;
}
}
}
return imageResponse;
}
public ImageResponse callService(String jobId) {
try {
String payload = Maps.of("JobId", jobId).toJSON();
Map<String, String> headers = createAuthorizationToken("QueryHunyuanImageJob", payload);
String resp = httpClient.post(config.getEndpoint(), headers, payload);
JSONObject resultJson = JSONObject.parseObject(resp).getJSONObject("Response");
JSONObject error = resultJson.getJSONObject("Error");
if (error != null && !error.isEmpty()) {
return ImageResponse.error(error.getString("Message"));
}
if (Objects.isNull(resultJson.get("JobStatusCode"))) {
return ImageResponse.error("response is no JobStatusCode");
}
Integer jobStatusCode = resultJson.getInteger("JobStatusCode");
if (Objects.equals(5, jobStatusCode)) {
//处理完成
if (Objects.isNull(resultJson.get("ResultImage"))) {
return ImageResponse.error("response is no ResultImage");
}
JSONArray imagesArray = resultJson.getJSONArray("ResultImage");
ImageResponse response = new ImageResponse();
for (int i = 0; i < imagesArray.size(); i++) {
String imageObj = imagesArray.getString(i);
response.addImage(imageObj);
}
return response;
}
if (Objects.equals(4, jobStatusCode)) {
//处理错误
return ImageResponse.error(resultJson.getString("JobErrorMsg"));
}
} catch (Exception e) {
return ImageResponse.error(e.getMessage());
}
return null;
}
public static String promptToPayload(GenerateImageRequest request) {
return Maps.of("Prompt", request.getPrompt())
.setIfNotEmpty("NegativePrompt", request.getNegativePrompt())
.setIfNotEmpty("Style", request.getSize())
.setIfNotEmpty("Resolution", request.getQuality())
.setIfNotEmpty("Num", request.getN())
.setIfNotEmpty(request.getOptions())
.toJSON();
}
private final static Charset UTF8 = StandardCharsets.UTF_8;
private final static String CT_JSON = "application/json; charset=utf-8";
public static byte[] hmac256(byte[] key, String msg) throws Exception {
Mac mac = Mac.getInstance("HmacSHA256");
SecretKeySpec secretKeySpec = new SecretKeySpec(key, mac.getAlgorithm());
mac.init(secretKeySpec);
return mac.doFinal(msg.getBytes(UTF8));
}
public static String sha256Hex(String s) throws Exception {
MessageDigest md = MessageDigest.getInstance("SHA-256");
byte[] d = md.digest(s.getBytes(UTF8));
return DatatypeConverter.printHexBinary(d).toLowerCase();
}
/**
* @return java.util.Map<java.lang.String, java.lang.String>
* @Author sunch
* @Description 封装参数
* @Date 17:34 2025/3/5
* @Param [action, payload]
*/
public Map<String, String> createAuthorizationToken(String action, String payload) {
try {
String service = config.getService();
String host = config.getHost();
String version = "2023-09-01";
String algorithm = "TC3-HMAC-SHA256";
String timestamp = String.valueOf(System.currentTimeMillis() / 1000);
SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd");
// 注意时区，否则容易出错
sdf.setTimeZone(TimeZone.getTimeZone("UTC"));
String date = sdf.format(new Date(Long.parseLong(timestamp + "000")));
// ************* 步骤 1：拼接规范请求串 *************
String httpRequestMethod = "POST";
String canonicalUri = "/";
String canonicalQueryString = "";
String canonicalHeaders = "content-type:application/json; charset=utf-8\n"
+ "host:" + host + "\n" + "x-tc-action:" + action.toLowerCase() + "\n";
String signedHeaders = "content-type;host;x-tc-action";
String hashedRequestPayload = sha256Hex(payload);
String canonicalRequest = httpRequestMethod + "\n" + canonicalUri + "\n" + canonicalQueryString + "\n"
+ canonicalHeaders + "\n" + signedHeaders + "\n" + hashedRequestPayload;
// System.out.println(canonicalRequest);
// ************* 步骤 2：拼接待签名字符串 *************
String credentialScope = date + "/" + service + "/" + "tc3_request";
String hashedCanonicalRequest = sha256Hex(canonicalRequest);
String stringToSign = algorithm + "\n" + timestamp + "\n" + credentialScope + "\n" + hashedCanonicalRequest;
// System.out.println(stringToSign);
// ************* 步骤 3：计算签名 *************
byte[] secretDate = hmac256(("TC3" + config.getApiKey()).getBytes(UTF8), date);
byte[] secretService = hmac256(secretDate, service);
byte[] secretSigning = hmac256(secretService, "tc3_request");
String signature = DatatypeConverter.printHexBinary(hmac256(secretSigning, stringToSign)).toLowerCase();
// System.out.println(signature);
// ************* 步骤 4：拼接 Authorization *************
String authorization = algorithm + " " + "Credential=" + config.getApiSecret() + "/" + credentialScope + ", "
+ "SignedHeaders=" + signedHeaders + ", " + "Signature=" + signature;
// System.out.println(authorization);
Map<String, String> headers = new HashMap<>();
headers.put("Authorization", authorization);
headers.put("Content-Type", CT_JSON);
headers.put("Host", host);
headers.put("X-TC-Action", action);
headers.put("X-TC-Timestamp", timestamp);
headers.put("X-TC-Version", version);
headers.put("X-TC-Region", config.getRegion());
return headers;
} catch (Exception e) {
throw new RuntimeException(e);
}
}
}

78
easy-agents-image/easy-agents-image-tencent/src/main/java/com/easyagents/image/tencent/TencentImageModelConfig.java Normal file
View File

@@ -0,0 +1,78 @@
/*
* Copyright (c) 2023-2026, Easy-Agents (fuhai999@gmail.com).
* <p>
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.easyagents.image.tencent;
public class TencentImageModelConfig {
private String endpoint = "https://hunyuan.tencentcloudapi.com";
private String apiKey;
private String apiSecret;
private String service = "hunyuan";
private String region = "ap-guangzhou";
public String getEndpoint() {
return endpoint;
}
public void setEndpoint(String endpoint) {
this.endpoint = endpoint;
}
public String getRegion() {
return region;
}
public void setRegion(String region) {
this.region = region;
}
public String getService() {
return service;
}
public void setService(String service) {
this.service = service;
}
public String getApiKey() {
return apiKey;
}
public void setApiKey(String apiKey) {
this.apiKey = apiKey;
}
public String getApiSecret() {
return apiSecret;
}
public void setApiSecret(String apiSecret) {
this.apiSecret = apiSecret;
}
public String getHost() {
String endpoint = getEndpoint();
if (endpoint.toLowerCase().startsWith("https://")) {
endpoint = endpoint.substring(8);
} else if (endpoint.toLowerCase().startsWith("http://")) {
endpoint = endpoint.substring(7);
}
return endpoint;
}
}

59
easy-agents-image/easy-agents-image-tencent/src/test/java/com/easyagents/image/test/TencentImageModelTest.java Normal file
View File

@@ -0,0 +1,59 @@
/*
* Copyright (c) 2023-2026, Easy-Agents (fuhai999@gmail.com).
* <p>
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* <p>
* http://www.apache.org/licenses/LICENSE-2.0
* <p>
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.easyagents.image.test;
import com.easyagents.core.model.image.GenerateImageRequest;
import com.easyagents.core.model.image.ImageModel;
import com.easyagents.core.model.image.ImageResponse;
import com.easyagents.image.tencent.TencentImageModel;
import com.easyagents.image.tencent.TencentImageModelConfig;
import org.junit.Test;
public class TencentImageModelTest {
@Test
public void testGenImage() throws InterruptedException {
Thread thread = new Thread(() -> {
TencentImageModelConfig config = new TencentImageModelConfig();
config.setApiSecret("*****************");
config.setApiKey("*****************");
ImageModel imageModel = new TencentImageModel(config);
GenerateImageRequest request = new GenerateImageRequest();
request.setPrompt("雨中, 竹林, 小路");
request.setN(1);
ImageResponse generate = imageModel.generate(request);
System.out.println(generate);
System.out.flush();
});
Thread thread2 = new Thread(() -> {
TencentImageModelConfig config = new TencentImageModelConfig();
config.setApiSecret("*****************");
config.setApiKey("*****************");
ImageModel imageModel = new TencentImageModel(config);
GenerateImageRequest request = new GenerateImageRequest();
request.setPrompt("雨中, 竹林, 小路, 人生");
request.setN(1);
ImageResponse generate = imageModel.generate(request);
System.out.println(generate);
System.out.flush();
});
thread.start();
thread2.start();
thread.join();
thread2.join();
}
}