初始化

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,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-openai</name>
<artifactId>easy-agents-image-openai</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>

View File

@@ -0,0 +1,89 @@
/*
* 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.openai;
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 OpenAIImageModel extends BaseImageModel<OpenAIImageModelConfig> {
private OpenAIImageModelConfig config;
private HttpClient httpClient = new HttpClient();
public OpenAIImageModel(OpenAIImageModelConfig config) {
super(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())
.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("not support edit image");
}
@Override
public ImageResponse vary(VaryImageRequest request) {
throw new UnsupportedOperationException("not support vary image");
}
}

View File

@@ -0,0 +1,29 @@
/*
* 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.openai;
import com.easyagents.core.model.config.BaseModelConfig;
public class OpenAIImageModelConfig extends BaseModelConfig {
private static final String endpoint = "https://api.openai.com";
private static final String model = "dall-e-3";
public OpenAIImageModelConfig() {
setEndpoint(endpoint);
setModel(model);
}
}

View File

@@ -0,0 +1,38 @@
/*
* 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.ImageResponse;
import com.easyagents.image.openai.OpenAIImageModel;
import com.easyagents.image.openai.OpenAIImageModelConfig;
import org.junit.Test;
public class OpenAIImageModelTest {
@Test
public void testGenImage(){
OpenAIImageModelConfig config = new OpenAIImageModelConfig();
config.setApiKey("sk-5gqOclb****");
OpenAIImageModel imageModel = new OpenAIImageModel(config);
GenerateImageRequest request = new GenerateImageRequest();
request.setPrompt("A cute little tiger standing in the high-speed train");
ImageResponse generate = imageModel.generate(request);
System.out.println(generate);
}
}