Use MCP_TRANSPORT environment variable

Replace TRANSPORT with MCP_TRANSPORT for transport type configuration.
This enables out-of-the-box compatibility with ToolHive, which sets
MCP_TRANSPORT as the default environment variable for MCP servers.

Resolves #53

Signed-off-by: Syed Azeez <[email protected]>

Author: syedazeez337

cmd/yardstick-client/README.md

@@ -31,8 +31,8 @@ The client connects to MCP servers using various transport mechanisms and allows
 ## Environment Variables
 
 Configuration can also be set via environment variables:
-- `TRANSPORT`: Override transport type
-- `ADDRESS`: Override server address  
+- `MCP_TRANSPORT`: Override transport type
+- `ADDRESS`: Override server address
 - `PORT`: Override server port
 - `COMMAND`: Override command for stdio transport
 
@@ -101,7 +101,7 @@ Call a specific tool with provided JSON arguments:
 
 ### Using environment variables
 ```bash
-export TRANSPORT=sse
+export MCP_TRANSPORT=sse
 export ADDRESS=remote.example.com
 export PORT=3000
 ./client -action=list-resources

cmd/yardstick-client/main.go

@@ -211,7 +211,7 @@ func parseConfig() Config {
 	}
 
 	// Use environment variables if provided
-	if t, ok := os.LookupEnv("TRANSPORT"); ok {
+	if t, ok := os.LookupEnv("MCP_TRANSPORT"); ok {
 		config.Transport = t
 	}
 	if a, ok := os.LookupEnv("ADDRESS"); ok {

cmd/yardstick-client/main_test.go

@@ -39,7 +39,7 @@ func TestConfig_ParseConfig(t *testing.T) {
 
 	// Save original env vars
 	originalEnvVars := map[string]string{}
-	envVars := []string{"TRANSPORT", "ADDRESS", "PORT", "COMMAND"}
+	envVars := []string{"MCP_TRANSPORT", "ADDRESS", "PORT", "COMMAND"}
 	for _, env := range envVars {
 		if val, exists := os.LookupEnv(env); exists {
 			originalEnvVars[env] = val
@@ -59,20 +59,20 @@ func TestConfig_ParseConfig(t *testing.T) {
 	// Test environment variable override
 	t.Run("environment variables", func(t *testing.T) {
 		// Set test env vars
-		os.Setenv("TRANSPORT", "streamable-http")
+		os.Setenv("MCP_TRANSPORT", "streamable-http")
 		os.Setenv("ADDRESS", "remote.example.com")
 		os.Setenv("PORT", "3000")
 		os.Setenv("COMMAND", "test-command")
 
 		defer func() {
-			os.Unsetenv("TRANSPORT")
+			os.Unsetenv("MCP_TRANSPORT")
 			os.Unsetenv("ADDRESS")
 			os.Unsetenv("PORT")
 			os.Unsetenv("COMMAND")
 		}()
 
 		// Test that env vars would be read (we can't call parseConfig due to flag redefinition)
-		transport, transportExists := os.LookupEnv("TRANSPORT")
+		transport, transportExists := os.LookupEnv("MCP_TRANSPORT")
 		assert.True(t, transportExists)
 		assert.Equal(t, "streamable-http", transport)
 

cmd/yardstick-server/README.md

@@ -64,12 +64,12 @@ docker run -it ghcr.io/stackloklabs/yardstick/server
 
 **SSE Transport:**
 ```bash
-docker run -p 8080:8080 -e TRANSPORT=sse -e PORT=8080 ghcr.io/stackloklabs/yardstick/server
+docker run -p 8080:8080 -e MCP_TRANSPORT=sse -e PORT=8080 ghcr.io/stackloklabs/yardstick/server
 ```
 
 **Streamable HTTP Transport:**
 ```bash
-docker run -p 8080:8080 -e TRANSPORT=streamable-http -e PORT=8080 ghcr.io/stackloklabs/yardstick/server
+docker run -p 8080:8080 -e MCP_TRANSPORT=streamable-http -e PORT=8080 ghcr.io/stackloklabs/yardstick/server
 ```
 
 ## Tools

cmd/yardstick-server/main.go

@@ -142,7 +142,7 @@ func parseConfig() {
 	flag.Parse()
 
 	// Use environment variables if provided, otherwise use flag values
-	if t, ok := os.LookupEnv("TRANSPORT"); ok {
+	if t, ok := os.LookupEnv("MCP_TRANSPORT"); ok {
 		transport = t
 	}
 	if p, ok := os.LookupEnv("PORT"); ok {