Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-08-22 02:58:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Split source_file parameter context into context and context_back.

Browse Source
This commit is contained in:
Bartosz Taudul
2026-01-30 01:07:15 +01:00
parent 81e8b1a4d8
commit e017f504c6
3 changed files with 19 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
profiler/src/llm/tools.json
View File

@@ -131,7 +131,11 @@
},
"context": {
"type": "integer",
"description": "How many lines to get before and after the target line."
"description": "How many lines to get after the target line. Default: 2."
},
"context_back": {
"type": "integer",
"description": "How many lines to get before the target line. Default: 2."
}
},
"required": ["file", "line"]

25
profiler/src/profiler/TracyLlmTools.cpp
View File

@@ -176,7 +176,7 @@ TracyLlmTools::ToolReply TracyLlmTools::HandleToolCalls( const std::string& tool
}
else if( tool == "source_file" )
{
return { .reply = SourceFile( Param( "file" ), ParamU32( "line" ), ParamOptU32( "context", 50 ) ) };
return { .reply = SourceFile( Param( "file" ), ParamU32( "line" ), ParamOptU32( "context", 2 ), ParamOptU32( "context_back", 2 ) ) };
}
else if( tool == "source_search" )
{
@@ -776,7 +776,7 @@ std::string TracyLlmTools::SearchManual( const std::string& query, TracyLlmApi&
return json.dump( 2, ' ', false, nlohmann::json::error_handler_t::replace );
}
std::string TracyLlmTools::SourceFile( const std::string& file, uint32_t line, uint32_t context ) const
std::string TracyLlmTools::SourceFile( const std::string& file, uint32_t line, uint32_t context, uint32_t contextBack ) const
{
if( line == 0 ) return "Error: Source file line number must be greater than 0.";
@@ -793,21 +793,22 @@ std::string TracyLlmTools::SourceFile( const std::string& file, uint32_t line, u
uint32_t minLine = line;
uint32_t maxLine = line+1;
while( context > 0 && ( minLine > 0 || maxLine < lines.size() ) )
while( ( context > 0 && maxLine < lines.size() ) || ( contextBack > 0 && minLine > 0 ) )
{
if( minLine > 0 )
if( context > 0 && maxLine < lines.size() )
{
size += lines[minLine].size() * 3 + 30;
if( size >= maxSize ) break;
minLine--;
}
if( maxLine < lines.size() )
{
size += lines[maxLine].size() * 3 + 30;
size += lines[maxLine].size() + 7;
if( size >= maxSize ) break;
maxLine++;
context--;
}
if( contextBack > 0 && minLine > 0 )
{
size += lines[minLine].size() + 7;
if( size >= maxSize ) break;
minLine--;
contextBack--;
}
context--;
}
nlohmann::json json = {

2
profiler/src/profiler/TracyLlmTools.hpp
View File

@@ -61,7 +61,7 @@ private:
std::string SearchWeb( std::string query );
std::string GetWebpage( const std::string& url );
std::string SearchManual( const std::string& query, TracyLlmApi& api, bool hasEmbeddingsModel );
std::string SourceFile( const std::string& file, uint32_t line, uint32_t context ) const;
std::string SourceFile( const std::string& file, uint32_t line, uint32_t context, uint32_t contextBack ) const;
std::string SourceSearch( std::string query, bool caseInsensitive, const std::string& path ) const;
void ManualEmbeddingsWorker( TracyLlmApi& api );