use toolsrv.Dial for local server construction

This commit is contained in:
Levi Neely 2026-07-29 23:29:23 +02:00
parent b2462c6a06
commit 6a6016c5e0
2 changed files with 10 additions and 9 deletions

View File

@ -153,12 +153,16 @@ func (s *Manager) CreateSession(args []string) (string, error) {
} }
} else { } else {
newToolServer = func() toolsrv.Runner { newToolServer = func() toolsrv.Runner {
srv := toolsrv.New(cwd) srv, _ := toolsrv.Dial(s.cfg.Ctx, toolsrv.LocalAddr(cwd), execOpts...)
for _, o := range execOpts { return srv
o(srv)
} }
return srv
}
} }
var promptEnv []string var promptEnv []string

View File

@ -168,10 +168,7 @@ func (s *Manager) restoreSession(ps *agent.PersistedAgent) error {
execOpts = append(execOpts, toolsrv.WithBuiltins(tools.Builtins())) execOpts = append(execOpts, toolsrv.WithBuiltins(tools.Builtins()))
} }
newToolServer = func() toolsrv.Runner { newToolServer = func() toolsrv.Runner {
srv := toolsrv.New(cwd) srv, _ := toolsrv.Dial(s.cfg.Ctx, toolsrv.LocalAddr(cwd), execOpts...)
for _, o := range execOpts {
o(srv)
}
return srv return srv
} }
promptEnv = agent.PromptEnv(cwd) promptEnv = agent.PromptEnv(cwd)