systemprompt-agent 0.2.0

Core Agent protocol module for systemprompt.io
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

use systemprompt_models::{AiMessage, AiRequest};

use super::super::ExecutionContext;

pub fn build_ai_request(context: &ExecutionContext, messages: Vec<AiMessage>) -> AiRequest {
    let tool_config = context.request_ctx.tool_model_config();

    let provider = tool_config
        .and_then(|c| c.provider.as_deref())
        .or(context.agent_runtime.provider.as_deref())
        .unwrap_or_else(|| context.ai_service.default_provider());
    let model = tool_config
        .and_then(|c| c.model.as_deref())
        .or(context.agent_runtime.model.as_deref())
        .unwrap_or_else(|| context.ai_service.default_model());
    let max_output_tokens = tool_config
        .and_then(|c| c.max_output_tokens)
        .or(context.agent_runtime.max_output_tokens)
        .unwrap_or_else(|| context.ai_service.default_max_output_tokens());

    if tool_config.is_some() {
        tracing::debug!(
            provider,
            model,
            max_output_tokens,
            "Using tool_model_config in planned strategy"
        );
    }

    AiRequest::builder(
        messages,
        provider,
        model,
        max_output_tokens,
        context.request_ctx.clone(),
    )
    .build()
}