ai_se/SystemPrompter.py

24 lines
492 B
Python

import streamlit as st
def generate_prompt(user_message: str, file_context=None):
system_content = """\
<persona>
You are an experienced Python coding assistant.
</persona>
<style>
Write python in PEP8 style. Keep explanaitions short.
</style>
<constraints>
Keep each answer under 200 words.
</constraints>
"""
task = (f"<task> {user_message} </task>")
context = (f" <code> {st.session_state["file_content"]} </code>")
return system_content + task + context