rig-ecs 0.44.0

rig inside a Bevy World: effects as entities, handlers as entities, the driver as a system.
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

//! The reflection wrappers for the rig-core types the agent's components
//! hold (the bus's wrappers, [`crate::bus::reflect`], cover the rest):
//! opaque remote wrappers, serialized through their serde form.
//!
//! ```
//! let mut registry = bevy_reflect::TypeRegistry::default();
//! registry.register::<rig_ecs::agent::reflect::UsageReflect>();
//! ```

use bevy_reflect::{ReflectDeserialize, ReflectSerialize, prelude::ReflectDefault};
use rig_core::completion::{
    Usage,
    message::{AssistantContent, CallId, Message, ToolChoice, ToolName},
};
use serde::{Deserialize, Serialize};

pub use crate::bus::reflect::ToolContextReflect;

crate::reflect::opaque_reflect! {
    /// A `serde_json::Value`, reflected.
    enum JsonReflect(serde_json::Value): Default, PartialEq;
    /// An `Option<serde_json::Value>`, reflected.
    enum OptionalJsonReflect(Option<serde_json::Value>): Default, PartialEq;
    /// An `Option<ToolChoice>`, reflected.
    enum ToolChoiceReflect(Option<ToolChoice>): Default, PartialEq;
    /// A [`CallId`], reflected.
    struct ToolCallIdReflect(CallId): PartialEq;
    /// A [`ToolName`], reflected.
    struct ToolNameReflect(ToolName): PartialEq;
    /// An assistant turn's parts, `Vec<AssistantContent>`, reflected.
    struct AssistantContentsReflect(Vec<AssistantContent>): Default, PartialEq;
    /// A [`Message`], reflected.
    enum MessageReflect(Message): PartialEq;
}

/// The wire [`Usage`], reflected.
#[bevy_reflect::reflect_remote(Usage)]
#[derive(Debug, Clone, Copy, Default, PartialEq, Serialize, Deserialize)]
#[reflect(opaque, Debug, Default, PartialEq, Serialize, Deserialize)]
pub struct UsageReflect {}