From 36b1d49c62882e4e827027068f9e0c9e15278037 Mon Sep 17 00:00:00 2001 From: jknapp Date: Sun, 16 Mar 2025 18:41:53 -0700 Subject: [PATCH] testing thinking --- claude_sonnet-3-7-bedrock.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py index 3c12a56..2dc9aaa 100644 --- a/claude_sonnet-3-7-bedrock.py +++ b/claude_sonnet-3-7-bedrock.py @@ -182,11 +182,11 @@ class Pipeline: # Handle reasoning content (Chain of Thought) if "reasoningContent" in delta and "text" in delta["reasoningContent"]: - yield {"type": "reasoning", "content": delta["reasoningContent"]["text"]} + yield delta["reasoningContent"]["text"] # Handle regular response text if "text" in delta: - yield {"type": "response", "content": delta["text"]} + yield delta["text"] def get_completion(self, model_id: str, payload: dict) -> str: # https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-runtime/client/converse.html