mock_types

Documentation

Overview

Package mock_types is a generated GoMock package.

Index

• type MockTGenerator
  • func NewMockTGenerator(ctrl *gomock.Controller) *MockTGenerator
  • func (m *MockTGenerator) AddEmbeddingDocument(ctx context.Context, id string, embedding [1536]float32, ...) error
  • func (m *MockTGenerator) CountTokens(ctx context.Context, content string) (int, error)
  • func (m *MockTGenerator) DeleteEmbeddingDocument(ctx context.Context, id string) error
  • func (m *MockTGenerator) DeleteEmbeddingDocuments(ctx context.Context, ids []string) error
  • func (m *MockTGenerator) EXPECT() *MockTGeneratorMockRecorder
  • func (m *MockTGenerator) FetchEmbedding(ctx context.Context, content ...string) ([][1536]float32, error)
  • func (m *MockTGenerator) GenerateChat(ctx context.Context, messages []types.Message, stream bool, functions string) (types.CompletionMessage, error)
  • func (m *MockTGenerator) GetEmbeddingDocument(ctx context.Context, id string) (types.Vector, bool, error)
  • func (m *MockTGenerator) ListAllEmbeddingsIds(ctx context.Context) (types.SearchResults, error)
  • func (m *MockTGenerator) OffsetTokens(ctx context.Context, content string, from, to int) (string, int, error)
  • func (m *MockTGenerator) RawTokens(ctx context.Context, content string) ([]string, error)
  • func (m *MockTGenerator) SearchWithEmbedding(ctx context.Context, embedding types.QueryFilter, k int) (types.SearchResults, error)
  • func (m *MockTGenerator) SpeechToText(ctx context.Context, audioContent *[]byte, language string) (string, error)
  • func (m *MockTGenerator) TextToSpeech(ctx context.Context, content, language, voice string) (*[]byte, error)
• type MockTGeneratorMockRecorder
  • func (mr *MockTGeneratorMockRecorder) AddEmbeddingDocument(ctx, id, embedding, metadata interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) CountTokens(ctx, content interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) DeleteEmbeddingDocument(ctx, id interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) DeleteEmbeddingDocuments(ctx, ids interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) FetchEmbedding(ctx interface{}, content ...interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) GenerateChat(ctx, messages, stream, functions interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) GetEmbeddingDocument(ctx, id interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) ListAllEmbeddingsIds(ctx interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) OffsetTokens(ctx, content, from, to interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) RawTokens(ctx, content interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) SearchWithEmbedding(ctx, embedding, k interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) SpeechToText(ctx, audioContent, language interface{}) *gomock.Call
  • func (mr *MockTGeneratorMockRecorder) TextToSpeech(ctx, content, language, voice interface{}) *gomock.Call

Types

type MockTGenerator

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

MockTGenerator is a mock of TGenerator interface.

func NewMockTGenerator

func NewMockTGenerator(ctrl *gomock.Controller) *MockTGenerator

NewMockTGenerator creates a new mock instance.

func (*MockTGenerator) AddEmbeddingDocument

func (m *MockTGenerator) AddEmbeddingDocument(ctx context.Context, id string, embedding [1536]float32, metadata types.Metadata) error

AddEmbeddingDocument mocks base method.

func (*MockTGenerator) CountTokens

func (m *MockTGenerator) CountTokens(ctx context.Context, content string) (int, error)

CountTokens mocks base method.

func (*MockTGenerator) DeleteEmbeddingDocument

func (m *MockTGenerator) DeleteEmbeddingDocument(ctx context.Context, id string) error

DeleteEmbeddingDocument mocks base method.

func (*MockTGenerator) DeleteEmbeddingDocuments

func (m *MockTGenerator) DeleteEmbeddingDocuments(ctx context.Context, ids []string) error

DeleteEmbeddingDocuments mocks base method.

func (*MockTGenerator) EXPECT

func (m *MockTGenerator) EXPECT() *MockTGeneratorMockRecorder

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

func (*MockTGenerator) FetchEmbedding

func (m *MockTGenerator) FetchEmbedding(ctx context.Context, content ...string) ([][1536]float32, error)

FetchEmbedding mocks base method.

func (*MockTGenerator) GenerateChat

func (m *MockTGenerator) GenerateChat(ctx context.Context, messages []types.Message, stream bool, functions string) (types.CompletionMessage, error)

GenerateChat mocks base method.

func (*MockTGenerator) GetEmbeddingDocument

func (m *MockTGenerator) GetEmbeddingDocument(ctx context.Context, id string) (types.Vector, bool, error)

GetEmbeddingDocument mocks base method.

func (*MockTGenerator) ListAllEmbeddingsIds

