xai

Documentation

Overview

Package xai provides a xai LLM provider for langrails.

It uses an OpenAI-compatible API and is a thin wrapper around the compat package.

Index

• type Option
  • func WithBaseURL(url string) Option
  • func WithHTTPClient(client *http.Client) Option
• type Provider
  • func New(apiKey string, opts ...Option) *Provider
  • func (p *Provider) Complete(ctx context.Context, req *langrails.CompletionRequest) (*langrails.CompletionResponse, error)
  • func (p *Provider) Stream(ctx context.Context, req *langrails.CompletionRequest) (<-chan langrails.StreamEvent, error)

Types

type Option

type Option func(*compat.Config)

Option configures the provider.

func WithBaseURL

func WithBaseURL(url string) Option

WithBaseURL sets a custom base URL.

func WithHTTPClient

func WithHTTPClient(client *http.Client) Option

WithHTTPClient sets a custom HTTP client.

type Provider

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

Provider implements langrails.Provider for xAI's API.

func New

func New(apiKey string, opts ...Option) *Provider

New creates a new xAI provider.

func (*Provider) Complete

func (p *Provider) Complete(ctx context.Context, req *langrails.CompletionRequest) (*langrails.CompletionResponse, error)

func (*Provider) Stream

func (p *Provider) Stream(ctx context.Context, req *langrails.CompletionRequest) (<-chan langrails.StreamEvent, error)