oy-cli 0.10.6

Local AI coding CLI for inspecting, editing, running commands, and auditing repositories
Documentation 
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

//! Structured reasoning tool for step-by-step problem solving.
//!
//! Allows the model to record explicit reasoning steps with numbered thoughts,
//! revisions, and branching comparisons.

use anyhow::Result;
use serde::Serialize;
use serde_json::Value;

use super::args::ThinkArgs;
use super::ToolContext;

#[derive(Debug, Clone, Serialize)]
pub(super) struct Thought {
    pub number: usize,
    pub content: String,
    pub mode: String,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub revises_thought: Option<usize>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub branch_from_thought: Option<usize>,
    #[serde(skip_serializing_if = "Option::is_none")]
    pub branch_id: Option<String>,
}

#[derive(Debug, Serialize)]
pub(super) struct ThinkOutput {
    pub thought: Thought,
    pub total_thoughts: usize,
    pub next_thought_needed: bool,
}

pub(super) fn tool_think(_ctx: &mut ToolContext, args: ThinkArgs) -> Result<Value> {
    let thought = Thought {
        number: args.thought_number,
        content: args.thought,
        mode: args.mode,
        revises_thought: args.revises_thought,
        branch_from_thought: args.branch_from_thought,
        branch_id: args.branch_id,
    };

    let output = ThinkOutput {
        thought,
        total_thoughts: args.total_thoughts,
        next_thought_needed: args.next_thought_needed,
    };

    Ok(serde_json::to_value(output)?)
}