volition-core 0.2.0

Core library for the Volition AI agent
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
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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106

// volition-agent-core/src/strategies/conversation.rs
use super::{DelegationResult, NextStep, Strategy};
use crate::errors::AgentError;
use crate::models::chat::{ApiResponse, ChatMessage};
use crate::UserInteraction;
use anyhow::Result;
use async_trait::async_trait;
use std::fmt;

pub struct ConversationStrategy<UI: UserInteraction + 'static> {
    conversation_history: Vec<ChatMessage>,
    inner_strategy: Box<dyn Strategy<UI> + Send + Sync>,
}

// Manual Debug implementation
impl<UI: UserInteraction + 'static> fmt::Debug for ConversationStrategy<UI> {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        f.debug_struct("ConversationStrategy")
         .field("conversation_history", &self.conversation_history)
         .field("inner_strategy", &self.inner_strategy.name())
         .finish()
    }
}

impl<UI: UserInteraction + 'static> ConversationStrategy<UI> {
    pub fn new(inner_strategy: Box<dyn Strategy<UI> + Send + Sync>) -> Self {
        Self {
            conversation_history: Vec::new(),
            inner_strategy,
        }
    }

    pub fn with_history(
        inner_strategy: Box<dyn Strategy<UI> + Send + Sync>,
        history: Vec<ChatMessage>,
    ) -> Self {
        Self {
            conversation_history: history,
            inner_strategy,
        }
    }
}

#[async_trait]
impl<UI: UserInteraction + 'static> Strategy<UI> for ConversationStrategy<UI> {
    fn name(&self) -> &'static str {
        "Conversation"
    }

    fn initialize_interaction(
        &mut self,
        state: &mut crate::AgentState,
    ) -> Result<NextStep, AgentError> {
        // If conversation_history is not empty, we are continuing a conversation.
        // Replace the state's messages entirely with the stored history.
        if !self.conversation_history.is_empty() {
            state.messages = self.conversation_history.clone();
        } else {
            // If history is empty (first turn), keep the initial message(s)
            // already present in the state (from AgentState::new).
            // We'll update self.conversation_history later in update_history.
        }
        self.inner_strategy.initialize_interaction(state)
    }

    fn process_api_response(
        &mut self,
        state: &mut crate::AgentState,
        response: ApiResponse,
    ) -> Result<NextStep, AgentError> {
        let next_step = self.inner_strategy.process_api_response(state, response)?;
        self.update_history(state);
        Ok(next_step)
    }

    fn process_tool_results(
        &mut self,
        state: &mut crate::AgentState,
        results: Vec<crate::ToolResult>,
    ) -> Result<NextStep, AgentError> {
        let next_step = self.inner_strategy.process_tool_results(state, results)?;
        self.update_history(state);
        Ok(next_step)
    }

    fn process_delegation_result(
        &mut self,
        state: &mut crate::AgentState,
        result: DelegationResult,
    ) -> Result<NextStep, AgentError> {
        let next_step = self.inner_strategy.process_delegation_result(state, result)?;
        self.update_history(state);
        Ok(next_step)
    }
}

impl<UI: UserInteraction + 'static> ConversationStrategy<UI> {
    // Simplified history update
    fn update_history(&mut self, state: &crate::AgentState) {
        self.conversation_history = state.messages.clone();
    }

    pub fn get_history(&self) -> &Vec<ChatMessage> {
        &self.conversation_history
    }
}