diff --git a/worker/orca-agent/src/database/models.py b/worker/orca-agent/src/database/models.py index 75bb54e..2d8034f 100644 --- a/worker/orca-agent/src/database/models.py +++ b/worker/orca-agent/src/database/models.py @@ -10,7 +10,7 @@ class Submission(SQLModel, table=True): """Task submission model.""" task_id: str - swarmBountyId: int = Field(primary_key=True) + swarmBountyId: str = Field(primary_key=True) status: str = "pending" pr_url: Optional[str] = None username: Optional[str] = None diff --git a/worker/orca-agent/src/server/routes/submission.py b/worker/orca-agent/src/server/routes/submission.py index 5ffcab7..7da4fbe 100644 --- a/worker/orca-agent/src/server/routes/submission.py +++ b/worker/orca-agent/src/server/routes/submission.py @@ -16,7 +16,7 @@ def fetch_submission(swarmBountyId): submission = ( db.query(Submission) .filter( - Submission.swarmBountyId == int(swarmBountyId), + Submission.swarmBountyId == swarmBountyId, ) .first() ) diff --git a/worker/orca-agent/src/server/services/repo_summary_service.py b/worker/orca-agent/src/server/services/repo_summary_service.py index 5b0e4b1..4e57006 100644 --- a/worker/orca-agent/src/server/services/repo_summary_service.py +++ b/worker/orca-agent/src/server/services/repo_summary_service.py @@ -26,6 +26,7 @@ def handle_task_creation(task_id, swarmBountyId, repo_url, db=None): result = workflow.run() if result.get("success"): + # Convert swarmBountyId to integer submission = Submission( task_id=task_id, swarmBountyId=swarmBountyId,