mocks

package
v0.2.18 Latest Latest
Warning

This package is not in the latest version of its module.

 Go to latest Published: Mar 10, 2025 License: MIT Imports: 7 Imported by: 0

Documentation

Overview

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Package mocks is a generated GoMock package.

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

This section is empty.

Types

type MockClient 

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

MockClient is a mock of Client interface.

func NewMockClient 

func NewMockClient(ctrl *gomock.Controller) *MockClient

NewMockClient creates a new mock instance.

func (*MockClient) Do 

func (m *MockClient) Do(req *http.Request) (*http.Response, error)

Do mocks base method.

func (*MockClient) EXPECT 

func (m *MockClient) EXPECT() *MockClientMockRecorder

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

func (*MockClient) Get 

func (m *MockClient) Get(url string) (*http.Response, error)

Get mocks base method.

func (*MockClient) Post 

func (m *MockClient) Post(url, bodyType, body string) (*http.Response, error)

Post mocks base method.

type MockClientMockRecorder 

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

MockClientMockRecorder is the mock recorder for MockClient.

func (*MockClientMockRecorder) Do 

func (mr *MockClientMockRecorder) Do(req any) *gomock.Call

Do indicates an expected call of Do.

func (*MockClientMockRecorder) Get 

func (mr *MockClientMockRecorder) Get(url any) *gomock.Call

Get indicates an expected call of Get.

func (*MockClientMockRecorder) Post 

func (mr *MockClientMockRecorder) Post(url, bodyType, body any) *gomock.Call

Post indicates an expected call of Post.

type MockLogger 

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

MockLogger is a mock of Logger interface.

func NewMockLogger 

func NewMockLogger(ctrl *gomock.Controller) *MockLogger

NewMockLogger creates a new mock instance.

func (*MockLogger) Debug 

func (m *MockLogger) Debug(message string, fields ...any)

Debug mocks base method.

func (*MockLogger) EXPECT 

func (m *MockLogger) EXPECT() *MockLoggerMockRecorder

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

func (*MockLogger) Error 

func (m *MockLogger) Error(message string, err error, fields ...any)

Error mocks base method.

func (*MockLogger) Fatal 

func (m *MockLogger) Fatal(message string, err error, fields ...any)

Fatal mocks base method.

func (*MockLogger) Info 

func (m *MockLogger) Info(message string, fields ...any)

Info mocks base method.

type MockLoggerMockRecorder 

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

MockLoggerMockRecorder is the mock recorder for MockLogger.

func (*MockLoggerMockRecorder) Debug 

func (mr *MockLoggerMockRecorder) Debug(message any, fields ...any) *gomock.Call

Debug indicates an expected call of Debug.

func (*MockLoggerMockRecorder) Error 

func (mr *MockLoggerMockRecorder) Error(message, err any, fields ...any) *gomock.Call

Error indicates an expected call of Error.

func (*MockLoggerMockRecorder) Fatal 

func (mr *MockLoggerMockRecorder) Fatal(message, err any, fields ...any) *gomock.Call

Fatal indicates an expected call of Fatal.

func (*MockLoggerMockRecorder) Info 

func (mr *MockLoggerMockRecorder) Info(message any, fields ...any) *gomock.Call

Info indicates an expected call of Info.

type MockOpenTelemetry 

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

MockOpenTelemetry is a mock of OpenTelemetry interface.

func NewMockOpenTelemetry 

func NewMockOpenTelemetry(ctrl *gomock.Controller) *MockOpenTelemetry

NewMockOpenTelemetry creates a new mock instance.

func (*MockOpenTelemetry) EXPECT 

func (m *MockOpenTelemetry) EXPECT() *MockOpenTelemetryMockRecorder

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

func (*MockOpenTelemetry) Init 

func (m *MockOpenTelemetry) Init(config config.Config) error

Init mocks base method.

func (*MockOpenTelemetry) RecordTokenUsage added in v0.2.15 

func (m *MockOpenTelemetry) RecordTokenUsage(ctx context.Context, provider, model string, promptTokens, completionTokens, totalTokens int64)

RecordTokenUsage mocks base method.

func (*MockOpenTelemetry) RecordTokenUsageWithTime added in v0.2.15 

func (m *MockOpenTelemetry) RecordTokenUsageWithTime(ctx context.Context, provider, model string, promptTokens, completionTokens, totalTokens int64, queueTime, promptTime, completionTime, totalTime float64)

RecordTokenUsageWithTime mocks base method.

type MockOpenTelemetryMockRecorder 

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

