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
64
65
66
67
68
69
70
71
72
73

//! `list_chat_attachments` — list VNodes attached on messages in the current conversation.

use async_trait::async_trait;
use serde_json::{Value, json};

use crate::{
    llm_tools::{LlmTool, ToolCtx},
    plugins::{
        filesystem::node,
        llm_assistant::{chat_attachments, genai::FunctionDeclaration},
    },
};

pub struct ListChatAttachmentsTool;

#[async_trait]
impl LlmTool for ListChatAttachmentsTool {
    fn name(&self) -> &str {
        "list_chat_attachments"
    }

    fn declaration(&self) -> FunctionDeclaration {
        FunctionDeclaration {
            name: "list_chat_attachments".into(),
            description: "List filesystem VNodes attached to messages in the current conversation \
                (device uploads and files selected from the filesystem). Returns each \
                file's id, name, and absolute VNode path. Use when you need vnode ids or \
                paths to pass to read_file or other tools."
                .into(),
            parameters: Some(json!({
                "type": "object",
                "properties": {}
            })),
        }
    }

    async fn run(&self, ctx: &ToolCtx<'_>, _args: Value) -> Result<Value, String> {
        let Some(session_id) = ctx.session_id.filter(|id| *id > 0) else {
            return Err("no active conversation session".into());
        };
        let refs = chat_attachments::list_session_attachment_refs(ctx.db, session_id)
            .await
            .map_err(|e| e.to_string())?;

        let mut out = Vec::with_capacity(refs.len());
        for att in refs {
            let Some(vnode) = node::get_by_id(ctx.db, att.vnode_id)
                .await
                .map_err(|e| e.to_string())?
            else {
                out.push(json!({
                    "id": att.vnode_id,
                    "name": att.display_name,
                    "path": Value::Null,
                    "missing": true,
                }));
                continue;
            };
            let path = node::get_path(ctx.db, &vnode).await;
            let name = if att.display_name.is_empty() {
                vnode.name
            } else {
                att.display_name
            };
            out.push(json!({
                "id": vnode.id,
                "name": name,
                "path": path,
            }));
        }
        Ok(json!({ "attachments": out }))
    }
}