openai

Documentation

Index

• func IsOpenAiEngine(engine string) bool
• type ExecuteToolStep
  • func (e ExecuteToolStep) Start(ctx context.Context, input ToolCompletionResponse) (steps.StepResult[map[string]interface{}], error)
• type Step
  • func NewStep(settings *settings.StepSettings) *Step
  • func (csf *Step) Interrupt()
  • func (csf *Step) Publish(publisher message.Publisher, topic string) error
  • func (csf *Step) Start(ctx context.Context, messages []*geppetto_context.Message) (steps.StepResult[string], error)
• type ToolCallMerger
  • func NewToolCallMerger() *ToolCallMerger
  • func (tcm *ToolCallMerger) AddToolCalls(toolCalls []go_openai.ToolCall)
  • func (tcm *ToolCallMerger) GetToolCalls() []go_openai.ToolCall
• type ToolCompletionResponse
• type ToolStep
  • func (csf *ToolStep) SetStreaming(b bool)
  • func (csf *ToolStep) Start(ctx context.Context, messages []*geppetto_context.Message) (steps.StepResult[ToolCompletionResponse], error)
• type Transcription
• type TranscriptionClient
  • func NewTranscriptionClient(apiKey, model, prompt, language string, temperature float32) *TranscriptionClient
  • func (tc *TranscriptionClient) TranscribeFile(mp3FilePath string, out chan<- Transcription, wg *sync.WaitGroup)

Functions

func IsOpenAiEngine

func IsOpenAiEngine(engine string) bool

Types

type ExecuteToolStep

type ExecuteToolStep struct {
	Tools map[string]interface{}
}

func (ExecuteToolStep) Start

func (e ExecuteToolStep) Start(
	ctx context.Context,
	input ToolCompletionResponse,
) (steps.StepResult[map[string]interface{}], error)

type Step

type Step struct {
	Settings *settings.StepSettings
	// contains filtered or unexported fields
}

func NewStep

func NewStep(settings *settings.StepSettings) *Step

func (*Step) Interrupt

func (csf *Step) Interrupt()

func (*Step) Publish

func (csf *Step) Publish(publisher message.Publisher, topic string) error

func (*Step) Start

func (csf *Step) Start(
	ctx context.Context,
	messages []*geppetto_context.Message,
) (steps.StepResult[string], error)

type ToolCallMerger

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

func NewToolCallMerger

func NewToolCallMerger() *ToolCallMerger

func (*ToolCallMerger) AddToolCalls

func (tcm *ToolCallMerger) AddToolCalls(toolCalls []go_openai.ToolCall)

func (*ToolCallMerger) GetToolCalls

func (tcm *ToolCallMerger) GetToolCalls() []go_openai.ToolCall

type ToolCompletionResponse

type ToolCompletionResponse struct {
	Role      string               `json:"role"`
	Content   string               `json:"content"`
	ToolCalls []go_openai.ToolCall `json:"tool_calls"`
}

type ToolStep

type ToolStep struct {
	Settings *settings.StepSettings
	Tools    []go_openai.Tool
}

func (*ToolStep) SetStreaming

func (csf *ToolStep) SetStreaming(b bool)

func (*ToolStep) Start

func (csf *ToolStep) Start(
	ctx context.Context,
	messages []*geppetto_context.Message,
) (steps.StepResult[ToolCompletionResponse], error)

type Transcription

type Transcription struct {
	File     string                `json:"file"`
	Response *openai.AudioResponse `json:"response"`
	// contains filtered or unexported fields
}

type TranscriptionClient

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

func NewTranscriptionClient

func NewTranscriptionClient(apiKey, model, prompt, language string, temperature float32) *TranscriptionClient

func (*TranscriptionClient) TranscribeFile

func (tc *TranscriptionClient) TranscribeFile(mp3FilePath string, out chan<- Transcription, wg *sync.WaitGroup)