diff --git a/claude_sonnet-3-7-bedrock.py b/claude_sonnet-3-7-bedrock.py index 5c00f7d..4d44bdb 100644 --- a/claude_sonnet-3-7-bedrock.py +++ b/claude_sonnet-3-7-bedrock.py @@ -183,13 +183,12 @@ class Pipeline: # Handle reasoning content (Chain of Thought) if "reasoningContent" in delta and "text" in delta["reasoningContent"]: - if '"
' in delta["reasoningContent"]["text"]: - yield "" - yield delta["reasoningContent"]["text"] + print(f"Reasoning content: {delta['reasoningContent']}") + yield delta["reasoningContent"]["text"] # Handle regular response text if "text" in delta: - yield 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