mirror of
https://github.com/x1xhlol/system-prompts-and-models-of-ai-tools.git
synced 2026-06-18 15:29:36 +00:00
100 lines
2.8 KiB
Python
100 lines
2.8 KiB
Python
"""
|
|
Outreach Agent — generates personalized cold outreach messages.
|
|
وكيل الوصول — يُنشئ رسائل وصول باردة مخصصة.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
from typing import Any, Literal
|
|
|
|
from auto_client_acquisition.agents.intake import Lead
|
|
from core.agents.base import BaseAgent
|
|
from core.config.models import Task
|
|
from core.llm.base import Message
|
|
from core.prompts import get_prompt
|
|
|
|
Channel = Literal["email", "whatsapp", "linkedin", "sms"]
|
|
|
|
|
|
@dataclass
|
|
class OutreachMessage:
|
|
channel: Channel
|
|
subject: str | None
|
|
body: str
|
|
locale: str
|
|
recipient_channel_value: str | None # email or phone
|
|
|
|
def to_dict(self) -> dict[str, Any]:
|
|
return {
|
|
"channel": self.channel,
|
|
"subject": self.subject,
|
|
"body": self.body,
|
|
"locale": self.locale,
|
|
"recipient_channel_value": self.recipient_channel_value,
|
|
}
|
|
|
|
|
|
class OutreachAgent(BaseAgent):
|
|
"""Generates opener messages for cold outreach."""
|
|
|
|
name = "outreach"
|
|
|
|
async def run(
|
|
self,
|
|
*,
|
|
lead: Lead,
|
|
channel: Channel = "email",
|
|
trigger: str = "We saw your profile",
|
|
**_: Any,
|
|
) -> OutreachMessage:
|
|
"""Generate a personalized cold opener."""
|
|
prompt = get_prompt(
|
|
"outreach_opener",
|
|
channel=channel,
|
|
locale=lead.locale,
|
|
name=lead.contact_name or "there",
|
|
company=lead.company_name or "your company",
|
|
trigger=trigger,
|
|
)
|
|
|
|
response = await self.router.run(
|
|
task=Task.PAGE_COPY,
|
|
messages=[Message(role="user", content=prompt)],
|
|
max_tokens=400,
|
|
temperature=0.6,
|
|
)
|
|
|
|
subject: str | None = None
|
|
body = response.content.strip()
|
|
if channel == "email":
|
|
subject = self._build_subject(lead)
|
|
|
|
recipient = lead.contact_email if channel == "email" else lead.contact_phone
|
|
|
|
message = OutreachMessage(
|
|
channel=channel,
|
|
subject=subject,
|
|
body=body,
|
|
locale=lead.locale,
|
|
recipient_channel_value=recipient,
|
|
)
|
|
|
|
self.log.info(
|
|
"outreach_generated",
|
|
lead_id=lead.id,
|
|
channel=channel,
|
|
locale=lead.locale,
|
|
)
|
|
return message
|
|
|
|
@staticmethod
|
|
def _build_subject(lead: Lead) -> str:
|
|
if lead.locale == "ar":
|
|
if lead.sector:
|
|
return f"فرصة سريعة لـ {lead.company_name or 'شركتكم'} في {lead.sector}"
|
|
return f"سؤال قصير لـ {lead.company_name or 'شركتكم'}"
|
|
if lead.sector:
|
|
return f"Quick idea for {lead.company_name or 'your team'} in {lead.sector}"
|
|
return f"Quick question for {lead.company_name or 'your team'}"
|