@@ -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