from pydantic import BaseModel

import dspy

dspy.settings.configure(lm=dspy.LM("openai/gpt-4o-mini"))


class CalendarEvent(BaseModel):
    name: str
    date: str
    participants: dict[str, str]


def write_invitation_letter(participant_name: str, event_info: CalendarEvent):
    # if isinstance(event_info, dict):
    #     event_info = CalendarEvent.model_validate(event_info)
    if participant_name not in event_info.participants:
        return None
    return f"It's my honor to invite {participant_name} to event {event_info.name} on {event_info.date}"


class InvitationSignature(dspy.Signature):
    participant_name: str = dspy.InputField(desc="The name of the participant to invite")
    event_info: CalendarEvent = dspy.InputField(desc="The information about the event")
    invitation_letter: str = dspy.OutputField(desc="The invitation letter to be sent to the participant")


react = dspy.ReAct(InvitationSignature, tools=[write_invitation_letter])

outputs = react(
    participant_name="Alice",
    event_info=CalendarEvent(
        name="Science Fair",
        date="Friday",
        participants={"Alice": "female", "Bob": "male"},
    ),
)
print(outputs)