synth-ai-core 0.3.0

Rust core for Synth AI SDKs - API client, streaming, tracing, and tunnels
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
74
75
76
77
78
79
80
81
82
83
84
85
86

//! Local API client for container communication.
//!
//! This module provides the client for communicating with containers:
//! - HTTP client with authentication
//! - Data contracts (request/response types)
//! - Health checking and waiting
//! - Rollout execution

pub mod auth;
pub mod client;
pub mod datasets;
pub mod health;
pub mod helpers;
pub mod llm_guards;
pub mod override_helpers;
pub mod proxy;
pub mod rollout_helpers;
pub mod trace_helpers;
pub mod tracing_utils;
pub mod types;
pub mod validation;
pub mod validators;
pub mod vendors;

pub use auth::{
    allowed_environment_api_keys, encrypt_for_backend, ensure_container_auth,
    is_api_key_header_authorized, mint_environment_api_key, normalize_environment_api_key,
    setup_environment_api_key, DEV_ENVIRONMENT_API_KEY_NAME, ENVIRONMENT_API_KEY_ALIASES_NAME,
    ENVIRONMENT_API_KEY_NAME, MAX_ENVIRONMENT_API_KEY_BYTES, SEALED_BOX_ALGORITHM,
};
pub use client::{EnvClient, ContainerClient};
pub use datasets::{ensure_split, normalise_seed, TaskDatasetSpec};
pub use health::container_health;
pub use helpers::{
    extract_api_key, get_default_max_completion_tokens, normalize_chat_completion_url,
    parse_tool_calls_from_response,
};
pub use llm_guards::is_direct_provider_call;
pub use override_helpers::{
    apply_context_overrides, get_agent_skills_path, get_applied_env_vars, MAX_ENV_VARS,
    MAX_ENV_VAR_VALUE_LENGTH, MAX_FILES_PER_OVERRIDE, MAX_FILE_SIZE_BYTES, MAX_TOTAL_SIZE_BYTES,
    PREFLIGHT_SCRIPT_TIMEOUT_SECONDS,
};
pub use proxy::{
    extract_message_text, inject_system_hint, normalize_response_format_for_groq,
    parse_tool_call_from_text, prepare_for_groq, prepare_for_openai,
    synthesize_tool_call_if_missing,
};
pub use rollout_helpers::build_rollout_response;
pub use trace_helpers::{
    build_trace_payload, build_trajectory_trace, extract_trace_correlation_id,
    include_event_history_in_response, include_event_history_in_trajectories,
    include_trace_correlation_id_in_response, validate_trace_correlation_id,
    verify_trace_correlation_id_in_response,
};
pub use tracing_utils::{
    resolve_sft_output_dir, resolve_tracing_db_url, tracing_env_enabled, unique_sft_path,
};
pub use types::{
    AuthInfo, DatasetInfo, HealthResponse, InferenceInfo, InfoResponse, LimitsInfo, RolloutEnvSpec,
    RolloutMetrics, RolloutPolicySpec, RolloutRequest, RolloutResponse, RolloutSafetyConfig,
    TaskDescriptor, TaskInfo,
};
pub use validation::{
    validate_artifact_size, validate_artifacts_list, validate_context_overrides,
    validate_context_snapshot, MAX_ARTIFACTS_PER_ROLLOUT, MAX_ARTIFACT_CONTENT_TYPE_LENGTH,
    MAX_ARTIFACT_METADATA_BYTES, MAX_CONTEXT_OVERRIDES_PER_ROLLOUT, MAX_CONTEXT_SNAPSHOT_BYTES,
    MAX_INLINE_ARTIFACT_BYTES, MAX_TOTAL_INLINE_ARTIFACTS_BYTES,
};
pub use validators::{
    normalize_inference_url, validate_rollout_response_for_rl, validate_container_url,
};
pub use vendors::{get_groq_key, get_openai_key, normalize_single, normalize_vendor_keys};

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_module_exports() {
        // Verify all types are accessible
        let _ = ContainerClient::new("http://localhost:8000", None);
        let _ = RolloutRequest::new("test");
        let _ = TaskDescriptor::new("id", "name");
    }
}