Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-09-03 00:48:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Do not display system prompt injected into user messages.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-12-28 18:15:44 +01:00
parent 0246cec0a7
commit 9183aef645
1 changed files with 15 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
profiler/src/profiler/TracyLlmChat.cpp
View File

@@ -152,7 +152,21 @@ bool TracyLlmChat::Turn( TurnRole role, const std::string& content )
}
else if( role != TurnRole::Assistant )
{
m_markdown.Print( content.c_str(), content.size() );
auto pos = content.find( "<SYSTEM_PROMPT>" );
if( pos == std::string::npos )
{
m_markdown.Print( content.c_str(), content.size() );
}
else
{
auto str = content;
while( ( pos = str.find( "<SYSTEM_PROMPT>" ) ) != std::string::npos )
{
auto pos2 = str.find( "</SYSTEM_PROMPT>" );
str = str.substr( 0, pos ) + str.substr( pos2 + sizeof( "</SYSTEM_PROMPT>" ) - 1 );
}
m_markdown.Print( str.c_str(), str.size() );
}
}
else if( content.starts_with( "<tool_output>\n" ) )
{