testing

Documentation

Index

• func ChatMockLatency(model providers.Model) *latency.MovingAverage
• type LangModelMock
  • func NewLangModelMock(ID string, healthy bool, avgLatency float64, weight int) LangModelMock
  • func (m *LangModelMock) ChatLatency() *latency.MovingAverage
  • func (m LangModelMock) Healthy() bool
  • func (m LangModelMock) ID() string
  • func (m LangModelMock) LatencyUpdateInterval() *fields.Duration
  • func (m LangModelMock) Weight() int
• type ProviderMock
  • func NewProviderMock(modelName *string, responses []RespMock) *ProviderMock
  • func NewStreamProviderMock(modelName *string, chatStreams []RespStreamMock) *ProviderMock
  • func (c *ProviderMock) Chat(_ context.Context, _ *schemas.ChatParams) (*schemas.ChatResponse, error)
  • func (c *ProviderMock) ChatStream(_ context.Context, _ *schemas.ChatParams) (clients.ChatStream, error)
  • func (c *ProviderMock) ModelName() string
  • func (c *ProviderMock) Provider() string
  • func (c *ProviderMock) SupportChatStream() bool
• type RespMock
  • func (m *RespMock) Resp() *schemas.ChatResponse
  • func (m *RespMock) RespChunk() *schemas.ChatStreamChunk
• type RespStreamMock
  • func NewRespStreamMock(chunk *[]RespMock) RespStreamMock
  • func NewRespStreamWithOpenErr(openErr error) RespStreamMock
  • func (m *RespStreamMock) Close() error
  • func (m *RespStreamMock) Open() error
  • func (m *RespStreamMock) Recv() (*schemas.ChatStreamChunk, error)

Functions

func ChatMockLatency

func ChatMockLatency(model providers.Model) *latency.MovingAverage

Types

type LangModelMock

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

LangModelMock

func NewLangModelMock

func NewLangModelMock(ID string, healthy bool, avgLatency float64, weight int) LangModelMock

func (*LangModelMock) ChatLatency

func (m *LangModelMock) ChatLatency() *latency.MovingAverage

func (LangModelMock) Healthy

func (m LangModelMock) Healthy() bool

func (LangModelMock) ID

func (m LangModelMock) ID() string

func (LangModelMock) LatencyUpdateInterval

func (m LangModelMock) LatencyUpdateInterval() *fields.Duration

func (LangModelMock) Weight

func (m LangModelMock) Weight() int

type ProviderMock

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

ProviderMock mocks a model provider

func NewProviderMock

func NewProviderMock(modelName *string, responses []RespMock) *ProviderMock

func NewStreamProviderMock

func NewStreamProviderMock(modelName *string, chatStreams []RespStreamMock) *ProviderMock

func (*ProviderMock) Chat

func (c *ProviderMock) Chat(_ context.Context, _ *schemas.ChatParams) (*schemas.ChatResponse, error)

func (*ProviderMock) ChatStream

func (c *ProviderMock) ChatStream(_ context.Context, _ *schemas.ChatParams) (clients.ChatStream, error)

func (*ProviderMock) ModelName

func (c *ProviderMock) ModelName() string

func (*ProviderMock) Provider

func (c *ProviderMock) Provider() string

func (*ProviderMock) SupportChatStream

func (c *ProviderMock) SupportChatStream() bool

type RespMock

type RespMock struct {
	Msg string
	Err error
}

RespMock mocks a chat response or a streaming chat chunk

func (*RespMock) Resp

func (m *RespMock) Resp() *schemas.ChatResponse

func (*RespMock) RespChunk

func (m *RespMock) RespChunk() *schemas.ChatStreamChunk

type RespStreamMock

type RespStreamMock struct {
	OpenErr error
	Chunks  *[]RespMock
	// contains filtered or unexported fields
}

RespStreamMock mocks a chat stream

func NewRespStreamMock

func NewRespStreamMock(chunk *[]RespMock) RespStreamMock

func NewRespStreamWithOpenErr

func NewRespStreamWithOpenErr(openErr error) RespStreamMock

func (*RespStreamMock) Close

func (m *RespStreamMock) Close() error

func (*RespStreamMock) Open

func (m *RespStreamMock) Open() error

func (*RespStreamMock) Recv

func (m *RespStreamMock) Recv() (*schemas.ChatStreamChunk, error)