Merge pull request #393 from UiPath/fix/job_context

Fix: job context

Author: alexenica

src/uipath/_services/jobs_service.py

@@ -553,8 +553,12 @@ def create_attachment(
 
         # Get job key from context if not explicitly provided
         context_job_key = None
-        if job_key is None and hasattr(self._execution_context, "job_key"):
-            context_job_key = self._execution_context.job_key
+        if job_key is None:
+            try:
+                context_job_key = self._execution_context.instance_key
+            except ValueError:
+                # Instance key is not set in environment
+                context_job_key = None
 
         # Check if a job is available
         if job_key is not None or context_job_key is not None:
@@ -690,8 +694,12 @@ async def main():
 
         # Get job key from context if not explicitly provided
         context_job_key = None
-        if job_key is None and hasattr(self._execution_context, "job_key"):
-            context_job_key = self._execution_context.job_key
+        if job_key is None:
+            try:
+                context_job_key = self._execution_context.instance_key
+            except ValueError:
+                # Instance key is not set in environment
+                context_job_key = None
 
         # Check if a job is available
         if job_key is not None or context_job_key is not None:

tests/sdk/services/test_jobs_service.py

@@ -579,16 +579,13 @@ def test_create_job_attachment_with_job_context(
         content = "Test attachment content"
         name = "test_attachment.txt"
 
-        # Set job key in execution context - add attribute if it doesn't exist
-        if not hasattr(execution_context, "job_key"):
-            # Add job_key attribute to ExecutionContext
-            execution_context.__dict__["job_key"] = job_key
-        else:
-            execution_context.job_key = job_key
-
-        # Create service with our execution context
+        # Set job key in environment - must be string
+        monkeypatch.setenv("UIPATH_JOB_KEY", str(job_key))
         monkeypatch.setenv("UIPATH_FOLDER_PATH", "test-folder-path")
-        service = JobsService(config=config, execution_context=execution_context)
+
+        # Create fresh execution context after setting environment variables
+        fresh_execution_context = ExecutionContext()
+        service = JobsService(config=config, execution_context=fresh_execution_context)
 
         # Mock the attachment service's upload method
         mock_upload = mocker.patch.object(