fix: timeout

Author: chebread

cmd/main.go

@@ -95,6 +95,8 @@ func main() {
 	flag.StringVar(&opts.tags, "tags", "", "Build tags to pass to `go run` capture phase (e.g. 'mock')")
 	flag.Parse()
 
+	fmt.Printf("[DEBUG] Parsed Timeout: %v\n", opts.timeout)
+
 	if showVersion {
 		fmt.Printf("lx %s\n", version)
 		return
@@ -873,6 +875,9 @@ func diagnoseLLMError(err error) string {
 	msg := err.Error()
 
 	switch {
+	case strings.Contains(msg, "timeout reached"):
+		return fmt.Sprintf("TIMEOUT: The operation exceeded the time limit. (%s)", msg)
+
 	case strings.Contains(msg, "API_KEY_INVALID"):
 		return "The API key is incorrect. Please double-check the api_key in 'lx-config.yaml'."
 	case strings.Contains(msg, "quota"):
@@ -883,6 +888,7 @@ func diagnoseLLMError(err error) string {
 		return "Your response has been blocked by security policy. Please edit the prompt."
 	case strings.Contains(msg, "connection") || strings.Contains(msg, "timeout"):
 		return "The network connection is unstable. Please check your Internet connection."
+
 	default:
 		return fmt.Sprintf("An unknown error has occurred: %v", err)
 	}
@@ -962,12 +968,21 @@ func (c *commandLLM) Generate(ctx context.Context, model string, prompt string)
 
 	cmd := exec.CommandContext(ctx, c.binPath, finalArgs...)
 
+	cmd.SysProcAttr = &syscall.SysProcAttr{Setpgid: true}
+
+	cmd.Cancel = func() error {
+		return syscall.Kill(-cmd.Process.Pid, syscall.SIGKILL)
+	}
+
 	var out bytes.Buffer
 	var stderr bytes.Buffer
 	cmd.Stdout = &out
 	cmd.Stderr = &stderr
 
 	if err := cmd.Run(); err != nil {
+		if ctx.Err() == context.DeadlineExceeded {
+			return "", fmt.Errorf("timeout reached (%s): process group killed", ctx.Err())
+		}
 		return "", fmt.Errorf("command execution failed: %v\nStderr: %s", err, stderr.String())
 	}