feat(pods): add tailLines parameter to pod logs retrieval with default 256 lines

Signed-off-by: iamsudip <[email protected]>

Author: iamsudip

pkg/kubernetes/pods.go

@@ -92,17 +92,27 @@ func (k *Kubernetes) PodsDelete(ctx context.Context, namespace, name string) (st
 		k.ResourcesDelete(ctx, &schema.GroupVersionKind{Group: "", Version: "v1", Kind: "Pod"}, namespace, name)
 }
 
-func (k *Kubernetes) PodsLog(ctx context.Context, namespace, name, container string, previous bool) (string, error) {
-	tailLines := int64(256)
+func (k *Kubernetes) PodsLog(ctx context.Context, namespace, name, container string, previous bool, tailLines int64) (string, error) {
 	pods, err := k.manager.accessControlClientSet.Pods(k.NamespaceOrDefault(namespace))
 	if err != nil {
 		return "", err
 	}
-	req := pods.GetLogs(name, &v1.PodLogOptions{
-		TailLines: &tailLines,
+
+	logOptions := &v1.PodLogOptions{
 		Container: container,
 		Previous:  previous,
-	})
+	}
+
+	// Only set tailLines if a value is provided (non-zero)
+	if tailLines > 0 {
+		logOptions.TailLines = &tailLines
+	} else {
+		// Default to 256 lines when not specified
+		defaultLines := int64(256)
+		logOptions.TailLines = &defaultLines
+	}
+
+	req := pods.GetLogs(name, logOptions)
 	res := req.Do(ctx)
 	if res.Error() != nil {
 		return "", res.Error()

pkg/mcp/pods_test.go

@@ -756,6 +756,41 @@ func TestPodsLog(t *testing.T) {
 				return
 			}
 		})
+
+		// Test with tailLines parameter
+		podsTailLines, err := c.callTool("pods_log", map[string]interface{}{
+			"namespace": "ns-1",
+			"name":      "a-pod-in-ns-1",
+			"tailLines": 100,
+		})
+		t.Run("pods_log with tailLines=100 returns pod log", func(t *testing.T) {
+			if err != nil {
+				t.Fatalf("call tool failed %v", err)
+				return
+			}
+			if podsTailLines.IsError {
+				t.Fatalf("call tool failed")
+				return
+			}
+		})
+
+		// Test with invalid tailLines parameter
+		podsInvalidTailLines, _ := c.callTool("pods_log", map[string]interface{}{
+			"namespace": "ns-1",
+			"name":      "a-pod-in-ns-1",
+			"tailLines": "invalid",
+		})
+		t.Run("pods_log with invalid tailLines returns error", func(t *testing.T) {
+			if !podsInvalidTailLines.IsError {
+				t.Fatalf("call tool should fail")
+				return
+			}
+			expectedErrorMsg := "failed to parse tailLines parameter: expected integer"
+			if errMsg := podsInvalidTailLines.Content[0].(mcp.TextContent).Text; !strings.Contains(errMsg, expectedErrorMsg) {
+				t.Fatalf("unexpected error message, expected to contain '%s', got '%s'", expectedErrorMsg, errMsg)
+				return
+			}
+		})
 	})
 }
 

pkg/toolsets/core/pods.go

@@ -201,6 +201,12 @@ func initPods() []api.ServerTool {
 						Type:        "string",
 						Description: "Name of the Pod container to get the logs from (Optional)",
 					},
+					"tailLines": {
+						Type:        "integer",
+						Description: "Number of lines to retrieve from the end of the logs (Optional, default: 256)",
+						Default:     api.ToRawMessage(int64(256)),
+						Minimum:     ptr.To(float64(0)),
+					},
 					"previous": {
 						Type:        "boolean",
 						Description: "Return previous terminated container logs (Optional)",
@@ -396,7 +402,24 @@ func podsLog(params api.ToolHandlerParams) (*api.ToolCallResult, error) {
 	if previous != nil {
 		previousBool = previous.(bool)
 	}
-	ret, err := params.PodsLog(params, ns.(string), name.(string), container.(string), previousBool)
+	// Extract tailLines parameter
+	tailLines := params.GetArguments()["tailLines"]
+	var tailLinesInt int64
+	if tailLines != nil {
+		// Convert to int64 - safely handle both float64 (JSON number) and int types
+		switch v := tailLines.(type) {
+		case float64:
+			tailLinesInt = int64(v)
+		case int:
+			tailLinesInt = int64(v)
+		case int64:
+			tailLinesInt = v
+		default:
+			return api.NewToolCallResult("", fmt.Errorf("failed to parse tailLines parameter: expected integer, got %T", tailLines)), nil
+		}
+	}
+
+	ret, err := params.PodsLog(params.Context, ns.(string), name.(string), container.(string), previousBool, tailLinesInt)
 	if err != nil {
 		return api.NewToolCallResult("", fmt.Errorf("failed to get pod %s log in namespace %s: %v", name, ns, err)), nil
 	} else if ret == "" {