adding project id to job table

Author: nishika26

backend/app/api/routes/llm.py

@@ -70,7 +70,7 @@ def llm_call(
 
     # Fetch job details to return immediate response
     job_crud = JobCrud(session=session)
-    job = job_crud.get(job_id=job_id)
+    job = job_crud.get(job_id=job_id, project_id=project_id)
 
     if not job:
         raise HTTPException(status_code=404, detail="Job not found")
@@ -106,16 +106,19 @@ def get_llm_call_status(
     Poll for LLM call job status and results.
     Returns job information with nested LLM response when complete.
     """
+
+    project_id = _current_user.project_.id
+
     job_crud = JobCrud(session=session)
-    job = job_crud.get(job_id=job_id)
+    job = job_crud.get(job_id=job_id, project_id=project_id)
 
     if not job:
         raise HTTPException(status_code=404, detail="Job not found")
 
     llm_call_response = None
     if job.status.value == JobStatus.SUCCESS:
         llm_calls = get_llm_calls_by_job_id(
-            session=session, job_id=job_id, project_id=_current_user.project_.id
+            session=session, job_id=job_id, project_id=project_id
         )
 
         if llm_calls:

backend/app/crud/jobs.py

@@ -12,11 +12,13 @@ class JobCrud:
     def __init__(self, session: Session):
         self.session = session
 
-    def create(self, job_type: JobType, trace_id: str | None = None) -> Job:
-        new_job = Job(
-            job_type=job_type,
-            trace_id=trace_id,
-        )
+    def create(
+        self,
+        job_type: JobType,
+        trace_id: str | None = None,
+        project_id: int | None = None,
+    ) -> Job:
+        new_job = Job(job_type=job_type, trace_id=trace_id, project_id=project_id)
         self.session.add(new_job)
         self.session.commit()
         self.session.refresh(new_job)
@@ -38,5 +40,14 @@ def update(self, job_id: UUID, job_update: JobUpdate) -> Job:
 
         return job
 
-    def get(self, job_id: UUID) -> Job | None:
-        return self.session.get(Job, job_id)
+    def get(self, job_id: UUID, project_id: int | None = None) -> Job | None:
+        job = self.session.get(Job, job_id)
+        if job is None:
+            return None
+        if (
+            project_id is not None
+            and job.project_id is not None
+            and job.project_id != project_id
+        ):
+            return None
+        return job

backend/app/models/job.py

@@ -40,6 +40,11 @@ class Job(SQLModel, table=True):
         description="Tracing ID for correlating logs and traces.",
         sa_column_kwargs={"comment": "Tracing ID for correlating logs and traces"},
     )
+    project_id: int | None = Field(
+        default=None,
+        description="Project ID of the project the job belongs to.",
+        sa_column_kwargs={"comment": "Project ID of the job's project"},
+    )
     error_message: str | None = Field(
         default=None,
         description="Error details if the job fails.",

backend/app/services/llm/jobs.py

@@ -46,7 +46,9 @@ def start_job(
     """Create an LLM job and schedule Celery task."""
     trace_id = correlation_id.get() or "N/A"
     job_crud = JobCrud(session=db)
-    job = job_crud.create(job_type=JobType.LLM_API, trace_id=trace_id)
+    job = job_crud.create(
+        job_type=JobType.LLM_API, trace_id=trace_id, project_id=project_id
+    )
 
     # Explicitly flush to ensure job is persisted before Celery task starts
     db.flush()
@@ -87,7 +89,9 @@ def start_chain_job(
     """Create an LLM Chain job and schedule Celery task."""
     trace_id = correlation_id.get() or "N/A"
     job_crud = JobCrud(session=db)
-    job = job_crud.create(job_type=JobType.LLM_CHAIN, trace_id=trace_id)
+    job = job_crud.create(
+        job_type=JobType.LLM_CHAIN, trace_id=trace_id, project_id=project_id
+    )
 
     # Explicitly flush to ensure job is persisted before Celery task starts
     db.flush()

backend/app/services/response/jobs.py

@@ -19,7 +19,9 @@ def start_job(
     """Create a response job and schedule Celery task."""
     trace_id = correlation_id.get() or "N/A"
     job_crud = JobCrud(session=db)
-    job = job_crud.create(job_type=JobType.RESPONSE, trace_id=trace_id)
+    job = job_crud.create(
+        job_type=JobType.RESPONSE, trace_id=trace_id, project_id=project_id
+    )
 
     try:
         task_id = start_response_job(