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

Indicate forgotten tool outputs in the chat window.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-05-25 03:27:42 +02:00
parent 168ae92b8b
commit 736a2d3ba2
1 changed files with 18 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -400,7 +400,8 @@ void TracyLlm::Draw()
const auto rw = ImGui::CalcTextSize( ICON_FA_ROBOT ).x;
const auto ew = ImGui::CalcTextSize( ICON_FA_CIRCLE_EXCLAMATION ).x;
const auto yw = ImGui::CalcTextSize( ICON_FA_REPLY ).x;
const auto mw = std::max( { uw, rw, ew, yw } );
const auto cw = ImGui::CalcTextSize( ICON_FA_RECYCLE ).x;
const auto mw = std::max( { uw, rw, ew, yw, cw } );
const auto posStart = ImGui::GetCursorPos().x;
const auto& role = line["role"].get_ref<const std::string&>();
@@ -411,9 +412,18 @@ void TracyLlm::Draw()
const auto isError = role == "error";
const auto isAssistant = role == "assistant";
const auto isToolResponse = isUser && line["content"].get_ref<const std::string&>().starts_with( "<tool_output>\n" );
const auto isForgotten = isToolResponse && line["content"].get_ref<const std::string&>() == ForgetMsg;
float diff, offset;
if( isToolResponse )
if( isForgotten )
{
diff = mw - cw;
offset = diff / 2;
ImGui::Dummy( ImVec2( offset, 0 ) );
ImGui::SameLine( 0, 0 );
ImGui::TextColored( style.Colors[ImGuiCol_TextDisabled], ICON_FA_RECYCLE );
}
else if( isToolResponse )
{
diff = mw - yw;
offset = diff / 2;
@@ -476,7 +486,12 @@ void TracyLlm::Draw()
assert( false );
}
if( isToolResponse )
if( isForgotten )
{
ImGui::TextUnformatted( "Tool response removed to save context space" );
treeIdx++;
}
else if( isToolResponse )
{
ImGui::PushID( treeIdx++ );
auto expand = ImGui::TreeNode( "Tool response..." );