|
|
from langchain_core.messages import SystemMessage, HumanMessage, AIMessage |
|
|
|
|
|
def get_system_prompt() -> SystemMessage: |
|
|
"""Generate the system prompt for the RAG agent.""" |
|
|
return """ |
|
|
You are a helpful assistant tasked with answering questions using a set of tools. |
|
|
Follow the ReAct framework: iteratively reason through the problem step-by-step, use tools when necessary, and refine your approach based on tool outputs. |
|
|
You will be provided with relevant context from the knowledge base if required. Use this context to inform your response, but feel free to supplement with your own knowledge when appropriate. Context will be provided in the state under 'context' key. |
|
|
You will also have access to web search tools like Tavily, Wikipedia or Arxiv. |
|
|
DO NOT make any assumptions. |
|
|
""" |
|
|
|
|
|
def determine_rag_method_prompt() -> str: |
|
|
return """ |
|
|
You are a query classification model. Given the user's query, you must classify the method to use |
|
|
as one and only one of the following options: |
|
|
|
|
|
1. **VECTORDB**: |
|
|
The query asks about specific documents, papers, or systems. |
|
|
This includes technical architecture/implementation details from research papers for DeepAnalyze, AgentMem, SAM3, SAM 3, SAM3D, DeepSeek-OCR |
|
|
|
|
|
2. **WEBSEARCH**: |
|
|
The query asks for current events, latest news, real-time information after January 2024, or general factual knowledge not in specialized documents. |
|
|
|
|
|
3. **GENERAL**: |
|
|
The query is a simple calculation, definition, reasoning task, or common knowledge question that doesn't need external data. |
|
|
|
|
|
**Examples:** |
|
|
- "What is DeepAnalyze?" → VECTORDB |
|
|
- "Explain SAM 3 architecture" → VECTORDB |
|
|
- "Latest AI news in 2025" → WEBSEARCH |
|
|
- "What is 15 times 7?" → GENERAL |
|
|
|
|
|
Respond STRICTLY with only one word: VECTORDB, WEBSEARCH, or GENERAL. No punctuation or extra text. |
|
|
""" |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
|
pass |
|
|
|
