Package edu.tufts.hrilab.llm

Class Chat

java.lang.Object

edu.tufts.hrilab.llm.Chat

Direct Known Subclasses:

LlamaHFChat

public class Chat
extends Object

Field Summary

Fields

Modifier and Type	Field	Description
String	context
List<Message>	messages
String	model
String	robotName
String	service
String	systemMessage
List<Integer>	tokenCounts
int	tokens
String	userName

Constructor Summary

Constructors

Constructor	Description
Chat()
Chat(Completion a, Completion b)
Chat(Message message)
Chat(Prompt systemPrompt)
Chat(Prompt systemPrompt, List<Message> msgs)
Chat(String systemText)
Chat(String user, String robot)
Chat(String user, String robot, String systemText)
Chat(String systemText, List<Message> msgs)
Chat(List<Message> msgs)

Method Summary

Modifier and Type	Method	Description
void	addCompletion(Completion completion)
void	addCompletion(Completion completion, boolean updateTokens)
void	addContext(String addition)
void	addInteraction(Message a, Message b)
void	addMessage(Message message)
void	addMessage(String role, String content)	Message methods
void	addMessages(List<Message> msgs)
void	addRobotMessage(String content)
void	addUserMessage(String content)
String	getContext()
List<Message>	getMessages()
String	getRobotRole()
String	getSystem()
String	getUserRole()
String	roleWrapper(String role)
void	setContext(Chat chat)
void	setContext(String contextText)
void	setMessages(Chat chat)
void	setMessages(List<Message> msgs)
void	setRobot(String robot)
void	setSystem(Chat chat)
void	setSystem(Prompt systemPrompt)	System + Context methods
void	setSystem(String systemText)
void	setUser(String user)	User methods
List<Message>	toPrompt()	Prompt methods
String	toPromptString()
Message	toSystemMessage()

Methods inherited from class java.lang.Object

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

Field Details

tokens

public int tokens

messages

public List<Message> messages

userName

public String userName

robotName

public String robotName

systemMessage

public String systemMessage

context

public String context

service

public String service

model

public String model

tokenCounts

public List<Integer> tokenCounts

Constructor Details

Chat

public Chat()

Chat

public Chat(String user,
 String robot)

Chat

public Chat(String systemText)

Chat

public Chat(Prompt systemPrompt)

Chat

public Chat(String user,
 String robot,
 String systemText)

Chat

public Chat(List<Message> msgs)

Chat

public Chat(Message message)

Chat

public Chat(String systemText,
 List<Message> msgs)

Chat

public Chat(Prompt systemPrompt,
 List<Message> msgs)

Chat

public Chat(Completion a,
 Completion b)

Method Details

setUser

public void setUser(String user)

User methods

setRobot

public void setRobot(String robot)

getUserRole

public String getUserRole()

getRobotRole

public String getRobotRole()

roleWrapper

public String roleWrapper(String role)

addMessage

public void addMessage(String role,
 String content)

Message methods

addMessage

public void addMessage(Message message)

addMessages

public void addMessages(List<Message> msgs)

setMessages

public void setMessages(List<Message> msgs)

setMessages

public void setMessages(Chat chat)

addUserMessage

public void addUserMessage(String content)

addRobotMessage

public void addRobotMessage(String content)

addInteraction

public void addInteraction(Message a,
 Message b)

addCompletion

public void addCompletion(Completion completion)

addCompletion

public void addCompletion(Completion completion,
 boolean updateTokens)

getMessages

public List<Message> getMessages()

setSystem

public void setSystem(Prompt systemPrompt)

System + Context methods

setSystem

public void setSystem(String systemText)

setSystem

public void setSystem(Chat chat)

getSystem

public String getSystem()

setContext

public void setContext(String contextText)

setContext

public void setContext(Chat chat)

addContext

public void addContext(String addition)

getContext

public String getContext()

toSystemMessage

public Message toSystemMessage()

toPrompt

public List<Message> toPrompt()

Prompt methods

toPromptString

public String toPromptString()