gemma / generator /prompt_builder.py
dasomaru's picture
Upload folder using huggingface_hub
2c5f455 verified
raw
history blame contribute delete
626 Bytes
def build_prompt(query: str, law_docs: list, exam_docs: list) -> str:
prompt = (
"๋‹น์‹ ์€ ๊ณต์ธ์ค‘๊ฐœ์‚ฌ ์‹œํ—˜ ๋ฌธ์ œ ์ถœ์ œ ์ „๋ฌธ๊ฐ€์ž…๋‹ˆ๋‹ค.\n\n"
"์•„๋ž˜๋Š” ๊ด€๋ จ ๋ฒ•๋ น๊ณผ ๊ธฐ์ถœ๋ฌธ์ œ์ž…๋‹ˆ๋‹ค:\n\n"
"[๋ฒ•๋ น ์ •๋ณด]\n"
)
for doc in law_docs:
prompt += f"- {doc}\n"
prompt += "\n[๊ธฐ์ถœ๋ฌธ์ œ ์ •๋ณด]\n"
for doc in exam_docs:
prompt += f"- {doc}\n"
prompt += f"\n์œ„ ์ •๋ณด๋ฅผ ์ฐธ๊ณ ํ•˜์—ฌ ์‚ฌ์šฉ์ž์˜ ์š”์ฒญ์— ์ •ํ™•ํ•˜๊ณ  ๋ช…ํ™•ํ•˜๊ฒŒ ๋‹ต๋ณ€ํ•˜์„ธ์š”.\n\n"
prompt += f"[์งˆ๋ฌธ]\n{query}\n\n[๋‹ต๋ณ€]\n"
return prompt