Add keep-alive stdin for daemons

Author: ibuildthecloud

pkg/engine/daemon.go

@@ -100,6 +100,12 @@ func (e *Engine) startDaemon(_ context.Context, tool types.Tool) (string, error)
 		return url, err
 	}
 
+	r, w, err := os.Pipe()
+	if err != nil {
+		return "", err
+	}
+
+	cmd.Stdin = r
 	cmd.Stderr = os.Stderr
 	cmd.Stdout = os.Stdout
 	log.Infof("launched [%s][%s] port [%d] %v", tool.Parameters.Name, tool.ID, port, cmd.Args)
@@ -121,6 +127,8 @@ func (e *Engine) startDaemon(_ context.Context, tool types.Tool) (string, error)
 		if err != nil {
 			log.Errorf("daemon exited tool [%s] %v: %v", tool.Parameters.Name, cmd.Args, err)
 		}
+		_ = r.Close()
+		_ = w.Close()
 
 		cancel(err)
 		stop()
@@ -133,7 +141,7 @@ func (e *Engine) startDaemon(_ context.Context, tool types.Tool) (string, error)
 	daemonWG.Add(1)
 	context.AfterFunc(ctx, func() {
 		if err := cmd.Process.Kill(); err != nil {
-			log.Errorf("daemon failed to kill tool [%s] process: %v", tool.Parameters.Name, err)
+			log.Debugf("daemon failed to kill tool [%s] process: %v", tool.Parameters.Name, err)
 		}
 		daemonWG.Done()
 	})

pkg/engine/engine.go

@@ -155,14 +155,18 @@ func (c *Context) appendTool(completion *types.CompletionRequest, parentTool typ
 		c.toolNames = map[string]struct{}{}
 	}
 
-	completion.Tools = append(completion.Tools, types.CompletionTool{
-		Function: types.CompletionFunctionDefinition{
-			ToolID:      subTool.ID,
-			Name:        PickToolName(subToolName, c.toolNames),
-			Description: subTool.Parameters.Description,
-			Parameters:  args,
-		},
-	})
+	if subTool.Instructions == "" {
+		log.Debugf("Skipping zero instruction tool %s (%s)", subToolName, subTool.ID)
+	} else {
+		completion.Tools = append(completion.Tools, types.CompletionTool{
+			Function: types.CompletionFunctionDefinition{
+				ToolID:      subTool.ID,
+				Name:        PickToolName(subToolName, c.toolNames),
+				Description: subTool.Parameters.Description,
+				Parameters:  args,
+			},
+		})
+	}
 
 	for _, export := range subTool.Export {
 		if err := c.appendTool(completion, subTool, export); err != nil {