Skip to content

Commit a79fd96

Browse files
committed
Update dbmem-rembed.c
1 parent 016567d commit a79fd96

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/dbmem-rembed.c

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -210,7 +210,7 @@ static int set_json_error_message (dbmem_remote_engine_t *engine) {
210210
return -1;
211211
}
212212

213-
#if ENABLE_DBMEM_DEBUG
213+
#if ENABLE_DBMEM_DEBUG_EMBEDDING
214214
static void dbmem_remote_debug_log_response(dbmem_remote_engine_t *engine, long http_code) {
215215
const char *response = engine->data ? engine->data : "";
216216
DEBUG_DBMEM_ALWAYS("[dbmem-rembed] vectors.space response (HTTP %ld): %s", http_code, response);
@@ -457,7 +457,7 @@ int dbmem_remote_compute_embedding (dbmem_remote_engine_t *engine, const char *t
457457
sqlite3_free(response_data);
458458
#endif
459459

460-
#if ENABLE_DBMEM_DEBUG
460+
#if ENABLE_DBMEM_DEBUG_EMBEDDING
461461
dbmem_remote_debug_log_response(engine, http_code);
462462
#endif
463463

@@ -552,7 +552,7 @@ int dbmem_remote_compute_embedding (dbmem_remote_engine_t *engine, const char *t
552552
result->embedding = engine->embedding;
553553

554554
// Update statistics
555-
engine->total_tokens_processed += prompt_tokens;
555+
engine->total_tokens_processed += result->n_tokens;
556556
engine->total_embeddings_generated++;
557557

558558
return 0;

0 commit comments

Comments
 (0)