From a06acdf721afbe3ba7ad2bbcad679f4777628f15 Mon Sep 17 00:00:00 2001 From: 0007 <0007@qq.com> Date: Wed, 27 Aug 2025 19:59:14 +0800 Subject: [PATCH] Add File --- .../agentsflex/image/qwen/QwenImageModel.java | 80 +++++++++++++++++++ 1 file changed, 80 insertions(+) create mode 100644 agents-flex-image/agents-flex-image-qwen/src/main/java/com/agentsflex/image/qwen/QwenImageModel.java diff --git a/agents-flex-image/agents-flex-image-qwen/src/main/java/com/agentsflex/image/qwen/QwenImageModel.java b/agents-flex-image/agents-flex-image-qwen/src/main/java/com/agentsflex/image/qwen/QwenImageModel.java new file mode 100644 index 0000000..f841136 --- /dev/null +++ b/agents-flex-image/agents-flex-image-qwen/src/main/java/com/agentsflex/image/qwen/QwenImageModel.java @@ -0,0 +1,80 @@ +/* + * Copyright (c) 2023-2025, Agents-Flex (fuhai999@gmail.com). + *

+ * 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 + *

+ * http://www.apache.org/licenses/LICENSE-2.0 + *

+ * 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.agentsflex.image.qwen; + +import com.agentsflex.core.image.*; +import com.agentsflex.core.llm.client.HttpClient; +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 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."); + } + + +}