Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions cmd/yardstick-client/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion cmd/yardstick-client/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -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 {
Expand Down
8 changes: 4 additions & 4 deletions cmd/yardstick-client/main_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand All @@ -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)

Expand Down
4 changes: 2 additions & 2 deletions cmd/yardstick-server/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down
2 changes: 1 addition & 1 deletion cmd/yardstick-server/main.go
Original file line number Diff line number Diff line change
Expand Up @@ -165,7 +165,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 {
Expand Down