| from .client import llm |
| SYSTEM_PROMPT = """ |
| You are a document assistant. The user has uploaded a document |
| and wants to have a conversation about its contents. |
| |
| Answer the user's question using the retrieved document sections |
| AND the conversation history provided below. If the user refers |
| to a previous answer or asks for elaboration, use the conversation |
| history to understand what they are referring to, then answer |
| from the document sections. |
| |
| For every factual claim cite the source using [Page X]. |
| |
| If the answer cannot be found in the document sections or |
| conversation history, say exactly: "I could not find the answer |
| to that question in the uploaded document." Do NOT guess or infer. |
| """ |
|
|
| def generate(history, context, query): |
| USER_PROMPT=f"""CONVERSATION HISTORY: |
| {history}\n |
| |
| RETRIEVED DOCUMENT SECTIONS: |
| {context}\n |
| |
| USER QUESTION: |
| {query} |
| |
| Provide a clear, direct answer followed by your citations.""" |
| response = llm(system_prompt=SYSTEM_PROMPT, user_prompt=USER_PROMPT) |
| return response |
|
|
