Fix Mentra nested Responses endpoints

vendor/mentra-provider/src/stream.rs

@@ -0,0 +1,101 @@
+use tokio::sync::mpsc;
+
+use crate::{
+    ReasoningProvenance, model::HostedToolSearchCall, model::HostedWebSearchCall,
+    model::ImageGenerationCall, model::ImageGenerationResult, model::ImageSource, model::Role,
+    model::TokenUsage, model::ToolResultContent, model::WebSearchAction,
+};
+
+pub type ProviderEventStream = mpsc::UnboundedReceiver<Result<ProviderEvent, crate::ProviderError>>;
+
+#[derive(Debug, Clone, PartialEq, Eq, Default)]
+pub struct ResponseHeaders {
+    pub values: Vec<(String, String)>,
+}
+
+#[derive(Debug, Clone, PartialEq)]
+pub enum ProviderEvent {
+    ResponseHeaders(ResponseHeaders),
+    ResponseCreated,
+    MessageStarted {
+        id: String,
+        model: String,
+        role: Role,
+    },
+    ContentBlockStarted {
+        index: usize,
+        kind: ContentBlockStart,
+    },
+    ContentBlockDelta {
+        index: usize,
+        delta: ContentBlockDelta,
+    },
+    ContentBlockStopped {
+        index: usize,
+    },
+    MessageDelta {
+        stop_reason: Option<String>,
+        usage: Option<TokenUsage>,
+    },
+    ReasoningSummaryDelta {
+        delta: String,
+        summary_index: i64,
+    },
+    ReasoningContentDelta {
+        delta: String,
+        content_index: i64,
+    },
+    ReasoningSummaryPartAdded {
+        summary_index: i64,
+    },
+    MessageStopped,
+}
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub enum ContentBlockStart {
+    Text,
+    Thinking {
+        encrypted_content: Option<String>,
+        id: Option<String>,
+        provenance: Option<ReasoningProvenance>,
+        redacted: bool,
+    },
+    Image {
+        source: ImageSource,
+    },
+    ToolUse {
+        id: String,
+        name: String,
+    },
+    ToolResult {
+        tool_use_id: String,
+        is_error: bool,
+        content: Option<ToolResultContent>,
+    },
+    HostedToolSearch {
+        call: HostedToolSearchCall,
+    },
+    HostedWebSearch {
+        call: HostedWebSearchCall,
+    },
+    ImageGeneration {
+        call: ImageGenerationCall,
+    },
+}
+
+#[derive(Debug, Clone, PartialEq, Eq)]
+pub enum ContentBlockDelta {
+    Text(String),
+    ThinkingText(String),
+    ThinkingSignature(String),
+    ThinkingEncryptedContent(String),
+    ToolUseInputJson(String),
+    ToolResultContent(ToolResultContent),
+    HostedToolSearchQuery(String),
+    HostedToolSearchStatus(String),
+    HostedWebSearchAction(WebSearchAction),
+    HostedWebSearchStatus(String),
+    ImageGenerationStatus(String),
+    ImageGenerationRevisedPrompt(String),
+    ImageGenerationResult(ImageGenerationResult),
+}