test

Documentation

Index

• type MockClient
  • func NewMockClient() *MockClient
  • func (m *MockClient) Chat(messages []chat.Message, options *chat.ChatOptions) (*chat.Message, chat.ResponseInfo, error)
  • func (c *MockClient) ChatWithFunctions(messages []chat.Message, functions []tools.FunctionConfig, ...) (*chat.Message, chat.ResponseInfo, error)
• type MockServer
  • func NewTestServer() *MockServer
  • func (m *MockServer) Close()
  • func (m *MockServer) Init()
  • func (m *MockServer) SetRequestHandler(handler func(w http.ResponseWriter, r *http.Request))
  • func (m *MockServer) Start()
  • func (m *MockServer) URL() string

Types

type MockClient

type MockClient struct {
	ChatOutput string
	Err        error
}

func NewMockClient

func NewMockClient() *MockClient

func (*MockClient) Chat

func (m *MockClient) Chat(messages []chat.Message, options *chat.ChatOptions) (*chat.Message, chat.ResponseInfo, error)

func (*MockClient) ChatWithFunctions

func (c *MockClient) ChatWithFunctions(messages []chat.Message, functions []tools.FunctionConfig, options *chat.ChatOptions) (*chat.Message, chat.ResponseInfo, error)

type MockServer

type MockServer struct {
	*httptest.Server

	RequestHandler func(w http.ResponseWriter, r *http.Request)
}

func NewTestServer

func NewTestServer() *MockServer

func (*MockServer) Close

func (m *MockServer) Close()

func (*MockServer) Init

func (m *MockServer) Init()

func (*MockServer) SetRequestHandler

func (m *MockServer) SetRequestHandler(handler func(w http.ResponseWriter, r *http.Request))

func (*MockServer) Start

func (m *MockServer) Start()

func (*MockServer) URL

func (m *MockServer) URL() string