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

Fail when embeddings were not calculated.

Browse Source
This commit is contained in:
Bartosz Taudul
2025-06-11 23:06:26 +02:00
parent 2578a0c04a
commit 7e0476a641
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -336,7 +336,12 @@ void TracyLlmTools::ManualEmbeddingsWorker( TracyLlmApi& api )
req["model"] = m_manualEmbeddingState.model;
nlohmann::json response;
api.Embeddings( req, response );
if( !api.Embeddings( req, response ) )
{
m_manualEmbeddingState.inProgress = false;
m_manualEmbeddingState.done = false;
return;
}
auto& data = response["data"];
for( size_t j=0; j<bsz; j++ )
@@ -800,7 +805,7 @@ std::string TracyLlmTools::SearchManual( const std::string& query, TracyLlmApi&
req["model"] = m_manualEmbeddingState.model;
nlohmann::json response;
api.Embeddings( req, response, true );
if( !api.Embeddings( req, response, true ) ) return "Error: Failed to get embedding for the query";
auto& embedding = response["data"][0]["embedding"];
if( embedding.empty() ) return "Error: Failed to get embedding for the query";