client

type BidiMessage struct {
	Data        []byte
	MessageType string
	Metadata    map[string]string
	Error       error
}

type BidirectionalPlugin interface {
	PluginProvider

	// InvokeServiceBidirectional handles bidirectional streaming communication.
	// Reads from inStream and writes to outStream until context is cancelled or streams are closed.
	InvokeServiceBidirectional(ctx context.Context, serviceName string, wsConnID string, authInfo string, inStream <-chan BidiMessage, outStream chan<- BidiMessage) error
}

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

type LocalPluginProvider interface {
	PluginProvider

	// StartEngine starts the engine.
	StartEngine(mode string) error

	// StopEngine stops the engine.
	StopEngine() error

	// GetConfig returns the engine configuration.
	GetConfig(ctx context.Context) (*types.EngineRecommendConfig, error)

	// CheckEngine checks if the engine is installed.
	CheckEngine() (bool, error)

	// InstallEngine installs the engine.
	InstallEngine(ctx context.Context) error

	// InitEnv initializes the environment.
	InitEnv() error

	// UpgradeEngine upgrades the engine.
	UpgradeEngine(ctx context.Context) error

	// PullModel pulls a model (blocking).
	PullModel(ctx context.Context, req *types.PullModelRequest, fn types.PullProgressFunc) (*types.ProgressResponse, error)

	// PullModelStream pulls a model (streaming).
	PullModelStream(ctx context.Context, req *types.PullModelRequest) (chan []byte, chan error)

	// DeleteModel deletes a model.
	DeleteModel(ctx context.Context, req *types.DeleteRequest) error

	// ListModels lists all downloaded models.
	ListModels(ctx context.Context) (*types.ListResponse, error)

	// LoadModel loads a model into memory.
	LoadModel(ctx context.Context, req *types.LoadRequest) error

	// UnloadModel unloads models from memory.
	UnloadModel(ctx context.Context, req *types.UnloadModelRequest) error

	// GetRunningModels returns currently running models.
	GetRunningModels(ctx context.Context) (*types.ListResponse, error)

	// GetVersion returns the engine version information.
	GetVersion(ctx context.Context, resp *types.EngineVersionResponse) (*types.EngineVersionResponse, error)
}

type PluginProvider interface {
	// GetManifest returns the plugin metadata.
	GetManifest() *types.PluginManifest

	// GetOperateStatus returns the operational status.
	GetOperateStatus() int

	// SetOperateStatus sets the operational status.
	SetOperateStatus(status int)

	// HealthCheck performs a health check.
	HealthCheck(ctx context.Context) error

	GetSupportModelList(ctx context.Context) ([]types.RecommendModelData, error)

	// InvokeService invokes a plugin service (core method).
	// serviceName: Service name (e.g., "chat", "embed")
	// request: Serialized request data
	// Returns: Serialized response data
	InvokeService(ctx context.Context, serviceName string, authInfo string, request []byte) ([]byte, error)
}

type RemotePluginProvider interface {
	PluginProvider

	// SetAuth sets authentication information.
	SetAuth(req *http.Request, authInfo string, credentials map[string]string) error

	// ValidateAuth validates if authentication information is valid.
	ValidateAuth(ctx context.Context) error

	// RefreshAuth refreshes authentication information (for OAuth).
	RefreshAuth(ctx context.Context) error
}

type StreamChunk struct {
	Data     []byte
	Error    error
	IsFinal  bool
	Metadata map[string]string
}

type StreamablePlugin interface {
	PluginProvider

	// InvokeServiceStream executes a streaming service request.
	// Returns a channel that sends data chunks and closes when complete.
	// authInfo: Authentication information for the request (can be empty for local plugins)
	InvokeServiceStream(ctx context.Context, serviceName string, authInfo string, request []byte) (<-chan StreamChunk, error)
}