From 7fc76228f79d2406854a56efab4e609b8249ac42 Mon Sep 17 00:00:00 2001 From: Meng Zhang Date: Tue, 3 Oct 2023 11:38:58 -0700 Subject: [PATCH] chore: add debug log for /chat interface --- crates/tabby/src/serve/chat.rs | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/crates/tabby/src/serve/chat.rs b/crates/tabby/src/serve/chat.rs index 2cf17d1..c6a7ea0 100644 --- a/crates/tabby/src/serve/chat.rs +++ b/crates/tabby/src/serve/chat.rs @@ -12,7 +12,7 @@ use axum_streams::StreamBodyAs; use prompt::ChatPromptBuilder; use serde::{Deserialize, Serialize}; use tabby_inference::{TextGeneration, TextGenerationOptions, TextGenerationOptionsBuilder}; -use tracing::instrument; +use tracing::{debug, instrument}; use utoipa::ToSchema; pub struct ChatState { @@ -69,6 +69,7 @@ pub async fn completions( Json(request): Json, ) -> Response { let (prompt, options) = parse_request(&state, request); + debug!("PROMPT: {}", prompt); let s = stream! { for await content in state.engine.generate_stream(&prompt, options).await { yield ChatCompletionChunk { content }