integration

Documentation

Overview

Package integration provides utilities for integration with external systems, services, and frameworks.

Index

• func ToOpenAIChatCompletionMessages(messages schema.ChatMessages) ([]openai.ChatCompletionMessage, error)
• type HTTPClient
• type SerpAPI
  • func NewSerpAPI(apiKey string) (*SerpAPI, error)
  • func (s *SerpAPI) Run(ctx context.Context, query string) (string, error)
• type SerpAPIOptions
• type Wikipedia
  • func NewWikipedia(optFns ...func(o *WikipediaOptions)) *Wikipedia
  • func (w *Wikipedia) Run(ctx context.Context, query string) (string, error)
• type WikipediaOptions

Functions

func ToOpenAIChatCompletionMessages

func ToOpenAIChatCompletionMessages(messages schema.ChatMessages) ([]openai.ChatCompletionMessage, error)

ToOpenAIChatCompletionMessages converts a slice of schema.ChatMessage to a slice of openai.ChatCompletionMessage. It extracts the necessary information from each message to create the corresponding OpenAI chat completion message.

Types

type HTTPClient

type HTTPClient interface {
	Do(req *http.Request) (*http.Response, error)
}

type SerpAPI

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

func NewSerpAPI

func NewSerpAPI(apiKey string) (*SerpAPI, error)

func (*SerpAPI) Run

func (s *SerpAPI) Run(ctx context.Context, query string) (string, error)

type SerpAPIOptions

type SerpAPIOptions struct {
	Parameter map[string]string
}

type Wikipedia

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

func NewWikipedia

func NewWikipedia(optFns ...func(o *WikipediaOptions)) *Wikipedia

func (*Wikipedia) Run

func (w *Wikipedia) Run(ctx context.Context, query string) (string, error)

type WikipediaOptions

type WikipediaOptions struct {
	LanguageCode string
	TopK         int
	DocMaxChars  int
	HTTPClient   HTTPClient
}