Scarfski/tracy
1
0
Fork 0
mirror of https://github.com/wolfpld/tracy.git synced 2026-09-04 09:28:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Go back to regex paths in source search.

Browse Source
This commit is contained in:
Bartosz Taudul
2026-01-25 16:29:53 +01:00
parent 27ecaed76e
commit 0a58d88be2
3 changed files with 19 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -154,9 +154,9 @@
"type": "boolean",
"description": "Case insensitive search."
},
"glob": {
"path": {
"type": "string",
"description": "File path match."
"description": "File path ECMAScript regex."
}
},
"required": ["query"]

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

@@ -181,7 +181,7 @@ TracyLlmTools::ToolReply TracyLlmTools::HandleToolCalls( const std::string& tool
else if( tool == "source_search" )
{
std::string empty;
return { .reply = SourceSearch( Param( "query" ), ParamOptBool( "case_insensitive", false ), ParamOptString( "glob", empty ) ) };
return { .reply = SourceSearch( Param( "query" ), ParamOptBool( "case_insensitive", false ), ParamOptString( "path", empty ) ) };
}
return { .reply = "Unknown tool call: " + tool };
}
@@ -825,53 +825,38 @@ std::string TracyLlmTools::SourceFile( const std::string& file, uint32_t line, u
return json.dump( 2, ' ', false, nlohmann::json::error_handler_t::replace );
}
static bool GlobMatch( const std::string& pattern, const std::string& text )
{
std::string rx;
for( char c : pattern )
{
if( c == '*' )
{
rx += ".*";
}
else if( c == '?' )
{
rx += ".";
}
else if( c == '.' || c == '+' || c == '^' || c == '$' || c == '|' || c == '(' || c == ')' || c == '[' || c == ']' || c == '{' || c == '}' )
{
rx += '\\';
rx += c;
}
else
{
rx += c;
}
}
return std::regex_search( text, std::regex( rx ) );
}
std::string TracyLlmTools::SourceSearch( std::string query, bool caseInsensitive, const std::string& glob ) const
std::string TracyLlmTools::SourceSearch( std::string query, bool caseInsensitive, const std::string& path ) const
{
auto& cache = m_worker.GetSourceFileCache();
nlohmann::json json = {};
if( caseInsensitive ) std::ranges::transform( query, query.begin(), []( char c ) { return std::tolower( c ); } );
std::regex rx;
std::regex rx, rxPath;
try
{
rx = std::regex( query );
}
catch( const std::regex_error& e )
{
return "Error: Invalid regex: " + std::string( e.what() );
return "Error: Invalid query regex: " + std::string( e.what() );
}
if( !path.empty() )
{
try
{
rxPath = std::regex( path );
}
catch( const std::regex_error& e )
{
return "Error: Invalid path regex: " + std::string( e.what() );
}
}
size_t total = 0;
for( auto& item : cache )
{
if( IsFrameExternal( item.first, nullptr ) ) continue;
if( !glob.empty() && !GlobMatch( glob, item.first ) ) continue;
if( !path.empty() && !std::regex_search( item.first, rxPath ) ) continue;
char* tmp = nullptr;
auto& mem = item.second;

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

@@ -62,7 +62,7 @@ private:
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 SourceSearch( std::string query, bool caseInsensitive, const std::string& glob ) const;
std::string SourceSearch( std::string query, bool caseInsensitive, const std::string& path ) const;
void ManualEmbeddingsWorker( TracyLlmApi& api );