kcode-k1-chat-thread-actions 0.2.0

Synchronous CurrentTime dispatch for K1 chat threads
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

use kcode_k1_chat_chatend::{DispatchCall, DispatchOutcome};
use serde_json::Value;

const INVALID_CURRENT_TIME_ARGUMENTS: &str = "invalid CurrentTime arguments";
const UNKNOWN_TOOL: &str = "unknown Ktool";

pub fn dispatch(calls: &[DispatchCall]) -> Vec<DispatchOutcome> {
    calls.iter().map(dispatch_call).collect()
}

fn dispatch_call(call: &DispatchCall) -> DispatchOutcome {
    let result = match call.name.as_str() {
        "CurrentTime" if is_empty_object(&call.arguments) => {
            Ok(kcode_k1_ktool_current_time::current_time())
        }
        "CurrentTime" => Err(INVALID_CURRENT_TIME_ARGUMENTS.to_owned()),
        _ => Err(UNKNOWN_TOOL.to_owned()),
    };
    DispatchOutcome::Terminal(result)
}

fn is_empty_object(arguments: &str) -> bool {
    matches!(
        serde_json::from_str::<Value>(arguments),
        Ok(Value::Object(fields)) if fields.is_empty()
    )
}

#[cfg(test)]
mod tests;