bullpen 0.1.0

Bullpen: an unofficial library for the Pplx API and Replit Modelfarm.
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

use std::collections::HashMap;

use serde::{Deserialize, Serialize};

use super::Metadata;
use crate::v1::models::ModelfarmChatModel;

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ChatMessage {
    pub content: String,
    pub author: String,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ChatExample {
    pub input: ChatMessage,
    pub output: ChatMessage,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ChatSession {
    pub context: String,
    pub examples: Vec<ChatExample>,
    pub messages: Vec<ChatMessage>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ModelfarmChatParameters {
    pub prompts: Vec<ChatSession>,
    pub temperature: f64,
    pub max_output_tokens: usize,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ModelfarmChatCompletionRequest {
    pub model: ModelfarmChatModel,
    pub parameters: ModelfarmChatParameters,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct Candidate {
    pub message: ChatMessage,
    pub metadata: Option<HashMap<String, serde_json::Value>>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ChatPromptResponse {
    pub candidates: Vec<Candidate>,
}

#[derive(Debug, Serialize, Deserialize, Clone)]
pub struct ModelfarmChatCompletionResponse {
    pub metadata: Option<Metadata>,
    pub responses: Vec<ChatPromptResponse>,
}