testing thinking

This commit is contained in:
jknapp 2025-03-16 20:16:39 -07:00
parent 97382fde4b
commit 5ed90be307

View File

@ -186,12 +186,12 @@ class Pipeline:
# Handle reasoning content (Chain of Thought)
if "reasoningContent" in delta and "text" in delta["reasoningContent"]:
if not thinking_block:
thinking_block = delta["reasoningContent"]["contentBlockIndex"]
thinking_block = delta["contentBlockIndex"]
yield '<thinking>\n'
#print(f"Reasoning content: {delta['reasoningContent']}")
yield delta["reasoningContent"]["text"]
if "contentBlockStop" in delta and delta["contentBlockDelta"]["contentBlockIndex"] == thinking_block:
if "contentBlockStop" in delta and delta["contentBlockStop"]["contentBlockIndex"] == thinking_block:
yield '</thinking>\n\n'
# Handle regular response text