rskit-agent 0.2.0-alpha.1

Agentic loop — Provider + Tools + Hooks in a turn-based execution engine
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

//! Completion request construction for agent turns.

use rskit_llm::types::{CompletionRequest, Message, ToolDefinition};
use rskit_tool::Registry;

use crate::config::AgentConfig;

pub(crate) fn build_completion_request(
    config: &AgentConfig,
    messages: &[Message],
) -> CompletionRequest {
    CompletionRequest {
        model: config.model.clone(),
        messages: messages.to_vec(),
        max_tokens: None,
        temperature: None,
        stream: false,
        tools: config
            .tools
            .as_ref()
            .map(|registry| tool_definitions(registry.as_ref())),
        tool_choice: None,
    }
}

fn tool_definitions(registry: &Registry) -> Vec<ToolDefinition> {
    registry
        .list()
        .into_iter()
        .map(|tool| ToolDefinition {
            name: tool.name,
            description: tool.description,
            input_schema: tool.input_schema,
            output_schema: tool.output_schema,
        })
        .collect()
}