func (m *MockTGenerator) ListAllEmbeddingsIds(ctx context.Context) (types.SearchResults, error)

ListAllEmbeddingsIds mocks base method.

func (*MockTGenerator) OffsetTokens

func (m *MockTGenerator) OffsetTokens(ctx context.Context, content string, from, to int) (string, int, error)

OffsetTokens mocks base method.

func (*MockTGenerator) RawTokens

func (m *MockTGenerator) RawTokens(ctx context.Context, content string) ([]string, error)

RawTokens mocks base method.

func (*MockTGenerator) SearchWithEmbedding

func (m *MockTGenerator) SearchWithEmbedding(ctx context.Context, embedding types.QueryFilter, k int) (types.SearchResults, error)

SearchWithEmbedding mocks base method.

func (*MockTGenerator) SpeechToText

func (m *MockTGenerator) SpeechToText(ctx context.Context, audioContent *[]byte, language string) (string, error)

SpeechToText mocks base method.

func (*MockTGenerator) TextToSpeech

func (m *MockTGenerator) TextToSpeech(ctx context.Context, content, language, voice string) (*[]byte, error)

TextToSpeech mocks base method.

type MockTGeneratorMockRecorder

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

MockTGeneratorMockRecorder is the mock recorder for MockTGenerator.

func (*MockTGeneratorMockRecorder) AddEmbeddingDocument

func (mr *MockTGeneratorMockRecorder) AddEmbeddingDocument(ctx, id, embedding, metadata interface{}) *gomock.Call

AddEmbeddingDocument indicates an expected call of AddEmbeddingDocument.

func (*MockTGeneratorMockRecorder) CountTokens

func (mr *MockTGeneratorMockRecorder) CountTokens(ctx, content interface{}) *gomock.Call

CountTokens indicates an expected call of CountTokens.

func (*MockTGeneratorMockRecorder) DeleteEmbeddingDocument

func (mr *MockTGeneratorMockRecorder) DeleteEmbeddingDocument(ctx, id interface{}) *gomock.Call

DeleteEmbeddingDocument indicates an expected call of DeleteEmbeddingDocument.

func (*MockTGeneratorMockRecorder) DeleteEmbeddingDocuments

func (mr *MockTGeneratorMockRecorder) DeleteEmbeddingDocuments(ctx, ids interface{}) *gomock.Call

DeleteEmbeddingDocuments indicates an expected call of DeleteEmbeddingDocuments.

func (*MockTGeneratorMockRecorder) FetchEmbedding

func (mr *MockTGeneratorMockRecorder) FetchEmbedding(ctx interface{}, content ...interface{}) *gomock.Call

FetchEmbedding indicates an expected call of FetchEmbedding.

func (*MockTGeneratorMockRecorder) GenerateChat

func (mr *MockTGeneratorMockRecorder) GenerateChat(ctx, messages, stream, functions interface{}) *gomock.Call

GenerateChat indicates an expected call of GenerateChat.

func (*MockTGeneratorMockRecorder) GetEmbeddingDocument

func (mr *MockTGeneratorMockRecorder) GetEmbeddingDocument(ctx, id interface{}) *gomock.Call

GetEmbeddingDocument indicates an expected call of GetEmbeddingDocument.

func (*MockTGeneratorMockRecorder) ListAllEmbeddingsIds

func (mr *MockTGeneratorMockRecorder) ListAllEmbeddingsIds(ctx interface{}) *gomock.Call

ListAllEmbeddingsIds indicates an expected call of ListAllEmbeddingsIds.

func (*MockTGeneratorMockRecorder) OffsetTokens

func (mr *MockTGeneratorMockRecorder) OffsetTokens(ctx, content, from, to interface{}) *gomock.Call

OffsetTokens indicates an expected call of OffsetTokens.

func (*MockTGeneratorMockRecorder) RawTokens

func (mr *MockTGeneratorMockRecorder) RawTokens(ctx, content interface{}) *gomock.Call

RawTokens indicates an expected call of RawTokens.

func (*MockTGeneratorMockRecorder) SearchWithEmbedding

func (mr *MockTGeneratorMockRecorder) SearchWithEmbedding(ctx, embedding, k interface{}) *gomock.Call

SearchWithEmbedding indicates an expected call of SearchWithEmbedding.

func (*MockTGeneratorMockRecorder) SpeechToText

func (mr *MockTGeneratorMockRecorder) SpeechToText(ctx, audioContent, language interface{}) *gomock.Call

SpeechToText indicates an expected call of SpeechToText.

func (*MockTGeneratorMockRecorder) TextToSpeech

func (mr *MockTGeneratorMockRecorder) TextToSpeech(ctx, content, language, voice interface{}) *gomock.Call

TextToSpeech indicates an expected call of TextToSpeech.