engine

package
v0.6.0 Latest Latest
Warning

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

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

Documentation

Index

Constants

View Source 
const (
	// Default configuration
	DefaultPort = "16677"
	DefaultHost = constants.DefaultHost + ":" + DefaultPort

	// ipex-llm-ollama related
	IpexLlamaDir     = "ipex-llm-ollama"
	OllamaBatchFile  = "ollama-serve.bat"
	OllamaStartShell = "start-ollama.sh"

	// Windows download URLs
	WindowsAllGPUURL   = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ollama-windows-amd64-all.zip"
	WindowsNvidiaURL   = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ollama-windows-amd64.zip"
	WindowsAMDURL      = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ollama-windows-amd64-rocm.zip"
	WindowsIntelArcURL = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ipex-llm-ollama.zip"
	WindowsBaseURL     = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ollama-windows-amd64-base.zip"

	// Linux download URLs
	LinuxNvidiaURL   = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ollama-cuda-linux-amd64.tgz"
	LinuxAMDURL      = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ollama-rocm-linux-amd64.tgz"
	LinuxBaseURL     = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ollama-linux-amd64.tgz"
	LinuxARMURL      = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ollama-cuda-linux-arm64.tgz"
	LinuxARMBaseURL  = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ollama-linux-arm64.tgz"
	LinuxIntelArcURL = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ipex-llm-ollama.zip"

	// macOS download URLs
	MacOSIntelURL = constants.BaseDownloadURL + constants.UrlDirPathMacOS + "/Ollama-darwin.zip"

	// Archive commands
	TarCommand     = "tar"
	TarExtractFlag = "-xf"
	TarDestFlag    = "-C"
	UnzipCommand   = "unzip"
	UnzipDestFlag  = "-d"
	MoveCommand    = "mv"

	OllamaStartShellWin   = `%s\\ollama.exe serve`
	OllamaStartShellLinux = `%s/ollama serve`
	OllamaStartShellMac   = `%s/ollama serve`

	OllamaModelDirLinux = "/var/lib/aog/engine/ollama/models"

	// Required version
	OllamaMinVersion = "0.7.1"
)
View Source 
const (
	ModelScopeSCHEME               = "https"
	ModelScopeEndpointCN           = "www.modelscope.cn"
	ModelScopeEndpointAI           = "www.modelscope.ai"
	ModelScopeGetModelFilesReqPath = "/api/v1/models/%s/repo/files?Revision=%s&Recursive=%s"
	ModelScopeModelDownloadReqPath = "/api/v1/models/%s/repo?Revision=%s&FilePath=%s"
	ModelScopeRevision             = "master"
	BufferSize                     = 64 * 1024

	// OpenVINO Server configuration
	OpenvinoGRPCPort     = "9000"
	OpenvinoGRPCHost     = constants.DefaultHost + ":" + OpenvinoGRPCPort
	OpenvinoHTTPPort     = "16666"
	OpenvinoHTTPHost     = constants.DefaultHost + ":" + OpenvinoHTTPPort
	OpenvinoVersion      = "2025.0.0"
	OpenvinoDefaultModel = "stable-diffusion-v-1-5-ov-fp16"

	// Download URLs
	OVMSWindowsDownloadURL       = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/ovms_windows.zip"
	OVMSLinuxRedHatDownloadURL   = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ovms_redhat_python_on.tar.gz"
	OVMSLinuxUbuntu22DownloadURL = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ovms_ubuntu22_python_on.tar.gz"
	OVMSLinuxUbuntu24DownloadURL = constants.BaseDownloadURL + constants.UrlDirPathLinux + "/" + version.AOGVersion + "/ovms_ubuntu24_python_on.tar.gz"

	ScriptsDownloadURL = constants.BaseDownloadURL + constants.UrlDirPathWindows + "/" + version.AOGVersion + "/scripts.zip"

	// Required Version
	OpenvinoMinVersion = "2025.2.0"
)
View Source 
const (
	GraphPBTxtSpeechToText = `` /* 486-byte string literal not displayed */

	GraphPBTxtTextToImage = `` /* 640-byte string literal not displayed */

	GraphPBTxtTextToSpeech = `` /* 562-byte string literal not displayed */

	InitShellWin = `` /* 166-byte string literal not displayed */

	InitShellLinuxUbuntu2204 = `` /* 271-byte string literal not displayed */

	InitShellLinuxUbuntu2404 = `` /* 271-byte string literal not displayed */

	InitShellLinuxREHL96 = `` /* 271-byte string literal not displayed */

	StartShellWin = `` /* 212-byte string literal not displayed */

	StartShellLinux = `` /* 155-byte string literal not displayed */

)

Variables

