aisdk 0.5.2

An open-source Rust library for building AI-powered applications, inspired by the Vercel AI SDK. It provides a robust, type-safe, and easy-to-use interface for interacting with various Large Language Models (LLMs).
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

//! OpenAI Chat Completions API provider.
//!
//! **INTERNAL USE ONLY**: This provider is not intended for direct use by end users.
//! It implements the OpenAI Chat Completions API (`/v1/chat/completions`) and serves
//! as a base for providers like Groq, OpenRouter, Together AI, and others that are
//! compatible with this API format.

pub(crate) mod client;
pub(crate) mod conversions;
pub(crate) mod embedding_model;
pub(crate) mod language_model;
#[macro_use]
pub mod macros;
pub mod settings;

use crate::core::DynamicModel;
use crate::core::capabilities::ModelName;
use client::ChatCompletionsOptions;
use settings::OpenAIChatCompletionsSettings;

/// Internal OpenAI Chat Completions API provider.
///
/// This is not intended for direct end-user use. Use the OpenAI provider for
/// the latest features, or wrap this provider for compatible APIs.
#[derive(Debug, Clone)]
pub struct OpenAIChatCompletions<M: ModelName> {
    /// Configuration settings for the provider
    pub settings: OpenAIChatCompletionsSettings,
    /// Request options for the API call
    pub(crate) options: ChatCompletionsOptions,
    _phantom: std::marker::PhantomData<M>,
}

impl<M: ModelName> Default for OpenAIChatCompletions<M> {
    fn default() -> Self {
        let settings = OpenAIChatCompletionsSettings::default();
        let options = ChatCompletionsOptions {
            model: M::MODEL_NAME.to_string(),
            messages: vec![],
            ..Default::default()
        };

        Self {
            settings,
            options,
            _phantom: std::marker::PhantomData,
        }
    }
}

impl OpenAIChatCompletions<DynamicModel> {
    /// Creates an OpenAIChatCompletions provider with a dynamic model name using default settings.
    ///
    /// **INTERNAL USE ONLY**: This method is for internal provider implementations.
    ///
    /// # Parameters
    ///
    /// * `model_name` - The model identifier (e.g., "gpt-4o", "llama-3.3-70b-specdec")
    ///
    /// # Returns
    ///
    /// A configured `OpenAIChatCompletions<DynamicModel>` provider instance.
    pub fn model_name(name: impl Into<String>) -> Self {
        let settings = OpenAIChatCompletionsSettings::default();
        let options = ChatCompletionsOptions {
            model: name.into(),
            messages: vec![],
            ..Default::default()
        };

        Self {
            settings,
            options,
            _phantom: std::marker::PhantomData,
        }
    }
}