diff --git a/agents-flex-core/src/main/java/com/agentsflex/core/prompt/ImagePrompt.java b/agents-flex-core/src/main/java/com/agentsflex/core/prompt/ImagePrompt.java new file mode 100644 index 0000000..ed42d2e --- /dev/null +++ b/agents-flex-core/src/main/java/com/agentsflex/core/prompt/ImagePrompt.java @@ -0,0 +1,128 @@ +/* + * 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.core.prompt;
+
+import com.agentsflex.core.message.HumanImageMessage;
+import com.agentsflex.core.message.Message;
+import com.agentsflex.core.util.ImageUtil;
+import com.agentsflex.core.util.StringUtil;
+
+import java.io.File;
+import java.io.InputStream;
+import java.util.Collections;
+import java.util.List;
+
+public class ImagePrompt extends TextPrompt {
+
+ private String imageUrl;
+ private File imageFile;
+ private String imageBase64;
+
+ public ImagePrompt(String content) {
+ super(content);
+ }
+
+ public ImagePrompt(String content, String imageUrl) {
+ super(content);
+ this.imageUrl = imageUrl;
+ }
+
+ public ImagePrompt(String content, File imageFile) {
+ super(content);
+ this.imageFile = imageFile;
+ }
+
+ public ImagePrompt(String content, InputStream imageStream) {
+ super(content);
+ this.imageBase64 = ImageUtil.imageStreamToBase64(imageStream);
+ }
+
+ public String getImageUrl() {
+ return imageUrl;
+ }
+
+ public void setImageUrl(String imageUrl) {
+ this.imageUrl = imageUrl;
+ }
+
+ public File getImageFile() {
+ return imageFile;
+ }
+
+ public void setImageFile(File imageFile) {
+ this.imageFile = imageFile;
+ }
+
+ public String getImageBase64() {
+ return imageBase64;
+ }
+
+ public void setImageBase64(String imageBase64) {
+ this.imageBase64 = imageBase64;
+ }
+
+
+ public String toUrl() {
+ if (StringUtil.hasText(imageUrl)) {
+ return imageUrl;
+ }
+
+ if (imageBase64 != null) {
+ return imageBase64;
+ }
+
+ if (imageFile != null) {
+ imageBase64 = ImageUtil.imageFileToBase64(imageFile);
+ return imageBase64;
+ }
+ return null;
+ }
+
+
+ public String toImageBase64() {
+ if (imageBase64 != null) {
+ return imageBase64;
+ }
+
+ if (StringUtil.hasText(imageUrl)) {
+ imageBase64 = ImageUtil.imageUrlToBase64(imageUrl);
+ return imageBase64;
+ }
+
+ if (imageFile != null) {
+ imageBase64 = ImageUtil.imageFileToBase64(imageFile);
+ return imageBase64;
+ }
+
+ return null;
+ }
+
+ @Override
+ public List