com.github.hakenadu.javalangchains.chains.llm.openai.completions.OpenAiCompletionsParameters Maven / Gradle / Ivy
package com.github.hakenadu.javalangchains.chains.llm.openai.completions;
import java.util.Set;
import com.github.hakenadu.javalangchains.chains.llm.openai.OpenAiParameters;
/**
* Parameters for calling an OpenAI Completions Model
*
* https://platform.openai.com/docs/api-reference/completions/create
*/
public class OpenAiCompletionsParameters extends OpenAiParameters {
/**
* From
* https://github.com/openai/openai-openapi/blob/master/openapi.yaml
*
* Up to 4 sequences where the API will stop generating further tokens. The
* returned text will not contain the stop sequence.
*/
private Set stop;
/**
* Creates an instance of {@link OpenAiCompletionsParameters}
*/
public OpenAiCompletionsParameters() {
super(OpenAiCompletionsParameters.class);
}
/**
* @return {@link #stop}
*/
public Set getStop() {
return stop;
}
/**
* @param stop {@link #stop}
*/
public void setStop(final Set stop) {
this.stop = stop;
}
/**
* @param stop {@link #stop}
* @return this
*/
public OpenAiCompletionsParameters stop(final Set stop) {
setStop(stop);
return this;
}
}