001/*
002 *  Copyright (c) 2023-2026, Agents-Flex (fuhai999@gmail.com).
003 *  <p>
004 *  Licensed under the Apache License, Version 2.0 (the "License");
005 *  you may not use this file except in compliance with the License.
006 *  You may obtain a copy of the License at
007 *  <p>
008 *  http://www.apache.org/licenses/LICENSE-2.0
009 *  <p>
010 *  Unless required by applicable law or agreed to in writing, software
011 *  distributed under the License is distributed on an "AS IS" BASIS,
012 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 *  See the License for the specific language governing permissions and
014 *  limitations under the License.
015 */
016package com.agentsflex.core.prompt;
017
018import com.agentsflex.core.message.*;
019
020import java.io.File;
021import java.util.ArrayList;
022import java.util.List;
023
024
025public class SimplePrompt extends Prompt {
026
027    protected SystemMessage systemMessage;
028    protected UserMessage userMessage;
029    protected AiMessage aiMessage;
030    protected List<ToolMessage> toolMessages;
031
032    public SimplePrompt() {
033        this.userMessage = new UserMessage();
034    }
035
036    public SimplePrompt(String content) {
037        this.userMessage = new UserMessage(content);
038    }
039
040
041    /// /// Audio
042    public List<String> getAudioUrls() {
043        return userMessage.getAudioUrls();
044    }
045
046    public void setAudioUrls(List<String> audioUrls) {
047        userMessage.setAudioUrls(audioUrls);
048    }
049
050    public void addAudioUrl(String audioUrl) {
051        userMessage.addAudioUrl(audioUrl);
052    }
053
054
055    /// ///  Video
056    public List<String> getVideoUrls() {
057        return userMessage.getVideoUrls();
058    }
059
060    public void setVideoUrls(List<String> videoUrls) {
061        userMessage.setVideoUrls(videoUrls);
062    }
063
064    public void addVideoUrl(String videoUrl) {
065        userMessage.addVideoUrl(videoUrl);
066    }
067
068
069    /// /// Images
070    public List<String> getImageUrls() {
071        return userMessage.getImageUrls();
072    }
073
074    public void setImageUrls(List<String> imageUrls) {
075        userMessage.setImageUrls(imageUrls);
076    }
077
078    public void addImageUrl(String imageUrl) {
079        userMessage.addImageUrl(imageUrl);
080    }
081
082    public void addImageFile(File imageFile) {
083        userMessage.addImageFile(imageFile);
084    }
085
086    public void addImageBytes(byte[] imageBytes, String mimeType) {
087        userMessage.addImageBytes(imageBytes, mimeType);
088    }
089
090
091    /// ////getter setter
092    public SystemMessage getSystemMessage() {
093        return systemMessage;
094    }
095
096    public void setSystemMessage(SystemMessage systemMessage) {
097        this.systemMessage = systemMessage;
098    }
099
100    public UserMessage getUserMessage() {
101        return userMessage;
102    }
103
104    public void setUserMessage(UserMessage userMessage) {
105        this.userMessage = userMessage;
106    }
107
108    public AiMessage getAiMessage() {
109        return aiMessage;
110    }
111
112    public void setAiMessage(AiMessage aiMessage) {
113        this.aiMessage = aiMessage;
114    }
115
116    public List<ToolMessage> getToolMessages() {
117        return toolMessages;
118    }
119
120    public void setToolMessages(List<ToolMessage> toolMessages) {
121        this.toolMessages = toolMessages;
122    }
123
124    @Override
125    public List<Message> getMessages() {
126        List<Message> messages = new ArrayList<>(2);
127        if (systemMessage != null) {
128            messages.add(systemMessage);
129        }
130        messages.add(userMessage);
131
132        if (aiMessage != null) {
133            messages.add(aiMessage);
134        }
135
136        if (toolMessages != null) {
137            messages.addAll(toolMessages);
138        }
139        return messages;
140    }
141}