ai

Documentation

Index

• Constants
• func EvalFromExperiment(exp *agentv1.Experiment, experimentFilePath string, cookie map[string]string, ...) (map[string]*parserv1.Cell, error)
• func NewClient(cfg config.OpenAIConfig) (*openai.Client, error)
• func NewClientWithKey(key string) (*openai.Client, error)
• func NewClientWithoutKey() *openai.Client
• func NewLlmJudge(client *openai.Client) *llmJudge
• type Agent
  • func NewAgent(opts AgentOptions) (*Agent, error)
  • func (a *Agent) BuildResponseParams(ctx context.Context, req *agentv1.GenerateRequest) (*responses.ResponseNewParams, []option.RequestOption, error)
  • func (a *Agent) Generate(ctx context.Context, req *connect.Request[agentv1.GenerateRequest], ...) error
• type AgentOptions
  • func (o *AgentOptions) FromAssistantConfig(cfg config.CloudAssistantConfig) error
• type Asserter

Constants

const (
	// DefaultInstructions is the default system prompt to use when generating responses
	DefaultInstructions = `` /* 963-byte string literal not displayed */

)

Functions

func EvalFromExperiment

func EvalFromExperiment(exp *agentv1.Experiment, experimentFilePath string, cookie map[string]string, client *openai.Client, log logr.Logger) (map[string]*parserv1.Cell, error)

EvalFromExperiment runs an experiment based on the Experiment config.

func NewClient

func NewClient(cfg config.OpenAIConfig) (*openai.Client, error)

NewClient helper function to create a new OpenAI client from a config

func NewClientWithKey

func NewClientWithKey(key string) (*openai.Client, error)

func NewClientWithoutKey

func NewClientWithoutKey() *openai.Client

NewClientWithoutKey returns an OpenAI client configured without an API key. This is intended for OAuth flows where the caller supplies the Authorization header per-request.

func NewLlmJudge

func NewLlmJudge(client *openai.Client) *llmJudge

Types

type Agent

type Agent struct {
	Client *openai.Client
	// contains filtered or unexported fields
}

Agent implements the AI Service https://buf.build/jlewi/foyle/file/main:foyle/v1alpha1/agent.proto#L44

func NewAgent

func NewAgent(opts AgentOptions) (*Agent, error)

func (*Agent) BuildResponseParams

func (a *Agent) BuildResponseParams(ctx context.Context, req *agentv1.GenerateRequest) (*responses.ResponseNewParams, []option.RequestOption, error)

func (*Agent) Generate

func (a *Agent) Generate(ctx context.Context, req *connect.Request[agentv1.GenerateRequest], resp *connect.ServerStream[agentv1.GenerateResponse]) error

type AgentOptions

type AgentOptions struct {
	VectorStores []string
	Client       *openai.Client

	// Instructions are the prompt to use when generating responses
	Instructions string

	// Extra OpenAI API headers for requests that use OAuth access tokens.
	OAuthOpenAIOrganization string
	OAuthOpenAIProject      string
}

AgentOptions are options for creating a new Agent

func (*AgentOptions) FromAssistantConfig

func (o *AgentOptions) FromAssistantConfig(cfg config.CloudAssistantConfig) error

FromAssistantConfig overrides the AgentOptions based on the values from the AssistantConfig

type Asserter

type Asserter interface {
	Assert(ctx context.Context, as *agentv1.Assertion, inputText string, cells map[string]*parserv1.Cell) error
}