Fix streamable http sampling

client/client.go

@@ -478,6 +478,22 @@ func (c *Client) handleSamplingRequestTransport(ctx context.Context, request tra
 			return nil, fmt.Errorf("failed to unmarshal params: %w", err)
 		}
 	}
+
+	// Fix content parsing - HTTP transport unmarshals TextContent as map[string]any
+	for i := range params.Messages {
+		if contentMap, ok := params.Messages[i].Content.(map[string]any); ok {
+			if textType, exists := contentMap["type"]; exists && textType == "text" {
+				if text, exists := contentMap["text"]; exists {
+					if textStr, ok := text.(string); ok {
+						params.Messages[i].Content = mcp.TextContent{
+							Type: "text",
+							Text: textStr,
+						}
+					}
+				}
+			}
+		}
+	}
 
 	// Create the MCP request
 	mcpRequest := mcp.CreateMessageRequest{

client/transport/streamable_http.go

@@ -601,10 +601,8 @@ func (c *StreamableHTTP) IsOAuthEnabled() bool {
 func (c *StreamableHTTP) listenForever(ctx context.Context) {
 	c.logger.Infof("listening to server forever")
 	for {
-		// Add timeout for individual connection attempts
-		connectCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
-		err := c.createGETConnectionToServer(connectCtx)
-		cancel()
+		// Use the original context for continuous listening - no timeout
+		err := c.createGETConnectionToServer(ctx)
 
 		if errors.Is(err, ErrGetMethodNotAllowed) {
 			// server does not support listening