OpenWebUI-Tools/Pipelines
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

testing thinking

Browse Source
This commit is contained in:
jknapp 2025-03-16 19:26:06 -07:00
parent 99a49cdec0
commit a33a73003e
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
claude_sonnet-3-7-bedrock.py
View File

@ -176,23 +176,18 @@ class Pipeline:
def stream_response(self, model_id: str, payload: dict) -> Generator:
# https://boto3.amazonaws.com/v1/documentation/api/latest/reference/services/bedrock-runtime/client/converse_stream.html
streaming_response = self.bedrock_runtime.converse_stream(**payload)
print(f"{streaming_response} for {payload}")
for chunk in streaming_response["stream"]:
think_start = False
if "contentBlockDelta" in chunk:
delta = chunk["contentBlockDelta"]["delta"]
# Handle reasoning content (Chain of Thought)
if "reasoningContent" in delta and "text" in delta["reasoningContent"]:
if not think_start:
think_start = True
yield "\n<thinking>"
# if '"<details type="reasoning" done="false">' in delta["reasoningContent"]["text"]:
yield delta["reasoningContent"]["text"]
# Handle regular response text
if "text" in delta:
if think_start:
yield "\n</thinking>\n"
think_start = False
yield delta["text"]
def get_completion(self, model_id: str, payload: dict) -> str: