refactor(bedrock): inline rig-bedrock module for crates.io compatibility

Move vendored rig-bedrock code from vendor/ into src/bedrock/ so it gets
published with the crate. This ensures users installing from crates.io
get the extended thinking fixes, since crates.io ignores path dependencies
and [patch.crates-io] sections.

Changes:
- Move vendor/rig-bedrock/src/* to src/bedrock/
- Update Cargo.toml: remove rig-bedrock dep, add AWS SDK deps directly
- Update imports from rig_bedrock:: to crate::bedrock::
- Remove vendor/rig-bedrock directory
- Add debug logging for extended thinking troubleshooting

The inlined module contains fixes for:
- Extended thinking with multi-turn tool use
- Reasoning block preservation in conversation history
- Content block sorting (Reasoning before ToolUse)

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude <noreply@anthropic.com>

Author: 2 people

Cargo.lock

@@ -73,8 +73,16 @@ rand = "0.8"
 futures-util = "0.3"
 
 # Agent dependencies (using Rig - LLM application framework)
-rig-core = { version = "0.27", features = ["derive"] }
-rig-bedrock = { version = "0.3", path = "vendor/rig-bedrock" }  # Vendored with fix for extended thinking + tool calls
+rig-core = { version = "0.27", features = ["derive", "image"] }
+
+# AWS Bedrock dependencies (inlined bedrock module with extended thinking fixes)
+async-stream = "0.3"
+aws-config = { version = "1", features = ["behavior-version-latest"] }
+aws-sdk-bedrockruntime = "1"
+aws-smithy-types = "1"
+base64 = "0.22"
+schemars = "0.8"
+tracing = "0.1"
 
 # Diff rendering for file confirmation UI
 similar = "2.6"
@@ -107,8 +115,3 @@ path = "examples/check_vulnerabilities.rs"
 name = "security_analysis"
 path = "examples/security_analysis.rs"
 
-# Patch rig-bedrock to fix extended thinking support with tool use
-# The upstream version drops Reasoning blocks when a tool call is present.
-# See: vendor/rig-bedrock/src/types/assistant_content.rs for the fix.
-[patch.crates-io]
-rig-bedrock = { version = "0.3", path = "vendor/rig-bedrock" }

Cargo.toml

@@ -495,7 +495,7 @@ pub async fn run_interactive(
                 }
                 ProviderType::Bedrock => {
                     // Bedrock provider via rig-bedrock - same pattern as OpenAI/Anthropic
-                    let client = rig_bedrock::client::Client::from_env();
+                    let client = crate::bedrock::client::Client::from_env();
 
                     // Extended thinking for Claude models via Bedrock
                     // This enables Claude to show its reasoning process before responding.
@@ -1477,7 +1477,7 @@ pub async fn run_query(
         }
         ProviderType::Bedrock => {
             // Bedrock provider via rig-bedrock - same pattern as Anthropic
-            let client = rig_bedrock::client::Client::from_env();
+            let client = crate::bedrock::client::Client::from_env();
             let model_name = model
                 .as_deref()
                 .unwrap_or("global.anthropic.claude-sonnet-4-5-20250929-v1:0");

src/agent/mod.rs

@@ -1,5 +1,5 @@
-use crate::image::ImageGenerationModel;
-use crate::{completion::CompletionModel, embedding::EmbeddingModel};
+use super::image::ImageGenerationModel;
+use super::{completion::CompletionModel, embedding::EmbeddingModel};
 use aws_config::{BehaviorVersion, Region};
 use rig::client::Nothing;
 use rig::prelude::*;

vendor/rig-bedrock/src/client.rs src/bedrock/client.rsvendor/rig-bedrock/src/client.rs renamed to src/bedrock/client.rs

@@ -1,6 +1,6 @@
 //! All supported models <https://docs.aws.amazon.com/bedrock/latest/userguide/models-supported.html>
 
-use crate::{
+use super::{
     client::Client,
     types::{
         assistant_content::AwsConverseOutput, completion_request::AwsCompletionRequest,
@@ -173,7 +173,7 @@ impl CompletionModel {
 
 impl completion::CompletionModel for CompletionModel {
     type Response = AwsConverseOutput;
-    type StreamingResponse = crate::streaming::BedrockStreamingResponse;
+    type StreamingResponse = super::streaming::BedrockStreamingResponse;
 
     type Client = Client;
 

vendor/rig-bedrock/src/completion.rs src/bedrock/completion.rsvendor/rig-bedrock/src/completion.rs renamed to src/bedrock/completion.rs

@@ -2,7 +2,7 @@ use aws_smithy_types::Blob;
 use rig::embeddings::{self, Embedding, EmbeddingError};
 use serde::{Deserialize, Serialize};
 
-use crate::{client::Client, types::errors::AwsSdkInvokeModelError};
+use super::{client::Client, types::errors::AwsSdkInvokeModelError};
 
 #[derive(Serialize)]
 #[serde(rename_all = "camelCase")]

vendor/rig-bedrock/src/embedding.rs src/bedrock/embedding.rsvendor/rig-bedrock/src/embedding.rs renamed to src/bedrock/embedding.rs

@@ -1,6 +1,6 @@
-use crate::client::Client;
-use crate::types::errors::AwsSdkInvokeModelError;
-use crate::types::text_to_image::{TextToImageGeneration, TextToImageResponse};
+use super::client::Client;
+use super::types::errors::AwsSdkInvokeModelError;
+use super::types::text_to_image::{TextToImageGeneration, TextToImageResponse};
 use aws_smithy_types::Blob;
 use rig::image_generation::{
     self, ImageGenerationError, ImageGenerationRequest, ImageGenerationResponse,

vendor/rig-bedrock/src/image.rs src/bedrock/image.rsvendor/rig-bedrock/src/image.rs renamed to src/bedrock/image.rs

@@ -1,5 +1,5 @@
-use crate::types::completion_request::AwsCompletionRequest;
-use crate::{completion::CompletionModel, types::errors::AwsSdkConverseStreamError};
+use super::types::completion_request::AwsCompletionRequest;
+use super::{completion::CompletionModel, types::errors::AwsSdkConverseStreamError};
 use async_stream::stream;
 use aws_sdk_bedrockruntime::types as aws_bedrock;
 use rig::completion::GetTokenUsage;

vendor/rig-bedrock/src/lib.rs src/bedrock/mod.rsvendor/rig-bedrock/src/lib.rs renamed to src/bedrock/mod.rs

@@ -6,7 +6,7 @@ use rig::{
 };
 use serde::{Deserialize, Serialize};
 
-use crate::types::message::RigMessage;
+use super::message::RigMessage;
 
 use super::{converse_output::InternalConverseOutput, json::AwsDocument};
 use rig::completion;
@@ -78,6 +78,29 @@ impl TryFrom<aws_bedrock::ContentBlock> for RigAssistantContent {
     type Error = CompletionError;
 
     fn try_from(value: aws_bedrock::ContentBlock) -> Result<Self, Self::Error> {
+        // Debug: Log incoming AWS content block
+        let block_type = match &value {
+            aws_bedrock::ContentBlock::Text(t) => format!("Text(len={})", t.len()),
+            aws_bedrock::ContentBlock::ToolUse(t) => {
+                format!("ToolUse(id={}, name={})", t.tool_use_id, t.name)
+            }
+            aws_bedrock::ContentBlock::ReasoningContent(r) => match r {
+                aws_bedrock::ReasoningContentBlock::ReasoningText(rt) => format!(
+                    "ReasoningContent::ReasoningText(len={}, has_sig={})",
+                    rt.text.len(),
+                    rt.signature.is_some()
+                ),
+                aws_bedrock::ReasoningContentBlock::RedactedContent(blob) => format!(
+                    "ReasoningContent::RedactedContent(blob_len={})",
+                    blob.as_ref().len()
+                ),
+                _ => "ReasoningContent::Unknown".to_string(),
+            },
+            aws_bedrock::ContentBlock::ToolResult(_) => "ToolResult".to_string(),
+            _ => "Other".to_string(),
+        };
+        tracing::debug!("Converting AWS ContentBlock to Rig: {}", block_type);
+
         match value {
             aws_bedrock::ContentBlock::Text(text) => {
                 Ok(RigAssistantContent(AssistantContent::Text(Text { text })))
@@ -91,18 +114,42 @@ impl TryFrom<aws_bedrock::ContentBlock> for RigAssistantContent {
             )),
             aws_bedrock::ContentBlock::ReasoningContent(reasoning_block) => match reasoning_block {
                 aws_bedrock::ReasoningContentBlock::ReasoningText(reasoning_text) => {
+                    tracing::debug!(
+                        "Converting ReasoningText: text_len={}, signature={:?}",
+                        reasoning_text.text.len(),
+                        reasoning_text
+                            .signature
+                            .as_ref()
+                            .map(|s| format!("{}...", &s[..s.len().min(20)]))
+                    );
                     Ok(RigAssistantContent(AssistantContent::Reasoning(
                         rig::message::Reasoning::new(&reasoning_text.text)
                             .with_signature(reasoning_text.signature),
                     )))
                 }
-                _ => Err(CompletionError::ProviderError(
-                    "AWS Bedrock returned unsupported ReasoningContentBlock variant".into(),
-                )),
+                aws_bedrock::ReasoningContentBlock::RedactedContent(blob) => {
+                    tracing::warn!(
+                        "AWS Bedrock returned RedactedContent (blob_len={}). This variant is not yet supported!",
+                        blob.as_ref().len()
+                    );
+                    Err(CompletionError::ProviderError(format!(
+                        "AWS Bedrock returned RedactedContent (blob_len={}). This variant needs to be handled for multi-turn conversations with extended thinking.",
+                        blob.as_ref().len()
+                    )))
+                }
+                _ => {
+                    tracing::error!("AWS Bedrock returned unknown ReasoningContentBlock variant");
+                    Err(CompletionError::ProviderError(
+                        "AWS Bedrock returned unsupported ReasoningContentBlock variant".into(),
+                    ))
+                }
             },
-            _ => Err(CompletionError::ProviderError(
-                "AWS Bedrock returned unsupported ContentBlock".into(),
-            )),
+            _ => {
+                tracing::warn!("AWS Bedrock returned unsupported ContentBlock type");
+                Err(CompletionError::ProviderError(
+                    "AWS Bedrock returned unsupported ContentBlock".into(),
+                ))
+            }
         }
     }
 }
@@ -190,12 +237,9 @@ impl TryFrom<RigAssistantContent> for aws_bedrock::ContentBlock {
 
 #[cfg(test)]
 mod tests {
-    use crate::types::{
-        assistant_content::RigAssistantContent, converse_output::InternalConverseOutput,
-        errors::TypeConversionError,
-    };
-
+    use super::super::{converse_output::InternalConverseOutput, errors::TypeConversionError};
     use super::AwsConverseOutput;
+    use super::RigAssistantContent;
     use aws_sdk_bedrockruntime::types as aws_bedrock;
     use rig::{OneOrMany, completion, message::AssistantContent};