File size: 460 Bytes
bb17288
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
class PromptManager:
    @staticmethod
    def format_prompt(user_input, context="", history=None):
        prompt = ""
        if context:
            prompt += f"Context: {context}\n"
        if history:
            prompt += "Conversation History:\n"
            for msg in history:
                prompt += f"User: {msg['user_message']}\nAI: {msg['ai_response']}\n"
        prompt += f"User: {user_input}\nPlease respond helpfully."
        return prompt