Documentation
¶
Index ¶
- func NewMockTool(name, description string) ToolDefinitionInterface
- func SetRunError(toolDef ToolDefinitionInterface, err error)
- func SetRunResult(toolDef ToolDefinitionInterface, result string)
- type MockOpenAIClient
- func (m *MockOpenAIClient) AddCreateChatCompletionFunction(name, args string)
- func (m *MockOpenAIClient) Ask(ctx context.Context, f Fragment) (Fragment, error)
- func (m *MockOpenAIClient) CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error)
- func (m *MockOpenAIClient) SetAskError(err error)
- func (m *MockOpenAIClient) SetAskResponse(content string)
- func (m *MockOpenAIClient) SetCreateChatCompletionError(err error)
- func (m *MockOpenAIClient) SetCreateChatCompletionResponse(response openai.ChatCompletionResponse)
- type MockTool
Constants ¶
This section is empty.
Variables ¶
This section is empty.
Functions ¶
func NewMockTool ¶
func NewMockTool(name, description string) ToolDefinitionInterface
func SetRunError ¶ added in v0.5.0
func SetRunError(toolDef ToolDefinitionInterface, err error)
SetRunError sets an error for a mock tool within a ToolDef
func SetRunResult ¶ added in v0.5.0
func SetRunResult(toolDef ToolDefinitionInterface, result string)
SetRunResult sets the result for a mock tool within a ToolDef
Types ¶
type MockOpenAIClient ¶
type MockOpenAIClient struct {
AskResponses []Fragment
AskResponseIndex int
CreateChatCompletionResponses []openai.ChatCompletionResponse
CreateChatCompletionIndex int
AskError error
CreateChatCompletionError error
FragmentHistory []Fragment
}
MockOpenAIClient implements the OpenAIClient for testing
func NewMockOpenAIClient ¶
func NewMockOpenAIClient() *MockOpenAIClient
func (*MockOpenAIClient) AddCreateChatCompletionFunction ¶
func (m *MockOpenAIClient) AddCreateChatCompletionFunction(name, args string)
func (*MockOpenAIClient) Ask ¶
func (m *MockOpenAIClient) Ask(ctx context.Context, f Fragment) (Fragment, error)
func (*MockOpenAIClient) CreateChatCompletion ¶
func (m *MockOpenAIClient) CreateChatCompletion(ctx context.Context, request openai.ChatCompletionRequest) (openai.ChatCompletionResponse, error)
func (*MockOpenAIClient) SetAskError ¶
func (m *MockOpenAIClient) SetAskError(err error)
func (*MockOpenAIClient) SetAskResponse ¶
func (m *MockOpenAIClient) SetAskResponse(content string)
Helper methods for setting up mock responses
func (*MockOpenAIClient) SetCreateChatCompletionError ¶
func (m *MockOpenAIClient) SetCreateChatCompletionError(err error)
func (*MockOpenAIClient) SetCreateChatCompletionResponse ¶
func (m *MockOpenAIClient) SetCreateChatCompletionResponse(response openai.ChatCompletionResponse)
type MockTool ¶
type MockTool struct {
// contains filtered or unexported fields
}
MockTool implements the Tool interface for testing
func GetMockTool ¶ added in v0.5.0
func GetMockTool(toolDef ToolDefinitionInterface) *MockTool
GetMockTool extracts the MockTool from a ToolDef (if it contains one)
func (*MockTool) SetRunError ¶
func (*MockTool) SetRunResult ¶
Click to show internal directories.
Click to hide internal directories.