model

package

v0.3.2 Latest Latest Go to latest Published: Dec 26, 2024 License: Apache-2.0 Imports: 5 Imported by: 0

Details

Package model is a generated GoMock package.

This section is empty.

This section is empty.

This section is empty.

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

MockChatModel is a mock of ChatModel interface.

func NewMockChatModel(ctrl *gomock.Controller) *MockChatModel

NewMockChatModel creates a new mock instance.

func (m *MockChatModel) BindTools(tools []*schema.ToolInfo) error

BindTools mocks base method.

func (m *MockChatModel) EXPECT() *MockChatModelMockRecorder

EXPECT returns an object that allows the caller to indicate expected use.

func (m *MockChatModel) Generate(ctx context.Context, input []*schema.Message, opts ...model.Option) (*schema.Message, error)

Generate mocks base method.

func (m *MockChatModel) Stream(ctx context.Context, input []*schema.Message, opts ...model.Option) (*schema.StreamReader[*schema.Message], error)

Stream mocks base method.

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

MockChatModelMockRecorder is the mock recorder for MockChatModel.

func (mr *MockChatModelMockRecorder) BindTools(tools any) *gomock.Call

BindTools indicates an expected call of BindTools.

func (mr *MockChatModelMockRecorder) Generate(ctx, input any, opts ...any) *gomock.Call

Generate indicates an expected call of Generate.

func (mr *MockChatModelMockRecorder) Stream(ctx, input any, opts ...any) *gomock.Call

Stream indicates an expected call of Stream.