Package edu.tufts.hrilab.llm.llama.request

Class LlamaCompletionRequestBody

java.lang.Object

edu.tufts.hrilab.llm.llama.request.LlamaCompletionRequestBody

public class LlamaCompletionRequestBody
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
String	prompt
float	repeat_penalty
String[]	stop
boolean	stream
float	temperature
int	top_k
float	top_p

Constructor Summary

Constructors

Constructor	Description
LlamaCompletionRequestBody()
LlamaCompletionRequestBody(Chat chat)
LlamaCompletionRequestBody(String p)

Method Summary

Modifier and Type	Method	Description
void	addStopWords(String[] stopWords)
void	setTemperature(float temperatureFloat)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Field Details

prompt

public String prompt

temperature

public float temperature

stream

public boolean stream

top_k

public int top_k

top_p

public float top_p

repeat_penalty

public float repeat_penalty

stop

public String[] stop

Constructor Details

LlamaCompletionRequestBody

public LlamaCompletionRequestBody()

LlamaCompletionRequestBody

public LlamaCompletionRequestBody(String p)

LlamaCompletionRequestBody

public LlamaCompletionRequestBody(Chat chat)

Method Details

addStopWords

public void addStopWords(String[] stopWords)

setTemperature

public void setTemperature(float temperatureFloat)