Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-10-17 14:25:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Ignore spaces after <think> and <tool> tags.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-06-19 13:35:05 +02:00
parent 91cfc2e5b8
commit ca02df8ad5
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -215,25 +215,25 @@ bool TracyLlmChat::Turn( TurnRole role, const std::string& content )
if( minPos == posThink )
{
pos += sizeof( "<think>" ) - 1;
while( content[pos] == '\n' ) pos++;
while( content[pos] == '\n' || content[pos] == ' ' ) pos++;
auto endThink = content.find( "</think>", pos );
ThinkScope();
if( m_thinkOpen ) PrintThink( content.c_str() + pos, std::min( end, endThink ) - pos );
if( endThink == std::string::npos ) break;
pos = endThink + sizeof( "</think>" ) - 1;
while( content[pos] == '\n' ) pos++;
while( content[pos] == '\n' || content[pos] == ' ' ) pos++;
}
else
{
assert( minPos == posTool );
pos += sizeof( "<tool>" ) - 1;
while( content[pos] == '\n' ) pos++;
while( content[pos] == '\n' || content[pos] == ' ' ) pos++;
auto endTool = content.find( "</tool>", pos );
ThinkScope();
if( m_thinkOpen ) PrintToolCall( content.c_str() + pos, std::min( end, endTool ) - pos );
if( endTool == std::string::npos ) break;
pos = endTool + sizeof( "</tool>" ) - 1;
while( content[pos] == '\n' ) pos++;
while( content[pos] == '\n' || content[pos] == ' ' ) pos++;
}
}
}