初始化
This commit is contained in:
38
easy-agents-image/easy-agents-image-qwen/pom.xml
Normal file
38
easy-agents-image/easy-agents-image-qwen/pom.xml
Normal file
@@ -0,0 +1,38 @@
|
||||
<?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-qwen</name>
|
||||
<artifactId>easy-agents-image-qwen</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.alibaba</groupId>
|
||||
<artifactId>dashscope-sdk-java</artifactId>
|
||||
<!-- 请将 'the-latest-version' 替换为最新版本号:https://mvnrepository.com/artifact/com.alibaba/dashscope-sdk-java -->
|
||||
<version>2.18.2</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>junit</groupId>
|
||||
<artifactId>junit</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@@ -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.qwen;
|
||||
|
||||
import com.easyagents.core.model.client.HttpClient;
|
||||
import com.easyagents.core.model.image.*;
|
||||
import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesis;
|
||||
import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesisParam;
|
||||
import com.alibaba.dashscope.aigc.imagesynthesis.ImageSynthesisResult;
|
||||
import org.slf4j.Logger;
|
||||
import org.slf4j.LoggerFactory;
|
||||
|
||||
import java.util.Map;
|
||||
import java.util.Objects;
|
||||
|
||||
public class QwenImageModel implements ImageModel {
|
||||
private static final Logger LOG = LoggerFactory.getLogger(QwenImageModel.class);
|
||||
private final QwenImageModelConfig config;
|
||||
private final HttpClient httpClient = new HttpClient();
|
||||
|
||||
public QwenImageModel(QwenImageModelConfig config) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse generate(GenerateImageRequest request) {
|
||||
try {
|
||||
ImageSynthesis is = new ImageSynthesis();
|
||||
ImageSynthesisParam param =
|
||||
ImageSynthesisParam.builder()
|
||||
.apiKey(config.getApiKey())
|
||||
.model(null != request.getModel() ? request.getModel() : config.getModel())
|
||||
.size(request.getSize())
|
||||
.prompt(request.getPrompt())
|
||||
.seed(Integer.valueOf(String.valueOf(request.getOptionOrDefault("seed",1))))
|
||||
.build();
|
||||
ImageSynthesisResult result = is.call(param);
|
||||
if (Objects.isNull(result.getOutput().getResults())){
|
||||
return ImageResponse.error(result.getOutput().getMessage());
|
||||
}
|
||||
ImageResponse imageResponse = new ImageResponse();
|
||||
for(Map<String, String> item :result.getOutput().getResults()) {
|
||||
imageResponse.addImage(item.get("url"));
|
||||
}
|
||||
return imageResponse;
|
||||
} catch (Exception e) {
|
||||
return ImageResponse.error(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse img2imggenerate(GenerateImageRequest request) {
|
||||
throw new IllegalStateException("QwenImageModel Can not support img2imggenerate.");
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse edit(EditImageRequest request) {
|
||||
throw new IllegalStateException("QwenImageModel Can not support edit image.");
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse vary(VaryImageRequest request) {
|
||||
throw new IllegalStateException("QwenImageModel Can not support vary image.");
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
@@ -0,0 +1,49 @@
|
||||
/*
|
||||
* 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.qwen;
|
||||
|
||||
public class QwenImageModelConfig {
|
||||
private String endpoint = "https://dashscope.aliyuncs.com";
|
||||
|
||||
private String model = "flux-schnell";
|
||||
|
||||
private String apiKey;
|
||||
|
||||
|
||||
public String getEndpoint() {
|
||||
return endpoint;
|
||||
}
|
||||
|
||||
public void setEndpoint(String endpoint) {
|
||||
this.endpoint = endpoint;
|
||||
}
|
||||
|
||||
public String getModel() {
|
||||
return model;
|
||||
}
|
||||
|
||||
public void setModel(String model) {
|
||||
this.model = model;
|
||||
}
|
||||
|
||||
public String getApiKey() {
|
||||
return apiKey;
|
||||
}
|
||||
|
||||
public void setApiKey(String apiKey) {
|
||||
this.apiKey = apiKey;
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,54 @@
|
||||
/*
|
||||
* 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.qwen.QwenImageModel;
|
||||
import com.easyagents.image.qwen.QwenImageModelConfig;
|
||||
import org.junit.Test;
|
||||
|
||||
public class QwenImageModelTest {
|
||||
|
||||
@Test
|
||||
public void testGenImage() throws InterruptedException {
|
||||
Thread thread = new Thread(() -> {
|
||||
QwenImageModelConfig config = new QwenImageModelConfig();
|
||||
config.setApiKey("******************");
|
||||
ImageModel imageModel = new QwenImageModel(config);
|
||||
GenerateImageRequest request = new GenerateImageRequest();
|
||||
request.setPrompt("雨中, 竹林, 小路");
|
||||
request.setModel("flux-schnell");
|
||||
ImageResponse generate = imageModel.generate(request);
|
||||
System.out.println(generate);
|
||||
});
|
||||
|
||||
// Thread thread2 = new Thread(() -> {
|
||||
// QwenImageModelConfig config = new QwenImageModelConfig();
|
||||
// config.setApiKey("******************");
|
||||
// ImageModel imageModel = new QwenImageModel(config);
|
||||
// GenerateImageRequest request = new GenerateImageRequest();
|
||||
// request.setPrompt("雨中, 竹林, 小路");
|
||||
// request.setModel("flux-schnell");
|
||||
// ImageResponse generate = imageModel.generate(request);
|
||||
// });
|
||||
thread.start();
|
||||
// thread2.start();
|
||||
thread.join();
|
||||
// thread2.join();
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user