View Source 
var OllamaOperateStatus = 1
View Source 
var OpenvinoOperateStatus = 1

Functions

func AsyncDownloadModelFile 

func AsyncDownloadModelFile(ctx context.Context, a AsyncDownloadModelFileData, engine *OpenvinoProvider)

func CheckFileDigest 

func CheckFileDigest(ExceptDigest string, FilePath string) bool

func DownloadModelFileRequest 

func DownloadModelFileRequest(ctx context.Context, reqData *ModelScopeFileReqData, reqHeader map[string]string) (chan []byte, chan error)

func GetHttpClient 

func GetHttpClient() *client.Client

func GetModelScopeDomain 

func GetModelScopeDomain(cnSite bool) string

func QuotePlus 

func QuotePlus(s string) string

func VersionCompare added in v0.6.0 

func VersionCompare(v1, v2 string) int

Types

type AsyncDownloadModelFileData 

type AsyncDownloadModelFileData struct {
	ModelName      string
	ModelType      string
	DataCh         chan []byte
	ErrCh          chan error
	ModelFiles     []ModelScopeFile
	LocalModelPath string
}

type ModelConfig 

type ModelConfig struct {
	Name      string `json:"name"`
	BasePath  string `json:"base_path"`
	GraphPath string `json:"graph_path"`
}

type ModelScopeFile 

type ModelScopeFile struct {
	Name     string `json:"Name"`
	Path     string `json:"Path"`
	Digest   string `json:"Sha256"`
	Size     int64  `json:"Size"`
	IsLFS    bool   `json:"IsLFS"`
	Revision string `json:"Revision"`
	Type     string `json:"Type"`
}

func GetModelFiles 

func GetModelFiles(ctx context.Context, reqData *ModelScopeFileReqData) ([]ModelScopeFile, error)

type ModelScopeFileData 

type ModelScopeFileData struct {
	Files []ModelScopeFile `json:"Files"`
}

type ModelScopeFileReqData 

type ModelScopeFileReqData struct {
	ModelName string `json:"model_name"`
	Revision  string `json:"revision"`
	Recursive string `json:"recursive"`
	FilePath  string `json:"file_path"`
	Stream    bool   `json:"stream"`
}

type ModelScopeFileRespData 

type ModelScopeFileRespData struct {
	Code int                `json:"Code"`
	Data ModelScopeFileData `json:"Data"`
}

type OllamaProvider 

type OllamaProvider struct {
	EngineConfig *types.EngineRecommendConfig
	// contains filtered or unexported fields
}

func NewOllamaProvider 

func NewOllamaProvider(config *types.EngineRecommendConfig) *OllamaProvider

func (*OllamaProvider) CheckEngine added in v0.6.0 

func (o *OllamaProvider) CheckEngine() bool

CheckEngine checks if Ollama engine is installed

func (*OllamaProvider) DeleteModel 

func (o *OllamaProvider) DeleteModel(ctx context.Context, req *types.DeleteRequest) error

func (*OllamaProvider) GetConfig 

func (o *OllamaProvider) GetConfig() *types.EngineRecommendConfig

func (*OllamaProvider) GetDefaultClient 

func (o *OllamaProvider) GetDefaultClient() *client.Client

func (*OllamaProvider) GetOperateStatus added in v0.6.0 

func (o *OllamaProvider) GetOperateStatus() int

func (*OllamaProvider) GetRunningModels 

func (o *OllamaProvider) GetRunningModels(ctx context.Context) (*types.ListResponse, error)

func (*OllamaProvider) GetVersion 

func (o *OllamaProvider) GetVersion(ctx context.Context, resp *types.EngineVersionResponse) (*types.EngineVersionResponse, error)

func (*OllamaProvider) HealthCheck 

func (o *OllamaProvider) HealthCheck() error

func (*OllamaProvider) InitEnv 

func (o *OllamaProvider) InitEnv() error

func (*OllamaProvider) InstallEngine 

func (o *OllamaProvider) InstallEngine(cover bool) error

func (*OllamaProvider) ListModels 

func (o *OllamaProvider) ListModels(ctx context.Context) (*types.ListResponse, error)

func (*OllamaProvider) LoadModel 

func (o *OllamaProvider) LoadModel(ctx context.Context, req *types.LoadRequest) error

func (*OllamaProvider) PullModel 

func (o *OllamaProvider) PullModel(ctx context.Context, req *types.PullModelRequest, fn types.PullProgressFunc) (*types.ProgressResponse, error)

func (*OllamaProvider) PullModelStream 

func (o *OllamaProvider) PullModelStream(ctx context.Context, req *types.PullModelRequest) (chan []byte, chan error)

