Add proxy_mode field to Workload API output (#1846)

- Add ProxyMode field to core.Workload struct
- Load proxy_mode from RunConfig for container workloads
- Include proxy_mode in remote workload creation
- Output proxy_mode in thv list --format json

This allows clients to see whether a workload uses SSE or
streamable-http proxy mode, which is important for proper
client configuration.

Signed-off-by: Juan Antonio Osorio <[email protected]>
Co-authored-by: Don Browne <[email protected]>

Author: JAORMX

docs/server/docs.go

@@ -27,6 +27,8 @@ type Workload struct {
 	ToolType string `json:"tool_type"`
 	// TransportType is the type of transport used for this workload.
 	TransportType types.TransportType `json:"transport_type"`
+	// ProxyMode is the proxy mode for stdio transport (sse or streamable-http).
+	ProxyMode string `json:"proxy_mode,omitempty"`
 	// Status is the current status of the workload.
 	Status runtime.WorkloadStatus `json:"status"`
 	// StatusContext provides additional context about the workload's status.

docs/server/swagger.json

@@ -1160,6 +1160,7 @@ func (d *defaultManager) getRemoteWorkloadsFromState(
 			URL:           runConfig.RemoteURL,
 			Port:          runConfig.Port,
 			TransportType: transportType,
+			ProxyMode:     string(runConfig.ProxyMode),
 			ToolType:      "remote",
 			Group:         runConfig.Group,
 			CreatedAt:     workloadStatus.CreatedAt,

docs/server/swagger.yaml

@@ -14,14 +14,15 @@ import (
 	"github.com/stacklok/toolhive/pkg/transport/types"
 )
 
-// minimalRunConfig represents just the group field from a run configuration
+// minimalRunConfig represents just the fields we need from a run configuration
 type minimalRunConfig struct {
-	Group string `json:"group,omitempty" yaml:"group,omitempty"`
+	Group     string `json:"group,omitempty" yaml:"group,omitempty"`
+	ProxyMode string `json:"proxy_mode,omitempty" yaml:"proxy_mode,omitempty"`
 }
 
-// loadGroupFromRunConfig attempts to load group information from the runconfig
-// Returns empty string if runconfig doesn't exist or doesn't have group info
-func loadGroupFromRunConfig(ctx context.Context, name string) (string, error) {
+// loadRunConfigFields attempts to load specific fields from the runconfig
+// Returns empty struct if runconfig doesn't exist
+func loadRunConfigFields(ctx context.Context, name string) (*minimalRunConfig, error) {
 	// Try to load the runconfig
 	runConfig, err := state.LoadRunConfig(ctx, name, func(r io.Reader) (*minimalRunConfig, error) {
 		var config minimalRunConfig
@@ -33,13 +34,13 @@ func loadGroupFromRunConfig(ctx context.Context, name string) (string, error) {
 	})
 	if err != nil {
 		if errors.IsRunConfigNotFound(err) {
-			return "", nil
+			return &minimalRunConfig{}, nil
 		}
-		return "", err
+		return nil, err
 	}
 
-	// Return the group from the runconfig
-	return runConfig.Group, nil
+	// Return the runconfig
+	return runConfig, nil
 }
 
 // WorkloadFromContainerInfo creates a Workload struct from the runtime container info.
@@ -89,7 +90,7 @@ func WorkloadFromContainerInfo(container *runtime.ContainerInfo) (core.Workload,
 	}
 
 	ctx := context.Background()
-	groupName, err := loadGroupFromRunConfig(ctx, name)
+	runConfig, err := loadRunConfigFields(ctx, name)
 	if err != nil {
 		return core.Workload{}, err
 	}
@@ -101,11 +102,12 @@ func WorkloadFromContainerInfo(container *runtime.ContainerInfo) (core.Workload,
 		URL:           url,
 		ToolType:      toolType,
 		TransportType: tType,
+		ProxyMode:     runConfig.ProxyMode,
 		Status:        container.State,
 		StatusContext: container.Status,
 		CreatedAt:     container.Created,
 		Port:          port,
 		Labels:        userLabels,
-		Group:         groupName,
+		Group:         runConfig.Group,
 	}, nil
 }