prompts.py

import requests
from smolagents import CODE_SYSTEM_PROMPT

from src.settings import settings

DOCUMENT_CONTEXT_PROMPT = """
<document>
{doc_content}
</document>
"""

CHUNK_CONTEXT_PROMPT = """
Here is the chunk we want to situate within the whole document
<chunk>
{chunk_content}
</chunk>

Please give a short succinct context to situate this chunk within the overall document for the purposes of improving search retrieval of the chunk.
Answer only with the succinct context and nothing else.
"""

QUERY_PROMPT = "Represent this sentence for searching relevant passages: "

SYSTEM_PROMPT_RULES = """
## Rules:
- When using the VideoEditorTool, call `await sample()` at the end of the code and use VisualFeedbackTool to analyze the samples
- <callout>DO NOT REPEAT THE SAME CODE THAT WAS ALREADY EXECUTED<callout>, the VideoEditorTool is persistent and will keep the state of the composition between calls.
- If you create a variable, you can reuse with the next evaluation. If you call `composition.add(video)` multiple times, it will add the same video multiple times.
- If the VisualFeedbackTool rejects the composition, you need to fix the issues and call `await sample()` again.
- When the VisualFeedbackTool accepts the composition, use call `await render()` with the VideoEditorTool **without anything else**.
"""

def get_system_prompt():
    result = requests.get(f"{settings.url}/system-prompt.txt")

    return f"{CODE_SYSTEM_PROMPT}\n\n{result.text}\n\n{SYSTEM_PROMPT_RULES}"