amico-mods 0.0.2

The plugins of the Amico AI Agent Framework
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
87
88
89
90
91
92
93
94
95

use amico::ai::completion::{Error as CompletionError, Model, ModelChoice, Request};
use rig::{
    completion::{self as rc, CompletionModel as _},
    providers as rp,
};
use std::fmt::Debug;

use super::rig_helpers::*;

/// Re-export providers from rig-core
/// so that SDK users do not need to add `rig-core` as a dependency
pub use rig::providers;

// Implement type convertions

/// The uniform completion wrapper for all rig providers
async fn provider_completion(
    provider: &RigProvider,
    model_name: &str,
    request: rc::CompletionRequest,
) -> Result<ModelChoice, CompletionError> {
    match provider {
        RigProvider::Anthropic(client) => client
            .completion_model(model_name)
            .completion(request)
            .await
            .map(into_amico_choice)
            .map_err(into_amico_err),
        RigProvider::Deepseek(client) => client
            .completion_model(model_name)
            .completion(request)
            .await
            .map(into_amico_choice)
            .map_err(into_amico_err),
        RigProvider::Gemini(client) => client
            .completion_model(model_name)
            .completion(request)
            .await
            .map(into_amico_choice)
            .map_err(into_amico_err),
        RigProvider::Openai(client) => client
            .completion_model(model_name)
            .completion(request)
            .await
            .map(into_amico_choice)
            .map_err(into_amico_err),
    }
}

/// OpenAI provider using `rig-core`
pub enum RigProvider {
    Anthropic(rp::anthropic::Client),
    Deepseek(rp::deepseek::Client),
    Gemini(rp::gemini::Client),
    Openai(rp::openai::Client),
}

impl Debug for RigProvider {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        f.write_str(&format!(
            "RigProvider::{}",
            match self {
                Self::Anthropic(_) => "Anthropic",
                Self::Deepseek(_) => "Deepseek",
                Self::Gemini(_) => "Gemini",
                Self::Openai(_) => "Openai",
            }
        ))
    }
}

impl RigProvider {
    pub fn anthropic(client: rp::anthropic::Client) -> Self {
        Self::Anthropic(client)
    }

    pub fn deepseek(client: rp::deepseek::Client) -> Self {
        Self::Deepseek(client)
    }

    pub fn gemini(client: rp::gemini::Client) -> Self {
        Self::Gemini(client)
    }

    pub fn openai(client: rp::openai::Client) -> Self {
        Self::Openai(client)
    }
}

impl Model for RigProvider {
    #[doc = " Completes a prompt with the provider."]
    async fn completion(&self, req: &Request) -> Result<ModelChoice, CompletionError> {
        provider_completion(self, &req.model, into_rig_request(req)).await
    }
}