jarvish 1.0.1

Next Generation AI Integrated Shell inspired by J.A.R.V.I.S. on Marvel's Iron Man
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82

//! エラー調査フロー
//!
//! コマンドが異常終了した場合に、AI にエラーの原因と修正案を調査させる。
//! Tool Call からの失敗は自動調査、ユーザー直接入力は確認プロンプト後に調査する。

use std::sync::atomic::Ordering;

use tracing::{info, warn};

use crate::ai::AiResponse;
use crate::cli::jarvis::{jarvis_ask_investigate, jarvis_notice};
use crate::engine::{execute, CommandResult, LoopAction};

use super::Shell;

impl Shell {
    /// コマンドが異常終了した場合にエラー調査を実行する。
    ///
    /// - `from_tool_call`: true の場合、ユーザー確認なしで自動調査
    /// - `from_tool_call`: false の場合、確認プロンプト後に調査
    pub(super) async fn investigate_error(
        &mut self,
        line: &str,
        result: &CommandResult,
        from_tool_call: bool,
    ) {
        let ai = match self.ai_client {
            Some(ref ai) => ai,
            None => return,
        };

        // 調査開始の判定
        let should_investigate = if from_tool_call {
            info!("Tool Call command failed, auto-investigating");
            true
        } else {
            jarvis_ask_investigate(result.exit_code)
        };

        if !should_investigate {
            return;
        }

        // BlackBox から最新コンテキストを取得（失敗したコマンドも含む）
        let context = self
            .black_box
            .as_ref()
            .and_then(|bb| bb.get_recent_context(5).ok())
            .unwrap_or_default();

        match ai.investigate_error(line, result, &context).await {
            Ok(conv_result) => match conv_result.response {
                AiResponse::Command(ref fix_cmd) => {
                    // AI が修正コマンドを提案 → 実行
                    jarvis_notice(fix_cmd);
                    let fix_result = execute(fix_cmd);
                    self.last_exit_code
                        .store(fix_result.exit_code, Ordering::Relaxed);
                    println!();

                    // 修正コマンドの結果も履歴に記録
                    if fix_result.action == LoopAction::Continue {
                        if let Some(ref bb) = self.black_box {
                            if let Err(e) = bb.record(fix_cmd, &fix_result) {
                                warn!("Failed to record fix command history: {e}");
                            }
                        }
                    }
                }
                AiResponse::NaturalLanguage(_) => {
                    // 会話コンテキストを保存
                    self.conversation_state = Some(conv_result.conversation);
                    // AI が自然言語で説明 → ストリーミング表示済み
                    println!();
                }
            },
            Err(e) => {
                warn!(error = %e, "Error investigation failed");
            }
        }
    }
}