Spaces:
Sleeping
Sleeping
Update app/tasks.py
Browse files- app/tasks.py +3 -5
app/tasks.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
from app.celery_app import celery
|
| 2 |
-
from app.
|
| 3 |
from app import models
|
| 4 |
from app.llm import generate_draft
|
| 5 |
from app.utils.emailer import send_email
|
|
@@ -28,7 +28,6 @@ def task_generate_draft(draft_id: int):
|
|
| 28 |
finally:
|
| 29 |
db.close()
|
| 30 |
|
| 31 |
-
|
| 32 |
@celery.task()
|
| 33 |
def task_deliver(draft_id: int):
|
| 34 |
db = SessionLocal()
|
|
@@ -46,7 +45,7 @@ def task_deliver(draft_id: int):
|
|
| 46 |
results.append({"email": res})
|
| 47 |
# X
|
| 48 |
if draft.deliver_x:
|
| 49 |
-
text = f"{draft.title}\n\n{(draft.body_md or '')[:220]}"
|
| 50 |
res = post_to_x(text)
|
| 51 |
delivery = models.Delivery(draft_id=draft.id, channel=models.DeliveryChannel.x, payload={"text": text[:280]}, result=res)
|
| 52 |
db.add(delivery); db.commit(); db.refresh(delivery)
|
|
@@ -58,9 +57,8 @@ def task_deliver(draft_id: int):
|
|
| 58 |
db.add(delivery); db.commit(); db.refresh(delivery)
|
| 59 |
results.append({"note": res})
|
| 60 |
|
| 61 |
-
|
| 62 |
draft.status = models.DraftStatus.sent
|
| 63 |
db.add(draft); db.commit(); db.refresh(draft)
|
| 64 |
return {"id": draft.id, "results": results}
|
| 65 |
finally:
|
| 66 |
-
db.close()
|
|
|
|
| 1 |
from app.celery_app import celery
|
| 2 |
+
from app.database import SessionLocal
|
| 3 |
from app import models
|
| 4 |
from app.llm import generate_draft
|
| 5 |
from app.utils.emailer import send_email
|
|
|
|
| 28 |
finally:
|
| 29 |
db.close()
|
| 30 |
|
|
|
|
| 31 |
@celery.task()
|
| 32 |
def task_deliver(draft_id: int):
|
| 33 |
db = SessionLocal()
|
|
|
|
| 45 |
results.append({"email": res})
|
| 46 |
# X
|
| 47 |
if draft.deliver_x:
|
| 48 |
+
text = f"{draft.title}\n\n{(draft.body_md or '')[:220]}"
|
| 49 |
res = post_to_x(text)
|
| 50 |
delivery = models.Delivery(draft_id=draft.id, channel=models.DeliveryChannel.x, payload={"text": text[:280]}, result=res)
|
| 51 |
db.add(delivery); db.commit(); db.refresh(delivery)
|
|
|
|
| 57 |
db.add(delivery); db.commit(); db.refresh(delivery)
|
| 58 |
results.append({"note": res})
|
| 59 |
|
|
|
|
| 60 |
draft.status = models.DraftStatus.sent
|
| 61 |
db.add(draft); db.commit(); db.refresh(draft)
|
| 62 |
return {"id": draft.id, "results": results}
|
| 63 |
finally:
|
| 64 |
+
db.close()
|