初始化

This commit is contained in:
2026-02-22 18:55:40 +08:00
commit 8392cdd861
496 changed files with 45020 additions and 0 deletions

View File

@@ -0,0 +1,54 @@
<?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>
<artifactId>easy-agents-image-volcengine</artifactId>
<name>easy-agents-image-volcengine</name>
<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>
<!--火山方舟 Java SDK-->
<dependency>
<groupId>com.volcengine</groupId>
<artifactId>volcengine-java-sdk-ark-runtime</artifactId>
<version>0.2.9</version>
</dependency>
<dependency>
<groupId>com.volcengine</groupId>
<artifactId>volc-sdk-java</artifactId>
<version>1.0.221</version>
</dependency>
<dependency>
<groupId>junit</groupId>
<artifactId>junit</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>com.google.guava</groupId>
<artifactId>guava</artifactId>
<version>33.5.0-jre</version>
<scope>test</scope>
</dependency>
</dependencies>
</project>

View File

@@ -0,0 +1,80 @@
/*
* 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.volcengine;
import com.easyagents.core.model.image.*;
import com.alibaba.fastjson2.JSONArray;
import com.alibaba.fastjson2.JSONObject;
import com.volcengine.service.visual.IVisualService;
import com.volcengine.service.visual.impl.VisualServiceImpl;
public class VolcengineImageModel implements ImageModel {
private VolcengineImageModelConfig config;
private IVisualService visualService = VisualServiceImpl.getInstance();
public VolcengineImageModel(VolcengineImageModelConfig config) {
this.config = config;
visualService.setAccessKey(config.getAccessKey());
visualService.setSecretKey(config.getSecretKey());
}
private ImageResponse processImageRequest(GenerateImageRequest request) {
JSONObject req = new JSONObject(request.getOptions());
ImageResponse responseimage = new ImageResponse();
try {
Object response = visualService.cvProcess(req);
if (response instanceof JSONObject) {
JSONObject jsonResponse = (JSONObject) response;
JSONObject dataObject = jsonResponse.getJSONObject("data");
JSONArray imageUrlsArray = dataObject.getJSONArray("image_urls");
for (int i = 0; i < imageUrlsArray.size(); i++) {
responseimage.addImage(imageUrlsArray.getString(i));
}
} else {
throw new RuntimeException("Unexpected response type: " + response.getClass().getName());
}
return responseimage;
} catch (Exception e) {
ImageResponse.error(e.getMessage());
e.printStackTrace(); // 记录堆栈跟踪方便调试
return responseimage;
}
}
@Override
public ImageResponse generate(GenerateImageRequest request) {
return processImageRequest(request);
}
@Override
public ImageResponse img2imggenerate(GenerateImageRequest request) {
return processImageRequest(request);
}
@Override
public ImageResponse edit(EditImageRequest request) {
throw new UnsupportedOperationException("not support edit image");
}
@Override
public ImageResponse vary(VaryImageRequest request) {
throw new UnsupportedOperationException("not support vary image");
}
}

View File

@@ -0,0 +1,40 @@
/*
* 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.volcengine;
import java.io.Serializable;
public class VolcengineImageModelConfig implements Serializable {
private String accessKey;
private String secretKey;
public String getAccessKey() {
return accessKey;
}
public void setAccessKey(String accessKey) {
this.accessKey = accessKey;
}
public String getSecretKey() {
return secretKey;
}
public void setSecretKey(String secretKey) {
this.secretKey = secretKey;
}
}

View File

@@ -0,0 +1,99 @@
import com.easyagents.core.model.image.GenerateImageRequest;
import com.easyagents.core.model.image.ImageResponse;
import com.easyagents.image.volcengine.VolcengineImageModel;
import com.easyagents.image.volcengine.VolcengineImageModelConfig;
import com.alibaba.fastjson2.JSONObject;
import org.junit.Test;
import java.io.File;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Base64;
import java.util.List;
public class VolcengineImageTest {
@Test
public void testGenImage(){
VolcengineImageModelConfig config = new VolcengineImageModelConfig();
config.setAccessKey("*********************");
config.setSecretKey("*********************");
VolcengineImageModel imageModel = new VolcengineImageModel(config);
GenerateImageRequest request = new GenerateImageRequest();
JSONObject req=new JSONObject();
//请求Body(查看接口文档请求参数-请求示例,将请求参数内容复制到此)参考通用2.1-文生图
req.put("req_key","high_aes_general_v21_L");
req.put("prompt","千军万马");
req.put("model_version","general_v2.1_L");
req.put("req_schedule_conf","general_v20_9B_pe");
req.put("llm_seed",-1);
req.put("seed",-1);
req.put("scale",3.5);
req.put("ddim_steps",25);
req.put("width",512);
req.put("height",512);
req.put("use_pre_llm",true);
req.put("use_sr",true);
req.put("sr_seed",-1);
req.put("sr_strength",0.4);
req.put("sr_scale",3.5);
req.put("sr_steps",20);
req.put("is_only_sr",false);
req.put("return_url",true);
// 创建子级JSONObject
JSONObject subData = new JSONObject();
subData.put("add_logo", true);
subData.put("position", 2);
subData.put("language", 0);
subData.put("opacity", 0.3);
subData.put("logo_text_content", "wangyangyang");
req.put("logo_info",subData);
request.setOptions(req);
ImageResponse generate = imageModel.generate(request);
System.out.println(generate);
}
@Test( )
public void testImg2ImgXLSft() throws IOException {
VolcengineImageModelConfig config = new VolcengineImageModelConfig();
config.setAccessKey("*********************");
config.setSecretKey("*********************");
VolcengineImageModel imageModel = new VolcengineImageModel(config);
GenerateImageRequest request = new GenerateImageRequest();
JSONObject req=new JSONObject();
req.put("req_key","i2i_xl_sft");
List<String> images=new ArrayList<>();
File file = new File(System.getProperty("user.dir"), "../../testresource/ark_demo_img_1.png");
// 将图片读取为字节数组
byte[] imageBytes = Files.readAllBytes(Paths.get(file.toURI()));
// 将字节数组编码为Base64
String base64String = Base64.getEncoder().encodeToString(imageBytes);
images.add(base64String);
// images.add("https://ark-project.tos-cn-beijing.volces.com/doc_image/ark_demo_img_1.png");
req.put("binary_data_base64",images);
req.put("prompt","根据图片内容生成风格、服装及发型一样的亚洲美女图片");
req.put("return_url",true);
request.setOptions(req);
ImageResponse generate = imageModel.img2imggenerate(request);
System.out.println(generate);
}
}

View File

@@ -0,0 +1,52 @@
import com.alibaba.fastjson2.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.volcengine.service.visual.IVisualService;
import com.volcengine.service.visual.impl.VisualServiceImpl;
public class VolcengineTest {
//以下是同步调用直接返回结果的方法按照实际需求可调用sdk同步及异步相关方法
public static void main(String[] args) {
IVisualService visualService = VisualServiceImpl.getInstance();
// call below method if you dont set ak and sk in /.vcloud/config
visualService.setAccessKey("AKLTNmU0M2RkNWZkMmZmNDQwYWI2NTZiMjA1ODYxY2M3MjE");
visualService.setSecretKey("TkdObFpXSTFZMlJtWldReE5EVTRPRGt4TkRsaE1EVTRaalpsTnpnMllURQ==");
JSONObject req=new JSONObject();
//请求Body(查看接口文档请求参数-请求示例,将请求参数内容复制到此)
req.put("req_key","high_aes_general_v21_L");
req.put("prompt","千军万马");
req.put("model_version","general_v2.1_L");
req.put("req_schedule_conf","general_v20_9B_pe");
req.put("llm_seed",-1);
req.put("seed",-1);
req.put("scale",3.5);
req.put("ddim_steps",25);
req.put("width",512);
req.put("height",512);
req.put("use_pre_llm",true);
req.put("use_sr",true);
req.put("sr_seed",-1);
req.put("sr_strength",0.4);
req.put("sr_scale",3.5);
req.put("sr_steps",20);
req.put("is_only_sr",false);
req.put("return_url",true);
//创建子级JSONObject
JSONObject subData = new JSONObject();
subData.put("add_logo", true);
subData.put("position", 2);
subData.put("language", 0);
subData.put("opacity", 0.3);
subData.put("logo_text_content", "wangyangyang");
req.put("logo_info",subData);
try {
Object response = visualService.cvProcess(req);
System.out.println(JSON.toJSONString(response));
} catch (Exception e) {
e.printStackTrace();
}
}
}