summarization

package
v0.1.1 Latest Latest
Warning

This package is not in the latest version of its module.

Go to latest
Published: Mar 6, 2026 License: MIT Imports: 10 Imported by: 0

Documentation

Index

Constants

View Source
const SystemPrompt = `` /* 381-byte string literal not displayed */

SystemPrompt is the default prompt for conversation summarization.

Variables

This section is empty.

Functions

This section is empty.

Types

type Client

type Client struct {
	// contains filtered or unexported fields
}

Client provides LLM completion capabilities via Iris.

func NewClient

func NewClient(cfg *config.Config) *Client

NewClient creates a new summarization client.

func (*Client) Complete

func (c *Client) Complete(ctx context.Context, messages []Message) (string, error)

Complete sends a completion request to Iris and returns the response.

func (*Client) SummarizeMessages

func (c *Client) SummarizeMessages(ctx context.Context, messages []*types.Message) (string, error)

SummarizeMessages summarizes a list of conversation messages.

type CompletionRequest

type CompletionRequest struct {
	Provider  string    `json:"provider"`
	Model     string    `json:"model"`
	Messages  []Message `json:"messages"`
	MaxTokens int       `json:"max_tokens,omitempty"`
}

CompletionRequest is the request body for Iris completions endpoint.

type CompletionResponse

type CompletionResponse struct {
	Content string `json:"content"`
	Model   string `json:"model,omitempty"`
	Usage   *Usage `json:"usage,omitempty"`
}

CompletionResponse is the response from Iris completions endpoint.

type Message

type Message struct {
	Role    string `json:"role"`
	Content string `json:"content"`
}

Message represents a chat message for the LLM.

type Usage

type Usage struct {
	PromptTokens     int `json:"prompt_tokens"`
	CompletionTokens int `json:"completion_tokens"`
	TotalTokens      int `json:"total_tokens"`
}

Usage contains token usage information.

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL