api

type Completion interface {
	// Complete returns a list of possible completions for the given input.
	Complete(ctx context.Context, messages []Message) (*Message, Reason, error)
}

type CompletionGRPCClient struct {
	Client CompletionClient
}

type CompletionGRPCServer struct {
	UnimplementedCompletionServer

	Impl Completion
}

type Embedding interface {
	// Embedding returns the embedding for the given input.
	Embedding(ctx context.Context, input string) ([]float32, error)
}

type EmbeddingGRPCClient struct {
	Client EmbeddingClient
}

type EmbeddingGRPCServer struct {
	UnimplementedEmbeddingServer

	Impl Embedding
}

type Interfaces interface {
	// Interfaces returns a list of interfaces that the plugin implements.
	Interfaces(ctx context.Context) ([]string, error)
}

type InterfacesGRPCClient struct {
	Client InterfacesClient
}

type InterfacesGRPCServer struct {
	UnimplementedInterfacesServer

	Impl Interfaces
}

type Memory interface {
	// Memorize memorizes each data string.
	Memorize(ctx context.Context, data []string) error

	// Recall recalls the memory closest to the data string. If a count is
	// specified, it returns the count closest memories.
	Recall(ctx context.Context, data string, count ...int) ([]string, error)
}

type MemoryGRPCClient struct {
	Client MemoryClient
}

type MemoryGRPCServer struct {
	UnimplementedMemoryServer

	Impl Memory
}

type Message struct {
	Content string `json:"content"`
	Name    string `json:"name"`
	Role    string `json:"role"`
}

type Plugin struct {
	plugin.Plugin
	// contains filtered or unexported fields
}