ai

type AiService ¶

type AiService interface {
	Chat(ctx context.Context, request *client.ChatRequest) (*client.ChatResponse, error)
	ListModels(ctx context.Context) *ModelList
}

type AiServiceImpl ¶

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

func NewAiService(cfg *config.Config, clients map[string]client.ChatClient, logger *zap.SugaredLogger) (*AiServiceImpl, error)

func (ai *AiServiceImpl) Chat(ctx context.Context, request *client.ChatRequest) (*client.ChatResponse, error)

func (ai *AiServiceImpl) ListModels(ctx context.Context) *ModelList

type AiServiceMock ¶

type AiServiceMock struct {
	// ChatFunc mocks the Chat method.
	ChatFunc func(ctx context.Context, request *client.ChatRequest) (*client.ChatResponse, error)

	// ListModelsFunc mocks the ListModels method.
	ListModelsFunc func(ctx context.Context) *ModelList
	// contains filtered or unexported fields
}

AiServiceMock is a mock implementation of AiService.

func TestSomethingThatUsesAiService(t *testing.T) {

	// make and configure a mocked AiService
	mockedAiService := &AiServiceMock{
		ChatFunc: func(ctx context.Context, request *client.ChatRequest) (*client.ChatResponse, error) {
			panic("mock out the Chat method")
		},
		ListModelsFunc: func(ctx context.Context) *ModelList {
			panic("mock out the ListModels method")
		},
	}

	// use mockedAiService in code that requires AiService
	// and then make assertions.

}

func (mock *AiServiceMock) Chat(ctx context.Context, request *client.ChatRequest) (*client.ChatResponse, error)

func (mock *AiServiceMock) ChatCalls() []struct {
	Ctx     context.Context
	Request *client.ChatRequest
}

len(mockedAiService.ChatCalls())

func (mock *AiServiceMock) ListModels(ctx context.Context) *ModelList

func (mock *AiServiceMock) ListModelsCalls() []struct {
	Ctx context.Context
}

len(mockedAiService.ListModelsCalls())

type ModelList ¶ added in v0.0.3

type ModelList struct {
	Models  []string `json:"models"`
	Default string   `json:"default"`
}