Class Chat

java.lang.Object
edu.tufts.hrilab.llm.Chat
Direct Known Subclasses:
LlamaHFChat

public class Chat extends Object
  • 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
  • Constructor Details

  • 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)
    • addRobotMessage

      public void addRobotMessage(String content, List<ToolCall> toolCalls)
    • addToolMessage

      public void addToolMessage(String content, String tool_call_id)
    • addToolVisionMessage

      public void addToolVisionMessage(String text, String imageUrl, String tool_call_id)
    • addUserVisionMessage

      public void addUserVisionMessage(String text, String imageUrl)
    • 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 TextMessage toSystemMessage()
    • toPrompt

      public List<Message> toPrompt()
      Prompt methods
    • toPromptString

      public String toPromptString()