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

Explicitly check for assistant role.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-05-12 22:36:36 +02:00
parent 2ad2fa0b4a
commit 3a60a67c95
1 changed files with 13 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -186,6 +186,7 @@ void TracyLlm::Draw()
const auto role = line["role"].get<std::string>();
const auto isUser = role == "user";
const auto isError = role == "error";
const auto isAssistant = role == "assistant";
float diff, offset;
if( isUser )
@@ -204,7 +205,7 @@ void TracyLlm::Draw()
ImGui::SameLine( 0, 0 );
ImGui::TextColored( ImVec4( 1.f, 0.25f, 0.25f, 1.f ), ICON_FA_CIRCLE_EXCLAMATION );
}
else
else if( isAssistant )
{
diff = mw - rw;
offset = diff / 2;
@@ -212,6 +213,11 @@ void TracyLlm::Draw()
ImGui::SameLine( 0, 0 );
ImGui::TextColored( ImVec4( 0.4f, 0.5f, 1.f, 1.f ), ICON_FA_ROBOT );
}
else
{
assert( false );
}
ImGui::SameLine( 0, 0 );
ImGui::Dummy( ImVec2( diff - offset, 0 ) );
ImGui::SameLine();
@@ -227,12 +233,16 @@ void TracyLlm::Draw()
{
ImGui::PushStyleColor( ImGuiCol_Text, ImVec4( 1.f, 0.25f, 0.25f, 1.f ) );
}
else
else if( isAssistant )
{
ImGui::PushStyleColor( ImGuiCol_Text, style.Colors[ImGuiCol_Text] );
}
else
{
assert( false );
}
if( !isUser && !isError )
if( isAssistant )
{
auto str = line["content"].get<std::string>();
if( strncmp( str.c_str(), "<think>", 7 ) == 0 )