lariv-rs 0.1.0

Compile-time plugin web application framework built on Axum, SeaORM, Maud, and HTMX
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

//! Part kind strings + IsPartType matching (concrete payloads before text catch-all).

use crate::genai::{Part, part_is_empty};

pub const KIND_INLINE_DATA: &str = "inlineData";
pub const KIND_FILE_DATA: &str = "fileData";
pub const KIND_FUNCTION_CALL: &str = "functionCall";
pub const KIND_FUNCTION_RESPONSE: &str = "functionResponse";
pub const KIND_EXECUTABLE_CODE: &str = "executableCode";
pub const KIND_CODE_EXECUTION_RESULT: &str = "codeExecutionResult";
pub const KIND_TOOL_CALL: &str = "toolCall";
pub const KIND_TOOL_RESPONSE: &str = "toolResponse";
pub const KIND_MEDIA_RESOLUTION: &str = "mediaResolution";
pub const KIND_TEXT: &str = "text";

/// Classify a part into a stored kind. Classification order checks before text catch-all.
pub fn classify_part_kind(part: &Part) -> Option<&'static str> {
    if part.inline_data.is_some() {
        return Some(KIND_INLINE_DATA);
    }
    if part.function_response.is_some() {
        return Some(KIND_FUNCTION_RESPONSE);
    }
    if part.media_resolution.is_some() {
        return Some(KIND_MEDIA_RESOLUTION);
    }
    if part.code_execution_result.is_some() {
        return Some(KIND_CODE_EXECUTION_RESULT);
    }
    if part.executable_code.is_some() {
        return Some(KIND_EXECUTABLE_CODE);
    }
    if part.tool_call.is_some() {
        return Some(KIND_TOOL_CALL);
    }
    if part.file_data.is_some() {
        return Some(KIND_FILE_DATA);
    }
    if part.function_call.is_some() {
        return Some(KIND_FUNCTION_CALL);
    }
    if part.tool_response.is_some() {
        return Some(KIND_TOOL_RESPONSE);
    }
    // text catch-all (Go IsPartType)
    if part.inline_data.is_none()
        && part.file_data.is_none()
        && part.function_call.is_none()
        && part.function_response.is_none()
        && part.code_execution_result.is_none()
        && part.executable_code.is_none()
        && part.media_resolution.is_none()
        && part.tool_call.is_none()
        && part.tool_response.is_none()
        && (!part.text.as_deref().unwrap_or("").is_empty()
            || part.thought
            || !part.thought_signature.as_deref().unwrap_or("").is_empty()
            || part_is_empty(part))
    {
        return Some(KIND_TEXT);
    }
    None
}