MockOpenTelemetryMockRecorder is the mock recorder for MockOpenTelemetry.

func (*MockOpenTelemetryMockRecorder) Init 

func (mr *MockOpenTelemetryMockRecorder) Init(config any) *gomock.Call

Init indicates an expected call of Init.

func (*MockOpenTelemetryMockRecorder) RecordTokenUsage added in v0.2.15 

func (mr *MockOpenTelemetryMockRecorder) RecordTokenUsage(ctx, provider, model, promptTokens, completionTokens, totalTokens any) *gomock.Call

RecordTokenUsage indicates an expected call of RecordTokenUsage.

func (*MockOpenTelemetryMockRecorder) RecordTokenUsageWithTime added in v0.2.15 

func (mr *MockOpenTelemetryMockRecorder) RecordTokenUsageWithTime(ctx, provider, model, promptTokens, completionTokens, totalTokens, queueTime, promptTime, completionTime, totalTime any) *gomock.Call

RecordTokenUsageWithTime indicates an expected call of RecordTokenUsageWithTime.

type MockProvider 

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

MockProvider is a mock of Provider interface.

func NewMockProvider 

func NewMockProvider(ctrl *gomock.Controller) *MockProvider

NewMockProvider creates a new mock instance.

func (*MockProvider) EXPECT 

func (m *MockProvider) EXPECT() *MockProviderMockRecorder

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

func (*MockProvider) GenerateTokens 

func (m *MockProvider) GenerateTokens(ctx context.Context, model string, messages []providers.Message, tools []providers.Tool, maxTokens int) (providers.GenerateResponse, error)

GenerateTokens mocks base method.

func (*MockProvider) GetAuthType 

func (m *MockProvider) GetAuthType() string

GetAuthType mocks base method.

func (*MockProvider) GetExtraHeaders 

func (m *MockProvider) GetExtraHeaders() map[string][]string

GetExtraHeaders mocks base method.

func (*MockProvider) GetID 

func (m *MockProvider) GetID() string

GetID mocks base method.

func (*MockProvider) GetName 

func (m *MockProvider) GetName() string

GetName mocks base method.

func (*MockProvider) GetToken 

func (m *MockProvider) GetToken() string

GetToken mocks base method.

func (*MockProvider) GetURL 

func (m *MockProvider) GetURL() string

GetURL mocks base method.

func (*MockProvider) ListModels 

func (m *MockProvider) ListModels(ctx context.Context) (providers.ListModelsResponse, error)

ListModels mocks base method.

func (*MockProvider) StreamTokens added in v0.1.11 

func (m *MockProvider) StreamTokens(ctx context.Context, model string, messages []providers.Message) (<-chan providers.GenerateResponse, error)

StreamTokens mocks base method.

type MockProviderMockRecorder 

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

MockProviderMockRecorder is the mock recorder for MockProvider.

func (*MockProviderMockRecorder) GenerateTokens 

func (mr *MockProviderMockRecorder) GenerateTokens(ctx, model, messages, tools, maxTokens any) *gomock.Call

GenerateTokens indicates an expected call of GenerateTokens.

func (*MockProviderMockRecorder) GetAuthType 

func (mr *MockProviderMockRecorder) GetAuthType() *gomock.Call

GetAuthType indicates an expected call of GetAuthType.

func (*MockProviderMockRecorder) GetExtraHeaders 

func (mr *MockProviderMockRecorder) GetExtraHeaders() *gomock.Call

GetExtraHeaders indicates an expected call of GetExtraHeaders.

func (*MockProviderMockRecorder) GetID 

func (mr *MockProviderMockRecorder) GetID() *gomock.Call

GetID indicates an expected call of GetID.

func (*MockProviderMockRecorder) GetName 

func (mr *MockProviderMockRecorder) GetName() *gomock.Call

GetName indicates an expected call of GetName.

func (*MockProviderMockRecorder) GetToken 

func (mr *MockProviderMockRecorder) GetToken() *gomock.Call

GetToken indicates an expected call of GetToken.

func (*MockProviderMockRecorder) GetURL 

func (mr *MockProviderMockRecorder) GetURL() *gomock.Call

GetURL indicates an expected call of GetURL.

func (*MockProviderMockRecorder) ListModels 

func (mr *MockProviderMockRecorder) ListModels(ctx any) *gomock.Call

ListModels indicates an expected call of ListModels.

func (*MockProviderMockRecorder) StreamTokens added in v0.1.11 

