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

Release lock when handling tool calls.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-05-17 01:11:33 +02:00
parent 0ba0590dbc
commit a35e996f4f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -605,7 +605,7 @@ static std::vector<std::string> SplitLines( const std::string& str )
bool TracyLlm::OnResponse( const ollama::response& response )
{
std::lock_guard lock( m_lock );
std::unique_lock lock( m_lock );
if( m_stop )
{
@@ -643,8 +643,10 @@ bool TracyLlm::OnResponse( const ollama::response& response )
isTool = true;
auto tool = lines[0];
lines.erase( lines.begin() );
lock.unlock();
const auto reply = HandleToolCalls( tool, lines );
const auto output = "<tool_output>\n" + reply.reply;
lock.lock();
if( reply.image.empty() )
{
m_chat->emplace_back( ollama::message( "user", output ) );