Spaces:
Sleeping
Sleeping
| # app/modules/processing/prompt_builder.py | |
| from typing import Any, List, Dict | |
| from runtime.session_store import get_session_controller | |
| def build_planning_prompt(message: str, rag_context: str = "") -> List[Dict[str, str]]: | |
| system_content = ( | |
| "You are an elite query planner for a search-augmented LLM. Your task is to determine " | |
| "if an external web search is absolutely mandatory to answer the user's question.\n\n" | |
| "TRIAE STRATEGY (Priority):\n" | |
| "1. INTERNAL KNOWLEDGE: If the question is about historical facts, static scientific truths, " | |
| "unchanging concepts, or general logic that you already know perfectly, rely entirely on your own knowledge.\n" | |
| "2. LOCAL RAG CONTEXT: Use the provided RAG context (Wikipedia) to complement or verify your knowledge.\n" | |
| "3. WEB SEARCH: Generate web search queries ONLY if the question requires real-time information (after 2025/2026), " | |
| "highly localized niche data not found in Wikipedia, or breaking updates.\n\n" | |
| "CRITICAL PERFORMANCE REQUIREMENT:\n" | |
| "- Keep your internal thinking (<think>...</think>) extremely short (under 3 lines). Do not ramble.\n" | |
| "- If internal knowledge or local RAG is sufficient, immediately output exactly 0 queries.\n" | |
| "- Output ONLY the raw queries, one per line, no numbers, no bullet points, no markdown markup outside thinking tags." | |
| ) | |
| rag_section = rag_context.strip() if rag_context else "None (No relevant local database records found)." | |
| user_content = ( | |
| f"User Question: {message}\n\n" | |
| f"Available Local RAG Context (Wikipedia snippets):\n{rag_section}\n\n" | |
| "Task:\n" | |
| "- Output 1 to 3 highly concise search queries ONLY if external web search is required.\n" | |
| "- Output exactly '0' if no external search is needed." | |
| ) | |
| return [ | |
| {"role": "system", "content": system_content}, | |
| {"role": "user", "content": user_content} | |
| ] | |
| def build_generation_prompt( | |
| message: str, | |
| docs: list, | |
| history: Any = None, | |
| hf_user: str = None, | |
| session_id: str = None, | |
| top_k: int = 5 | |
| ) -> List[Dict[str, str]]: | |
| system_content = ( | |
| "You are an advanced, direct response assistant. Your objective is to synthesize your internal " | |
| "expertise with the provided context to answer the user comprehensively and clearly in Markdown format.\n\n" | |
| "CRITICAL INSTRUCTIONS FOR SPEED & TIMEOUT PREVENTION:\n" | |
| "- ABSOLUTELY NO INTERNAL REASONING OR THINKING PROCESS ALLOWED. Do not output <think> tags or chains of thought.\n" | |
| "- Start generating the final answer directly from the first token.\n" | |
| "- Prioritize your comprehensive internal knowledge for historical or logical accuracy, and use the provided context to add specific references or URLs.\n" | |
| "- Cite sources accurately using the exact (Title)[URL] format when referencing the provided context elements." | |
| ) | |
| sorted_docs = sorted(docs or [], key=lambda d: d.get("score", 0), reverse=True) | |
| top_docs = sorted_docs[:top_k] | |
| context_entries = [] | |
| for d in top_docs: | |
| entry = ( | |
| f"Query Source: {d.get('query', 'N/A')}\n" | |
| f"Title: {d.get('title', 'Untitled')}\n" | |
| f"URL: {d.get('url', '')}\n" | |
| f"Content: {d.get('snippet') or d.get('summary') or d.get('content') or ''}\n" | |
| ) | |
| context_entries.append(entry.strip()) | |
| context_str = "\n---\n".join(context_entries) if context_entries else "No external or local context provided. Rely entirely on your internal parameters." | |
| history_str = "" | |
| if history: | |
| if history == "history_ctr": | |
| history_controller = get_session_controller(hf_user, session_id) | |
| try: | |
| history_str = history_controller.build_prompt_history(message, max_tokens=1024, top_k=10) | |
| except Exception: | |
| pass | |
| else: | |
| try: | |
| last_items = history[-4:] if isinstance(history, list) else [str(history)] | |
| history_str = "\n".join(f"{r}: {c}" for r, c in last_items) | |
| except Exception: | |
| pass | |
| user_content = "" | |
| if history_str: | |
| user_content += f"Relevant Context from Past Conversation turns:\n{history_str}\n\n" | |
| user_content += ( | |
| f"Retrieved Reference Context (Local RAG & Web Search):\n{context_str}\n\n" | |
| f"Instructions:\n" | |
| f"- Answer the user message using your internal knowledge and the reference context above.\n" | |
| f"- Respond in the same language as the user's message.\n" | |
| f"- Do not mention that you were forbidden to think. Just start the answer.\n\n" | |
| f"User Message: {message}" | |
| ) | |
| return [ | |
| {"role": "system", "content": system_content}, | |
| {"role": "user", "content": user_content} | |
| ] |