mirror of
https://github.com/langgenius/dify-plugin-daemon.git
synced 2026-07-22 01:35:24 -04:00
wangxiaolei
101 lines
3.2 KiB
Go
101 lines
3.2 KiB
Go
package local_runtime
|
|
|
|
import (
|
|
_ "embed"
|
|
"fmt"
|
|
|
|
"github.com/langgenius/dify-plugin-daemon/pkg/utils/log"
|
|
"go.opentelemetry.io/otel/attribute"
|
|
)
|
|
|
|
func (p *LocalPluginRuntime) InitPythonEnvironment() error {
|
|
// root span for python env init
|
|
_, span := p.startSpan("python.init_env", attribute.String("plugin.identity", p.Config.Identity()))
|
|
defer span.End()
|
|
|
|
// prepare uv environment
|
|
uvPath, err := p.prepareUV()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to find uv path: %w", err)
|
|
}
|
|
|
|
// check if virtual environment exists
|
|
venv, err := p.checkPythonVirtualEnvironment()
|
|
switch err {
|
|
case ErrVirtualEnvironmentInvalid:
|
|
// remove the venv and rebuild it
|
|
log.Warn("virtual environment for %s is invalid; deleting and recreating", p.Config.Identity())
|
|
err = p.deleteVirtualEnvironment()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
// create virtual environment
|
|
venv, err = p.createVirtualEnvironment(uvPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create virtual environment: %w", err)
|
|
}
|
|
case ErrVirtualEnvironmentNotFound:
|
|
// create virtual environment
|
|
venv, err = p.createVirtualEnvironment(uvPath)
|
|
if err != nil {
|
|
return fmt.Errorf("failed to create virtual environment: %w", err)
|
|
}
|
|
case nil:
|
|
// PATCH:
|
|
// plugin sdk version less than 0.0.1b70 contains a memory leak bug
|
|
// to reach a better user experience, we will patch it here using a patched file
|
|
// https://github.com/langgenius/dify-plugin-sdks/commit/161045b65f708d8ef0837da24440ab3872821b3b
|
|
dependencyFilePath, err := p.getDependencyFilePath()
|
|
if err != nil {
|
|
log.Error("failed to get dependency file path for patching", "error", err)
|
|
} else if err := p.patchPluginSdk(
|
|
dependencyFilePath,
|
|
venv.pythonInterpreterPath,
|
|
); err != nil {
|
|
log.Error("failed to patch the plugin sdk", "error", err)
|
|
}
|
|
|
|
// everything is good, return nil
|
|
return nil
|
|
default:
|
|
return fmt.Errorf("failed to check virtual environment: %w", err)
|
|
}
|
|
|
|
// detect dependency file type and install dependencies
|
|
dependencyFileType, err := p.detectDependencyFileType()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to detect dependency file: %w", err)
|
|
}
|
|
|
|
if err := p.installDependencies(uvPath, dependencyFileType); err != nil {
|
|
return fmt.Errorf("failed to install dependencies: %w", err)
|
|
}
|
|
|
|
// pre-compile the plugin to avoid costly compilation on first invocation
|
|
if err := p.preCompile(venv.pythonInterpreterPath); err != nil {
|
|
return fmt.Errorf("failed to pre-compile the plugin: %w", err)
|
|
}
|
|
|
|
// PATCH:
|
|
// plugin sdk version less than 0.0.1b70 contains a memory leak bug
|
|
// to reach a better user experience, we will patch it here using a patched file
|
|
// https://github.com/langgenius/dify-plugin-sdks/commit/161045b65f708d8ef0837da24440ab3872821b3b
|
|
dependencyFilePath, err := p.getDependencyFilePath()
|
|
if err != nil {
|
|
log.Error("failed to get dependency file path for patching", "error", err)
|
|
} else if err := p.patchPluginSdk(
|
|
dependencyFilePath,
|
|
venv.pythonInterpreterPath,
|
|
); err != nil {
|
|
log.Error("failed to patch the plugin sdk", "error", err)
|
|
}
|
|
|
|
// mark the virtual environment as valid if everything goes well
|
|
if err := p.markVirtualEnvironmentAsValid(); err != nil {
|
|
log.Error("failed to mark the virtual environment as valid", "error", err)
|
|
}
|
|
|
|
return nil
|
|
}
|