openagent 0.1.10

OpenAI Agent Kit
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

//! OpenAI API

pub mod batch;
pub mod chat;
pub mod embedding;
pub mod file;
pub mod response;
pub mod r#type;

// self
use crate::_prelude::*;

/// JSON event handler for OpenAI API events.
pub struct ApiEventHandler<T>(PhantomData<T>);
impl<T> ApiEventHandler<T> {
	/// Create a new JSON event handler.
	pub fn new() -> Self {
		Self(PhantomData)
	}
}
impl<T> Default for ApiEventHandler<T> {
	fn default() -> Self {
		Self::new()
	}
}
impl<T> EventHandler for ApiEventHandler<T>
where
	T: Send + DeserializeOwned,
{
	type Event = T;

	fn handle_data(&self, data: String) -> Result<Self::Event> {
		Ok(serde_json::from_str(&data)?)
	}

	fn handle_unexpected(&self, unexpected: String) -> Result<()> {
		if let Ok(e) = serde_json::from_str::<ApiErrorWrapper>(&unexpected) {
			Err(Error::Api(e.error))
		} else {
			Err(Error::any(unexpected))
		}
	}
}