425d873 · crush · git.iko.soy

on main help sign in

fix(bash): use model name instead of ID

425d873Amolith1 year ago

internal/agent/common_test.go+8−1

@@ -175,8 +175,15 @@ func coderAgent(r *recorder.Recorder, env fakeEnv, large, small fantasy.Language

175 if err != nil {

176 return nil, err

177 }

178+

179+ // Get the model name for the bash tool

180+ modelName := large.Model() // fallback to ID if Name not available

181+ if model := cfg.GetModel(large.Provider(), large.Model()); model != nil {

182+ modelName = model.Name

183+ }

184+

178 allTools := []fantasy.AgentTool{

185 allTools := []fantasy.AgentTool{

179− tools.NewBashTool(env.permissions, env.workingDir, cfg.Options.Attribution, large.Model()),

186+ tools.NewBashTool(env.permissions, env.workingDir, cfg.Options.Attribution, modelName),

180 tools.NewDownloadTool(env.permissions, env.workingDir, r.GetDefaultClient()),

187 tools.NewDownloadTool(env.permissions, env.workingDir, r.GetDefaultClient()),

181 tools.NewEditTool(env.lspClients, env.permissions, env.history, env.workingDir),

188 tools.NewEditTool(env.lspClients, env.permissions, env.history, env.workingDir),

182 tools.NewMultiEditTool(env.lspClients, env.permissions, env.history, env.workingDir),

189 tools.NewMultiEditTool(env.lspClients, env.permissions, env.history, env.workingDir),

internal/agent/coordinator.go+3−1

@@ -330,7 +330,9 @@ func (c *coordinator) buildTools(ctx context.Context, agent config.Agent) ([]fan

330 // Get the model name for the agent

331 modelName := ""

332 if modelCfg, ok := c.cfg.Models[agent.Model]; ok {

333+ if model := c.cfg.GetModel(modelCfg.Provider, modelCfg.Model); model != nil {

333− modelName = modelCfg.Model

334+ modelName = model.Name

335+ }

334 }

336 }

335

337

336 allTools = append(allTools,

338 allTools = append(allTools,