Spaces:
Building
Building
File size: 1,492 Bytes
24ae72d b9ccd0b 24ae72d b9ccd0b 24ae72d b9ccd0b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
# /home/bk_anupam/code/LLM_agents/RAG_BOT/agent/prompts.py
import os
import sys
from langchain_core.prompts import ChatPromptTemplate
# Add the project root to the Python path to import Config
project_root = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..'))
sys.path.insert(0, project_root)
from RAG_BOT.config import Config
# --- Helper Prompt Functions ---
def get_evaluate_context_chat_prompt():
return ChatPromptTemplate.from_messages(
[
("system", Config.get_evaluate_context_prompt()),
(
"human",
"Original User Question: {original_query}\n\nRetrieved Context:\n{context}"
),
]
)
def get_reframe_question_chat_prompt():
return ChatPromptTemplate.from_messages(
[
("system", Config.get_reframe_question_prompt()),
(
"human",
"Original User Question: {original_query}\nQuery Used for Failed Retrieval: {failed_query}"
),
]
)
def get_final_answer_chat_prompt(language_code: str):
"""Creates the chat prompt template for generating the final answer, using the specified language."""
system_template = Config.get_final_answer_system_prompt_template(language_code) # Use language
human_template = Config.get_final_answer_human_prompt_template()
return ChatPromptTemplate.from_messages([
("system", system_template),
("human", human_template),
])
|