67 lines
2.8 KiB
Python
67 lines
2.8 KiB
Python
from django.utils import timezone
|
|
from openai import AsyncOpenAI
|
|
|
|
from core.models import AI, Manipulation, Person
|
|
|
|
|
|
def generate_prompt(msg: dict, person: Person, manip: Manipulation, chat_history: str):
|
|
"""
|
|
Generate a structured prompt using the attributes of the provided Person and Manipulation models.
|
|
"""
|
|
|
|
now = timezone.now()
|
|
persona = manip.persona
|
|
|
|
system_message = (
|
|
"You are my digital persona, responding on my behalf while embodying my personality, preferences, and unique style.\n\n"
|
|
"### Persona Profile ###\n"
|
|
f"- **MBTI:** {persona.mbti} ({persona.mbti_identity} balance)\n"
|
|
f"- **Tone:** {persona.tone} | **Humor:** {persona.humor_style}\n"
|
|
f"- **Core Values:** {persona.core_values}\n"
|
|
f"- **Communication Style:** {persona.communication_style}\n"
|
|
f"- **Flirting Style:** {persona.flirting_style}\n"
|
|
f"- **Likes:** {persona.likes} | **Dislikes:** {persona.dislikes}\n"
|
|
f"- **Response Tactics:** {persona.response_tactics}\n"
|
|
f"- **Persuasion Techniques:** {persona.persuasion_tactics}\n"
|
|
f"- **Boundaries:** {persona.boundaries} | **Adaptability:** {persona.adaptability}%\n\n"
|
|
"### Contact Information ###\n"
|
|
f"- **Summary:** {person.summary or 'N/A'}\n"
|
|
f"- **Profile:** {person.profile or 'N/A'}\n"
|
|
f"- **Revealed Details:** {person.revealed or 'N/A'}\n"
|
|
f"- **Sentiment Score:** {person.sentiment}\n"
|
|
f"- **Timezone:** {person.timezone or 'N/A'}\n"
|
|
f"- **Last Interaction:** {person.last_interaction or 'Never'}\n"
|
|
f"- **Current Date/Time:** {now}\n\n"
|
|
"### Conversation Context ###\n"
|
|
f"{chat_history if chat_history else 'No prior chat history.'}\n\n"
|
|
"### Response Guidelines ###\n"
|
|
"- **Engagement**: Keep responses engaging, with a balance of wit, depth, and confidence.\n"
|
|
"- **Flirting**: Be direct, playful, and, when appropriate, subtly provocative—without hesitation.\n"
|
|
"- **Pauses**: Use double newlines (`\\n\\n`) to pause where it enhances realism.\n"
|
|
"- **Flow Awareness**: Maintain continuity, avoid redundancy, and adjust response length based on interaction.\n"
|
|
)
|
|
|
|
user_message = f"[{msg['timestamp']}] <{person.name}> {msg['text']}"
|
|
|
|
return [
|
|
{"role": "system", "content": system_message},
|
|
{"role": "user", "content": user_message},
|
|
]
|
|
|
|
|
|
async def run_context_prompt(
|
|
prompt: list[str],
|
|
ai: AI,
|
|
):
|
|
cast = {"api_key": ai.api_key}
|
|
if ai.base_url is not None:
|
|
cast["base_url"] = ai.base_url
|
|
client = AsyncOpenAI(**cast)
|
|
response = await client.chat.completions.create(
|
|
model=ai.model,
|
|
messages=prompt,
|
|
)
|
|
content = response.choices[0].message.content
|
|
|
|
return content
|