Skip to content
Merged
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion tools/server/server-context.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -77,6 +77,7 @@ struct server_slot {
size_t last_nl_pos = 0;

std::string generated_text;
std::string debug_generated_text;
llama_tokens generated_tokens;

// idx of draft tokens in the main batch
Expand Down Expand Up @@ -425,7 +426,7 @@ struct server_slot {

if (!only_metrics) {
res["prompt"] = ptask->tokens.detokenize(ctx, true);
res["generated"] = generated_text;
res["generated"] = generated_text.empty()?debug_generated_text:generated_text;
Comment thread
ngxson marked this conversation as resolved.
Resolved
}
}

Expand Down Expand Up @@ -1442,6 +1443,12 @@ struct server_context_impl {
res->id_slot = slot.id;

res->index = slot.task->index;

// keep copy of last generated text for debugging purposes
if (slots_debug > 0) {
Comment thread
ngxson marked this conversation as resolved.
Resolved
slot.debug_generated_text = slot.generated_text;
}

// in stream mode, content and tokens are already in last partial chunk
if (slot.task->params.stream) {
res->content = "";
Expand Down
Loading