chat

type Command struct {
	// Name of the command.
	//
	// If Matches is nil, the command is executed when the input matches the name.
	Name string

	// Description of the command.
	Description string

	// Matches is a function that checks if the command matches the input.
	//
	// If Matches is nil, the command is executed when the input matches the name.
	// If Matches is not nil, the command is executed when Matches returns true.
	Matches func(input string) bool

	// Run is the function that executes the command.
	Run CommandFunc
}

type CommandFunc func(ctx context.Context, session *Session, input string)

type ReqRespPair struct {
	Model          string                       `json:"model,omitzero"`
	Req            openai.ChatCompletionMessage `json:"req,omitzero"`
	ReqTokens      int64                        `json:"req_tokens,omitzero"`
	Resp           openai.ChatCompletionMessage `json:"resp,omitzero"`
	RespTokens     int64                        `json:"resp_tokens,omitzero"`
	EmbeddingModel string                       `json:"embedding_model,omitzero"`
	Embeddings     [][]float64                  `json:"embedding,omitzero"`
}

type Session struct {
	Client                     *openai.Client
	ChatModel                  string
	EmbeddingModel             string
	StorageBackend             storage.Backend[string, ReqRespPair]
	Messages                   []openai.ChatCompletionMessage
	CurrentTokensUsed          int64
	SummarizeContextWindowSize int64

	Terminal   *term.Terminal
	OutWriter  *bufio.Writer
	TermWidth  int
	TermHeight int
	Commands   []Command
}