func (*OllamaProvider) SetOperateStatus added in v0.6.0 

func (o *OllamaProvider) SetOperateStatus(status int)

func (*OllamaProvider) SetProcessManager added in v0.6.0 

func (o *OllamaProvider) SetProcessManager(pm *process.EngineProcessManager)

SetProcessManager sets the process manager for the provider

func (*OllamaProvider) StartEngine 

func (o *OllamaProvider) StartEngine(mode string) error

func (*OllamaProvider) StopEngine 

func (o *OllamaProvider) StopEngine() error

func (*OllamaProvider) UnloadModel 

func (o *OllamaProvider) UnloadModel(ctx context.Context, req *types.UnloadModelRequest) error

func (*OllamaProvider) UpgradeEngine added in v0.6.0 

func (o *OllamaProvider) UpgradeEngine() error

type OpenvinoModelServerConfig 

type OpenvinoModelServerConfig struct {
	MediapipeConfigList []ModelConfig `json:"mediapipe_config_list"`
	ModelConfigList     []interface{} `json:"model_config_list"`
}

type OpenvinoProvider 

type OpenvinoProvider struct {
	EngineConfig *types.EngineRecommendConfig
	// contains filtered or unexported fields
}

func NewOpenvinoProvider 

func NewOpenvinoProvider(config *types.EngineRecommendConfig) *OpenvinoProvider

func (*OpenvinoProvider) CheckEngine added in v0.6.0 

func (o *OpenvinoProvider) CheckEngine() bool

CheckEngine checks if OpenVINO engine is installed

func (*OpenvinoProvider) DeleteModel 

func (o *OpenvinoProvider) DeleteModel(ctx context.Context, req *types.DeleteRequest) error

func (*OpenvinoProvider) GetConfig 

func (o *OpenvinoProvider) GetConfig() *types.EngineRecommendConfig

func (*OpenvinoProvider) GetDefaultClient 

func (o *OpenvinoProvider) GetDefaultClient() *client.GRPCClient

func (*OpenvinoProvider) GetOperateStatus added in v0.6.0 

func (o *OpenvinoProvider) GetOperateStatus() int

func (*OpenvinoProvider) GetRunningModels 

func (o *OpenvinoProvider) GetRunningModels(ctx context.Context) (*types.ListResponse, error)

func (*OpenvinoProvider) GetVersion 

func (o *OpenvinoProvider) GetVersion(ctx context.Context, resp *types.EngineVersionResponse) (*types.EngineVersionResponse, error)

func (*OpenvinoProvider) HealthCheck 

func (o *OpenvinoProvider) HealthCheck() error

func (*OpenvinoProvider) InitEnv 

func (o *OpenvinoProvider) InitEnv() error

func (*OpenvinoProvider) InstallEngine 

func (o *OpenvinoProvider) InstallEngine(cover bool) error

func (*OpenvinoProvider) ListModels 

func (o *OpenvinoProvider) ListModels(ctx context.Context) (*types.ListResponse, error)

func (*OpenvinoProvider) LoadModel 

func (o *OpenvinoProvider) LoadModel(ctx context.Context, req *types.LoadRequest) error

func (*OpenvinoProvider) PullModel 

func (o *OpenvinoProvider) PullModel(ctx context.Context, req *types.PullModelRequest, fn types.PullProgressFunc) (*types.ProgressResponse, error)

func (*OpenvinoProvider) PullModelStream 

func (o *OpenvinoProvider) PullModelStream(ctx context.Context, req *types.PullModelRequest) (chan []byte, chan error)

func (*OpenvinoProvider) SetOperateStatus added in v0.6.0 

func (o *OpenvinoProvider) SetOperateStatus(status int)

func (*OpenvinoProvider) SetProcessManager added in v0.6.0 

func (o *OpenvinoProvider) SetProcessManager(pm *process.EngineProcessManager)

SetProcessManager sets the process manager for the provider

func (*OpenvinoProvider) StartEngine 

func (o *OpenvinoProvider) StartEngine(mode string) error

func (*OpenvinoProvider) StopEngine 

func (o *OpenvinoProvider) StopEngine() error

func (*OpenvinoProvider) UnloadModel 

func (o *OpenvinoProvider) UnloadModel(ctx context.Context, req *types.UnloadModelRequest) error

func (*OpenvinoProvider) UpgradeEngine added in v0.6.0 

func (o *OpenvinoProvider) UpgradeEngine() error

Source Files

View all Source files

Jump to

Keyboard shortcuts

? : This menu
/ : Search site
f or F : Jump to
y or Y : Canonical URL
go.dev uses cookies from Google to deliver and enhance the quality of its services and to analyze traffic. Learn more.