NeuroDock/src/main/java/me/zacharias/chat/ollama/OllamaToolRespnce.java

package me.zacharias.chat.ollama;

import org.json.JSONObject;

public class OllamaToolRespnce extends OllamaMessage {
    private final String tool;
    private final String response;
    public OllamaToolRespnce(String tool, String response) {
        super(OllamaMessageRole.TOOL, new JSONObject().put("tool", tool).put("result", response).toString());
        this.tool = tool;
        this.response = response;
    }

    public String getTool() {
        return tool;
    }

    public String getResponse() {
        return response;
    }
}