Implement AI orchestration wedding demo
This commit is contained in:
41
backend/app/seed.py
Normal file
41
backend/app/seed.py
Normal file
@@ -0,0 +1,41 @@
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from app.infrastructure.database import Note, Reservation, SessionLocal, Task, init_db
|
||||
from app.memory.memory_manager import MemoryManager
|
||||
|
||||
|
||||
def seed() -> None:
|
||||
init_db()
|
||||
session = SessionLocal()
|
||||
try:
|
||||
if session.query(Task).count() == 0:
|
||||
session.add_all(
|
||||
[
|
||||
Task(title="Review venue shortlist", assignee="Maria", status="pending"),
|
||||
Task(title="Confirm catering tasting", assignee="Nikos", status="in_progress"),
|
||||
]
|
||||
)
|
||||
if session.query(Note).count() == 0:
|
||||
session.add_all(
|
||||
[
|
||||
Note(content="Customer prefers white flowers.", category="preference"),
|
||||
Note(content="Budget conversation should stay under 20,000 EUR.", category="budget"),
|
||||
]
|
||||
)
|
||||
if session.query(Reservation).count() == 0:
|
||||
session.add(
|
||||
Reservation(
|
||||
vendor="Aegean Garden Venue",
|
||||
reservation_date=datetime.utcnow() + timedelta(days=45),
|
||||
status="pending",
|
||||
)
|
||||
)
|
||||
session.commit()
|
||||
memory = MemoryManager()
|
||||
memory.seed_demo()
|
||||
finally:
|
||||
session.close()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
seed()
|
||||
Reference in New Issue
Block a user