tokn-convert 0.2.0-rc.3

Request and response conversion pipeline across tokn endpoint formats
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

pub use tokn_core::provider;

pub mod chat;
pub mod error;
pub mod ir;
pub mod messages;
pub mod responses;
pub mod sse;
pub mod tools;
pub mod usage;

use crate::provider::Endpoint;
use serde_json::Value;

pub use error::Result;

pub fn convert_request(from: Endpoint, to: Endpoint, body: &Value) -> Result<Value> {
  if from == to {
    return Ok(body.clone());
  }
  let req = match from {
    Endpoint::ChatCompletions => chat::request_from_value(body)?,
    Endpoint::Responses => responses::request_from_value(body)?,
    Endpoint::Messages => messages::request_from_value(body)?,
  };
  match to {
    Endpoint::ChatCompletions => chat::request_to_value(&req),
    Endpoint::Responses => responses::request_to_value(&req),
    Endpoint::Messages => messages::request_to_value(&req),
  }
}

pub fn convert_response(from: Endpoint, to: Endpoint, body: &Value) -> Result<Value> {
  if from == to {
    return Ok(body.clone());
  }
  let resp = match from {
    Endpoint::ChatCompletions => chat::response_from_value(body)?,
    Endpoint::Responses => responses::response_from_value(body)?,
    Endpoint::Messages => messages::response_from_value(body)?,
  };
  match to {
    Endpoint::ChatCompletions => chat::response_to_value(&resp),
    Endpoint::Responses => responses::response_to_value(&resp),
    Endpoint::Messages => messages::response_to_value(&resp),
  }
}