初始化
This commit is contained in:
33
easy-agents-image/easy-agents-image-gitee/pom.xml
Normal file
33
easy-agents-image/easy-agents-image-gitee/pom.xml
Normal file
@@ -0,0 +1,33 @@
|
||||
<?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-gitee</name>
|
||||
<artifactId>easy-agents-image-gitee</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>
|
||||
<scope>compile</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>junit</groupId>
|
||||
<artifactId>junit</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
</project>
|
||||
@@ -0,0 +1,87 @@
|
||||
/*
|
||||
* 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.gitee;
|
||||
|
||||
import com.easyagents.core.model.client.HttpClient;
|
||||
import com.easyagents.core.model.image.*;
|
||||
import com.easyagents.core.util.Maps;
|
||||
import com.easyagents.core.util.StringUtil;
|
||||
import com.alibaba.fastjson2.JSON;
|
||||
import com.alibaba.fastjson2.JSONArray;
|
||||
import com.alibaba.fastjson2.JSONObject;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.Map;
|
||||
|
||||
public class GiteeImageModel implements ImageModel {
|
||||
private GiteeImageModelConfig config;
|
||||
private HttpClient httpClient = new HttpClient();
|
||||
|
||||
public GiteeImageModel(GiteeImageModelConfig config) {
|
||||
this.config = config;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse generate(GenerateImageRequest request) {
|
||||
Map<String, String> headers = new HashMap<>();
|
||||
headers.put("Content-Type", "application/json");
|
||||
headers.put("Authorization", "Bearer " + config.getApiKey());
|
||||
|
||||
String payload = Maps.of("model", config.getModel())
|
||||
.set("prompt", request.getPrompt())
|
||||
.setIfNotNull("n", request.getN())
|
||||
.set("size", request.getSize())
|
||||
.set("response_format", "url")
|
||||
.toJSON();
|
||||
|
||||
String url = config.getEndpoint() + "/v1/images/generations";
|
||||
String responseJson = httpClient.post(url, headers, payload);
|
||||
|
||||
if (StringUtil.noText(responseJson)) {
|
||||
return ImageResponse.error("response is no text");
|
||||
}
|
||||
|
||||
JSONObject root = JSON.parseObject(responseJson);
|
||||
JSONArray images = root.getJSONArray("data");
|
||||
if (images == null || images.isEmpty()) {
|
||||
return ImageResponse.error("image data is empty: " + responseJson);
|
||||
}
|
||||
ImageResponse response = new ImageResponse();
|
||||
for (int i = 0; i < images.size(); i++) {
|
||||
JSONObject imageObj = images.getJSONObject(i);
|
||||
response.addImage(imageObj.getString("url"));
|
||||
}
|
||||
|
||||
return response;
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse img2imggenerate(GenerateImageRequest request) {
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
@Override
|
||||
public ImageResponse edit(EditImageRequest request) {
|
||||
throw new UnsupportedOperationException("GiteeImageModel Can not support edit image.");
|
||||
}
|
||||
|
||||
@Override
|
||||
public ImageResponse vary(VaryImageRequest request) {
|
||||
throw new UnsupportedOperationException("GiteeImageModel 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.gitee;
|
||||
|
||||
import java.io.Serializable;
|
||||
|
||||
public class GiteeImageModelConfig implements Serializable {
|
||||
private String endpoint = "https://ai.gitee.com";
|
||||
private String model = "flux-1-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,51 @@
|
||||
/*
|
||||
* 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.Image;
|
||||
import com.easyagents.core.model.image.ImageModel;
|
||||
import com.easyagents.core.model.image.ImageResponse;
|
||||
import com.easyagents.image.gitee.GiteeImageModel;
|
||||
import com.easyagents.image.gitee.GiteeImageModelConfig;
|
||||
import org.junit.Test;
|
||||
|
||||
import java.io.File;
|
||||
|
||||
public class GiteeImageModelTest {
|
||||
|
||||
@Test
|
||||
public void testGenImage(){
|
||||
GiteeImageModelConfig config = new GiteeImageModelConfig();
|
||||
config.setApiKey("****");
|
||||
|
||||
ImageModel imageModel = new GiteeImageModel(config);
|
||||
|
||||
GenerateImageRequest request = new GenerateImageRequest();
|
||||
request.setPrompt("A cute little tiger standing in the high-speed train");
|
||||
request.setSize(1024,1024);
|
||||
ImageResponse generate = imageModel.generate(request);
|
||||
if (generate != null && generate.getImages() != null){
|
||||
int index = 0;
|
||||
for (Image image : generate.getImages()) {
|
||||
image.writeToFile(new File("/Users/michael/Desktop/test/image"+(index++)+".jpg"));
|
||||
}
|
||||
}
|
||||
|
||||
System.out.println(generate);
|
||||
}
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user