test

package
v0.0.13 Latest Latest
Warning

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

Go to latest
Published: Sep 9, 2025 License: Apache-2.0 Imports: 10 Imported by: 0

Documentation

Index

Constants

This section is empty.

Variables

This section is empty.

Functions

func Clone

func Clone[T any](src T) T

func Must

func Must[T any](v T, err error) T

func RestoreEnv

func RestoreEnv(originalEnv []string)

func WriteObject

func WriteObject(w http.ResponseWriter, obj any)

Types

type ChatModel

type ChatModel struct {
	StreamReader  func(input []*schema.Message, opts ...model.Option) (*schema.StreamReader[*schema.Message], error)
	WithToolsFunc func(tools []*schema.ToolInfo) (model.ToolCallingChatModel, error)
}

func (*ChatModel) Generate

func (c *ChatModel) Generate(_ context.Context, _ []*schema.Message, _ ...model.Option) (*schema.Message, error)

func (*ChatModel) Stream

func (c *ChatModel) Stream(_ context.Context, input []*schema.Message, opts ...model.Option) (*schema.StreamReader[*schema.Message], error)

func (*ChatModel) WithTools

func (c *ChatModel) WithTools(tools []*schema.ToolInfo) (model.ToolCallingChatModel, error)

type InferenceProvider

type InferenceProvider struct {
	api.BasicInferenceProvider
	Initialized bool                       `json:"-"`
	Llm         model.ToolCallingChatModel `json:"-"`
}

func NewInferenceProvider

func NewInferenceProvider(name string, options ...InferenceProviderOption) *InferenceProvider

func (*InferenceProvider) GetInference

func (*InferenceProvider) Initialize

func (i *InferenceProvider) Initialize(_ context.Context)

type InferenceProviderOption

type InferenceProviderOption func(*InferenceProvider)

func WithInferenceAvailable

func WithInferenceAvailable() InferenceProviderOption

func WithInferenceLocal

func WithInferenceLocal() InferenceProviderOption

func WithInferencePublic

func WithInferencePublic() InferenceProviderOption

type MockHandlerFunc

type MockHandlerFunc func(http.ResponseWriter, *http.Request) (handled bool)

type MockServer

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

func NewMockServer

func NewMockServer() *MockServer

func (*MockServer) Close

func (m *MockServer) Close()

func (*MockServer) Handle

func (m *MockServer) Handle(handler MockHandlerFunc)

func (*MockServer) URL

func (m *MockServer) URL() string

type ToolsProvider

type ToolsProvider struct {
	api.BasicToolsProvider
	Initialized bool        `json:"-"`
	Tools       []*api.Tool `json:"-"`
}

func NewToolsProvider

func NewToolsProvider(name string, options ...ToolsProviderOption) *ToolsProvider

func (*ToolsProvider) GetTools

func (t *ToolsProvider) GetTools(_ context.Context) ([]*api.Tool, error)

func (*ToolsProvider) Initialize

func (t *ToolsProvider) Initialize(_ context.Context)

type ToolsProviderOption

type ToolsProviderOption func(*ToolsProvider)

func WithToolsAvailable

func WithToolsAvailable() ToolsProviderOption

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL