diff --git a/pkg/agent/loop.go b/pkg/agent/loop.go index 4f3a81b6a..49042d9f1 100644 --- a/pkg/agent/loop.go +++ b/pkg/agent/loop.go @@ -108,7 +108,7 @@ func registerSharedTools(cfg *config.Config, msgBus *bus.MessageBus, registry *A // Vision tool if cfg.Tools.Vision.Enabled { - agent.Tools.Register(tools.NewAnalyzeImageTool(cfg.Tools.Vision)) + agent.Tools.Register(tools.NewAnalyzeImageTool(cfg.Tools.Vision, cfg.WorkspacePath(), cfg.Agents.Defaults.RestrictToWorkspace)) } // Message tool diff --git a/pkg/tools/vision.go b/pkg/tools/vision.go index 5b5ff394f..daeee23de 100644 --- a/pkg/tools/vision.go +++ b/pkg/tools/vision.go @@ -31,10 +31,10 @@ type AnalyzeImageTool struct { restrict bool } -func NewAnalyzeImageTool(opts config.VisionToolConfig) *AnalyzeImageTool { +func NewAnalyzeImageTool(opts config.VisionToolConfig, workspace string, restrict bool) *AnalyzeImageTool { return &AnalyzeImageTool{ - workspace: opts.Workspace, - restrict: opts.Restrict, + workspace: workspace, + restrict: restrict, apiKey: opts.ApiKey, apiURL: opts.ApiURL, model: opts.Model,