func (mr *MockProviderMockRecorder) StreamTokens(ctx, model, messages any) *gomock.Call

StreamTokens indicates an expected call of StreamTokens.

type MockProviderRegistry added in v0.1.11 

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

MockProviderRegistry is a mock of ProviderRegistry interface.

func NewMockProviderRegistry added in v0.1.11 

func NewMockProviderRegistry(ctrl *gomock.Controller) *MockProviderRegistry

NewMockProviderRegistry creates a new mock instance.

func (*MockProviderRegistry) BuildProvider added in v0.1.11 

func (m *MockProviderRegistry) BuildProvider(providerID string, client providers.Client) (providers.Provider, error)

BuildProvider mocks base method.

func (*MockProviderRegistry) EXPECT added in v0.1.11 

func (m *MockProviderRegistry) EXPECT() *MockProviderRegistryMockRecorder

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

func (*MockProviderRegistry) GetProviders added in v0.1.11 

func (m *MockProviderRegistry) GetProviders() map[string]*providers.Config

GetProviders mocks base method.

type MockProviderRegistryMockRecorder added in v0.1.11 

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

MockProviderRegistryMockRecorder is the mock recorder for MockProviderRegistry.

func (*MockProviderRegistryMockRecorder) BuildProvider added in v0.1.11 

func (mr *MockProviderRegistryMockRecorder) BuildProvider(providerID, client any) *gomock.Call

BuildProvider indicates an expected call of BuildProvider.

func (*MockProviderRegistryMockRecorder) GetProviders added in v0.1.11 

func (mr *MockProviderRegistryMockRecorder) GetProviders() *gomock.Call

GetProviders indicates an expected call of GetProviders.

type MockRouter 

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

MockRouter is a mock of Router interface.

func NewMockRouter 

func NewMockRouter(ctrl *gomock.Controller) *MockRouter

NewMockRouter creates a new mock instance.

func (*MockRouter) EXPECT 

func (m *MockRouter) EXPECT() *MockRouterMockRecorder

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

func (*MockRouter) GenerateProvidersTokenHandler 

func (m *MockRouter) GenerateProvidersTokenHandler(c *gin.Context)

GenerateProvidersTokenHandler mocks base method.

func (*MockRouter) HealthcheckHandler 

func (m *MockRouter) HealthcheckHandler(c *gin.Context)

HealthcheckHandler mocks base method.

func (*MockRouter) ListAllModelsHandler 

func (m *MockRouter) ListAllModelsHandler(c *gin.Context)

ListAllModelsHandler mocks base method.

func (*MockRouter) ListModelsHandler 

func (m *MockRouter) ListModelsHandler(c *gin.Context)

ListModelsHandler mocks base method.

func (*MockRouter) NotFoundHandler 

func (m *MockRouter) NotFoundHandler(c *gin.Context)

NotFoundHandler mocks base method.

func (*MockRouter) ProxyHandler 

func (m *MockRouter) ProxyHandler(c *gin.Context)

ProxyHandler mocks base method.

type MockRouterMockRecorder 

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

MockRouterMockRecorder is the mock recorder for MockRouter.

func (*MockRouterMockRecorder) GenerateProvidersTokenHandler 

func (mr *MockRouterMockRecorder) GenerateProvidersTokenHandler(c any) *gomock.Call

GenerateProvidersTokenHandler indicates an expected call of GenerateProvidersTokenHandler.

func (*MockRouterMockRecorder) HealthcheckHandler 

func (mr *MockRouterMockRecorder) HealthcheckHandler(c any) *gomock.Call

HealthcheckHandler indicates an expected call of HealthcheckHandler.

func (*MockRouterMockRecorder) ListAllModelsHandler 

func (mr *MockRouterMockRecorder) ListAllModelsHandler(c any) *gomock.Call

ListAllModelsHandler indicates an expected call of ListAllModelsHandler.

func (*MockRouterMockRecorder) ListModelsHandler 

func (mr *MockRouterMockRecorder) ListModelsHandler(c any) *gomock.Call

ListModelsHandler indicates an expected call of ListModelsHandler.

func (*MockRouterMockRecorder) NotFoundHandler 

func (mr *MockRouterMockRecorder) NotFoundHandler(c any) *gomock.Call

NotFoundHandler indicates an expected call of NotFoundHandler.

func (*MockRouterMockRecorder) ProxyHandler 

func (mr *MockRouterMockRecorder) ProxyHandler(c any) *gomock.Call

ProxyHandler indicates an expected call of ProxyHandler.

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.