espionox 0.1.42

A library for creating multi-agent workflows in Rust
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

use super::super::streaming::{CompletionStreamStatus, StreamResponse};
use serde::Deserialize;

impl StreamResponse for OpenAiStreamResponse {}

#[derive(Debug, Deserialize, Clone)]
pub struct OpenAiStreamResponse {
    pub choices: Vec<StreamChoice>,
}

#[derive(Debug, Deserialize, Clone)]
struct StreamChoice {
    pub delta: StreamDelta,
}

#[derive(Debug, Deserialize, Clone)]
struct StreamDelta {
    pub role: Option<String>,
    pub content: Option<String>,
}

impl Into<CompletionStreamStatus> for OpenAiStreamResponse {
    fn into(self) -> CompletionStreamStatus {
        match self.choices[0].delta.content.to_owned() {
            Some(response) => CompletionStreamStatus::Working(
                response
                    .trim_start_matches('"')
                    .trim_end_matches('"')
                    .to_string(),
            ),
            None => CompletionStreamStatus::Finished,
        }
    }
}