openai

Documentation

Index

• type Client
  • func New(apiKey, model string) *Client
  • func NewWithOptions(model string, opts ...option.RequestOption) *Client
  • func (c *Client) CountTokens(ctx context.Context, req agent.MessagesRequest) (int, error)
  • func (c *Client) CreateMessages(ctx context.Context, req agent.MessagesRequest, ...) (agent.MessagesResponse, error)
  • func (c *Client) SetModel(model string)

Types

type Client

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

Client wraps the OpenAI client to implement agent.LLMClient

func New

func New(apiKey, model string) *Client

New creates a new Client

func NewWithOptions

func NewWithOptions(model string, opts ...option.RequestOption) *Client

NewWithOptions creates a new Client with custom options

func (*Client) CountTokens

func (c *Client) CountTokens(ctx context.Context, req agent.MessagesRequest) (int, error)

CountTokens estimates the number of tokens in a request. OpenAI doesn't provide a direct token counting API, so this is a rough estimation. A more accurate implementation would use tiktoken or similar tokenizer.

func (*Client) CreateMessages

func (c *Client) CreateMessages(ctx context.Context, req agent.MessagesRequest, onUpdate func(agent.MessagesResponse)) (agent.MessagesResponse, error)

func (*Client) SetModel

func (c *Client) SetModel(model string)

SetModel sets the model for the client