Merge pull request #680 from helixml/feature/helix_mcp_proxy

Model Context Protocol (MCP) support

api/cmd/helix/root.go

@@ -10,6 +10,7 @@ import (
 	"github.com/helixml/helix/api/pkg/cli/app"
 	"github.com/helixml/helix/api/pkg/cli/fs"
 	"github.com/helixml/helix/api/pkg/cli/knowledge"
+	"github.com/helixml/helix/api/pkg/cli/mcp"
 	"github.com/helixml/helix/api/pkg/cli/secret"
 )
 
@@ -33,6 +34,7 @@ func NewRootCmd() *cobra.Command {
 	RootCmd.AddCommand(fs.New())
 	RootCmd.AddCommand(fs.NewUploadCmd()) // Shortcut for upload
 	RootCmd.AddCommand(secret.New())
+	RootCmd.AddCommand(mcp.New())
 
 	// Commands available on all platforms
 	RootCmd.AddCommand(newServeCmd())

api/pkg/cli/knowledge/search.go

@@ -0,0 +1,54 @@
+package knowledge
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/helixml/helix/api/pkg/client"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	searchCmd.Flags().String("app", "", "App ID to search within")
+	searchCmd.Flags().String("knowledge", "", "Knowledge ID to search within")
+	searchCmd.Flags().String("prompt", "", "Search prompt")
+
+	_ = searchCmd.MarkFlagRequired("app")
+	_ = searchCmd.MarkFlagRequired("prompt")
+
+	rootCmd.AddCommand(searchCmd)
+}
+
+var searchCmd = &cobra.Command{
+	Use:   "search",
+	Short: "Search knowledge",
+	Long:  `Search through knowledge using a prompt`,
+	RunE: func(cmd *cobra.Command, _ []string) error {
+		apiClient, err := client.NewClientFromEnv()
+		if err != nil {
+			return err
+		}
+
+		appID, _ := cmd.Flags().GetString("app")
+		knowledgeID, _ := cmd.Flags().GetString("knowledge")
+		prompt, _ := cmd.Flags().GetString("prompt")
+
+		results, err := apiClient.SearchKnowledge(cmd.Context(), &client.KnowledgeSearchQuery{
+			AppID:       appID,
+			KnowledgeID: knowledgeID,
+			Prompt:      prompt,
+		})
+		if err != nil {
+			return fmt.Errorf("failed to search knowledge: %w", err)
+		}
+
+		jsonBytes, err := json.MarshalIndent(results, "", "  ")
+		if err != nil {
+			return fmt.Errorf("failed to marshal results to JSON: %w", err)
+		}
+
+		fmt.Println(string(jsonBytes))
+
+		return nil
+	},
+}

api/pkg/cli/mcp/cmd.go

@@ -0,0 +1,19 @@
+package mcp
+
+import (
+	"github.com/spf13/cobra"
+)
+
+var rootCmd = &cobra.Command{
+	Use:     "mcp",
+	Short:   "Helix model context protocol proxy",
+	Aliases: []string{"mcp"},
+	Long:    `TODO`,
+	Run: func(*cobra.Command, []string) {
+		// Do Stuff Here
+	},
+}
+
+func New() *cobra.Command {
+	return rootCmd
+}

api/pkg/cli/mcp/mcp_configure.go

@@ -0,0 +1,3 @@
+package mcp
+
+// TODO: add a command that configures the mcp server in the claude config file