mirror of
https://github.com/dograh-hq/dograh.git
synced 2026-06-07 07:55:16 +02:00
67479e98fd
* fix timestamps in tuner accumelator * chore: refactor strip_thought_ids_from_messages --------- Co-authored-by: Abhishek Kumar <abhishek@a6k.me>
42 lines
1.2 KiB
Python
42 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
from copy import deepcopy
|
|
|
|
from pipecat.utils.context.message_sanitization import (
|
|
strip_thought_from_id,
|
|
strip_thought_ids_from_messages,
|
|
)
|
|
|
|
|
|
def test_strip_thought_from_id():
|
|
assert strip_thought_from_id("call_123__thought__abc") == "call_123"
|
|
assert strip_thought_from_id("call_123") == "call_123"
|
|
assert strip_thought_from_id(None) is None
|
|
|
|
|
|
def test_strip_thought_ids_from_messages_does_not_mutate_input():
|
|
messages = [
|
|
{
|
|
"role": "assistant",
|
|
"tool_calls": [
|
|
{
|
|
"id": "call_1__thought__hidden",
|
|
"type": "function",
|
|
"function": {"name": "lookup", "arguments": "{}"},
|
|
}
|
|
],
|
|
},
|
|
{
|
|
"role": "tool",
|
|
"tool_call_id": "call_1__thought__hidden",
|
|
"content": '{"status":"ok"}',
|
|
},
|
|
]
|
|
original = deepcopy(messages)
|
|
|
|
cleaned = strip_thought_ids_from_messages(messages)
|
|
|
|
assert messages == original
|
|
assert cleaned is not messages
|
|
assert cleaned[0]["tool_calls"][0]["id"] == "call_1"
|
|
assert cleaned[1]["tool_call_id"] == "call_1"
|