artificial-openai 0.2.0

OpenAI backend adapter for the Artificial prompt-engineering SDK
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

use artificial_core::generic::{GenericFunctionCall, GenericFunctionCallIntent};
use serde::{Deserialize, Serialize};

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct ToolCall {
    pub id: String,
    pub function: ToolCallFunction,
    pub r#type: ToolType,
}

impl From<ToolCall> for GenericFunctionCallIntent {
    fn from(val: ToolCall) -> Self {
        GenericFunctionCallIntent {
            id: val.id,
            function: val.function.into(),
        }
    }
}

impl From<GenericFunctionCallIntent> for ToolCall {
    fn from(value: GenericFunctionCallIntent) -> Self {
        Self {
            id: value.id,
            function: value.function.into(),
            r#type: ToolType::Function,
        }
    }
}

#[derive(Debug, Clone, Deserialize, Serialize)]
#[serde(rename_all = "snake_case")]
pub enum ToolType {
    Function,
}

#[derive(Debug, Clone, Deserialize, Serialize)]
pub struct ToolCallFunction {
    pub name: String,
    pub arguments: String,
}

impl From<ToolCallFunction> for GenericFunctionCall {
    fn from(val: ToolCallFunction) -> Self {
        GenericFunctionCall {
            name: val.name,
            arguments: serde_json::from_str(&val.arguments)
                .unwrap_or(serde_json::Value::String(val.arguments)),
        }
    }
}

impl From<GenericFunctionCall> for ToolCallFunction {
    fn from(value: GenericFunctionCall) -> Self {
        Self {
            name: value.name,
            arguments: value.arguments.to_string(),
        }
    }
}