Skip to content

Commit d97edca

Browse files
[#76] Add logger for final prompt
1 parent 9996595 commit d97edca

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

backend/app/services/chat_job_service.py

Lines changed: 11 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ async def execute_chat_job(
1717
data: dict,
1818
callback_url: str,
1919
app_default_prompt: str,
20-
knowledge_base_ids: List[int] = []
20+
knowledge_base_ids: List[int] = [],
2121
):
2222
"""Background job executor for chat jobs (non-streaming)."""
2323
job = JobService.get_job(db, job_id)
@@ -58,6 +58,7 @@ async def execute_chat_job(
5858

5959
# combined prompt
6060
final_prompt = qa_prompt + "\n\n" + app_final_prompt
61+
logger.info(f"Chat job final prompt: {final_prompt}")
6162

6263
state = {
6364
"query": query,
@@ -80,11 +81,15 @@ async def execute_chat_job(
8081
citations = []
8182
for context in result_state.get("context", []):
8283
doc_metadata = context.metadata or {}
83-
citations.append({
84-
"document": doc_metadata.get("source") or doc_metadata.get("title"),
85-
"chunk": context.page_content,
86-
"page": doc_metadata.get("page_label") or doc_metadata.get("page"),
87-
})
84+
citations.append(
85+
{
86+
"document": doc_metadata.get("source")
87+
or doc_metadata.get("title"),
88+
"chunk": context.page_content,
89+
"page": doc_metadata.get("page_label")
90+
or doc_metadata.get("page"),
91+
}
92+
)
8893

8994
output = {"answer": answer, "citations": citations}
9095

0 commit comments

Comments
 (0)