OllamaChatRequest.java

Go to the documentation of this file.

package io.github.ollama4j.models.chat;
 
import java.util.List;
 
import io.github.ollama4j.models.request.OllamaCommonRequest;
import io.github.ollama4j.tools.Tools;
import io.github.ollama4j.utils.OllamaRequestBody;
 
import lombok.Getter;
import lombok.Setter;

@Getter
@Setter
public class OllamaChatRequest extends OllamaCommonRequest implements OllamaRequestBody {
 
  private List<OllamaChatMessage> messages;
 
  private List<Tools.PromptFuncDefinition> tools;
 
  public OllamaChatRequest() {}
 
  public OllamaChatRequest(String model, List<OllamaChatMessage> messages) {
    this.model = model;
    this.messages = messages;
  }
 
  @Override
  public boolean equals(Object o) {
    if (!(o instanceof OllamaChatRequest)) {
      return false;
    }
 
    return this.toString().equals(o.toString());
  }
 
}