Function and jobs refactor - resolve circular dependencies

client/qiskit_serverless/core/client.py

@@ -28,20 +28,19 @@
 """
 import warnings
 from abc import ABC, abstractmethod
-from typing import Optional, List, Dict, Any, Union
+from typing import Optional, List
 
-from qiskit_ibm_runtime import QiskitRuntimeService
-
-from qiskit_serverless.core.job import (
-    Job,
-    Configuration,
+from qiskit_serverless.core.job import Job, JobClient
+from qiskit_serverless.core.function import (
+    QiskitFunction,
+    RunnableQiskitFunction,
+    RunService,
 )
-from qiskit_serverless.core.function import QiskitFunction
 from qiskit_serverless.utils import JsonSerializable
 from qiskit_serverless.visualizaiton import Widget
 
 
-class BaseClient(JsonSerializable, ABC):
+class BaseClient(JobClient, RunService, JsonSerializable, ABC):
     """
     A client class for specifying custom compute resources.
 
@@ -140,76 +139,27 @@ def get_jobs(self, **kwargs) -> List[Job]:
         )
         return self.jobs(**kwargs)
 
-    @abstractmethod
-    def run(
-        self,
-        program: Union[QiskitFunction, str],
-        arguments: Optional[Dict[str, Any]] = None,
-        config: Optional[Configuration] = None,
-    ) -> Job:
-        """Execute a program as a async job.
-
-        Example:
-            >>> serverless = QiskitServerless()
-            >>> program = QiskitFunction(
-            >>>     "job.py",
-            >>>     arguments={"arg1": "val1"},
-            >>>     dependencies=["requests"]
-            >>> )
-            >>> job = serverless.run(program)
-            >>> # <Job | ...>
-
-        Args:
-            arguments: arguments to run program with
-            program: Program object
-
-        Returns:
-            Job
-        """
-
-    @abstractmethod
-    def status(self, job_id: str) -> str:
-        """Check status."""
-
-    @abstractmethod
-    def stop(
-        self, job_id: str, service: Optional[QiskitRuntimeService] = None
-    ) -> Union[str, bool]:
-        """Stops job/program."""
-
-    @abstractmethod
-    def result(self, job_id: str) -> Any:
-        """Return results."""
-
-    @abstractmethod
-    def logs(self, job_id: str) -> str:
-        """Return logs."""
-
-    @abstractmethod
-    def filtered_logs(self, job_id: str, **kwargs) -> str:
-        """Return filtered logs."""
-
     #########################
     ####### Functions #######
     #########################
 
     @abstractmethod
-    def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
+    def upload(self, program: QiskitFunction) -> Optional[RunnableQiskitFunction]:
         """Uploads program."""
 
     @abstractmethod
-    def functions(self, **kwargs) -> List[QiskitFunction]:
+    def functions(self, **kwargs) -> List[RunnableQiskitFunction]:
         """Returns list of available programs."""
 
     @abstractmethod
     def function(
         self, title: str, provider: Optional[str] = None
-    ) -> Optional[QiskitFunction]:
+    ) -> Optional[RunnableQiskitFunction]:
         """Returns program based on parameters."""
 
     def get(
         self, title: str, provider: Optional[str] = None
-    ) -> Optional[QiskitFunction]:
+    ) -> Optional[RunnableQiskitFunction]:
         """Returns program based on parameters."""
         warnings.warn(
             "`get` method has been deprecated. "
@@ -219,7 +169,7 @@ def get(
         )
         return self.function(title, provider=provider)
 
-    def list(self, **kwargs) -> List[QiskitFunction]:
+    def list(self, **kwargs) -> List[RunnableQiskitFunction]:
         """Returns list of available programs."""
         warnings.warn(
             "`list` method has been deprecated. "

client/qiskit_serverless/core/clients/local_client.py

@@ -48,7 +48,7 @@
     Job,
     Configuration,
 )
-from qiskit_serverless.core.function import QiskitFunction
+from qiskit_serverless.core.function import QiskitFunction, RunnableQiskitFunction
 from qiskit_serverless.exception import QiskitServerlessException
 from qiskit_serverless.serializers.program_serializers import (
     QiskitObjectsEncoder,
@@ -163,7 +163,7 @@ def filtered_logs(self, job_id: str, **kwargs):
     ####### Functions #######
     #########################
 
-    def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
+    def upload(self, program: QiskitFunction) -> Optional[RunnableQiskitFunction]:
         # check if entrypoint exists
         if not os.path.exists(os.path.join(program.working_dir, program.entrypoint)):
             raise QiskitServerlessException(
@@ -182,14 +182,14 @@ def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
             "client": self,
         }
         self._patterns.append(pattern)
-        return QiskitFunction.from_json(pattern)
+        return RunnableQiskitFunction.from_json(pattern)
 
-    def functions(self, **kwargs) -> List[QiskitFunction]:
+    def functions(self, **kwargs) -> List[RunnableQiskitFunction]:
         """Returns list of programs."""
         return [QiskitFunction.from_json(program) for program in self._patterns]
 
     def function(
         self, title: str, provider: Optional[str] = None
-    ) -> Optional[QiskitFunction]:
+    ) -> Optional[RunnableQiskitFunction]:
         functions = {function.title: function for function in self.functions()}
         return functions.get(title)

client/qiskit_serverless/core/clients/ray_client.py

@@ -42,7 +42,7 @@
     Configuration,
     Job,
 )
-from qiskit_serverless.core.function import QiskitFunction
+from qiskit_serverless.core.function import QiskitFunction, RunnableQiskitFunction
 from qiskit_serverless.serializers.program_serializers import (
     QiskitObjectsEncoder,
 )
@@ -157,16 +157,16 @@ def filtered_logs(self, job_id: str, **kwargs) -> str:
     ####### Functions #######
     #########################
 
-    def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
+    def upload(self, program: QiskitFunction) -> Optional[RunnableQiskitFunction]:
         """Uploads program."""
         raise NotImplementedError("Upload is not available for RayClient.")
 
-    def functions(self, **kwargs) -> List[QiskitFunction]:
+    def functions(self, **kwargs) -> List[RunnableQiskitFunction]:
         """Returns list of available programs."""
         raise NotImplementedError("get_programs is not available for RayClient.")
 
     def function(
         self, title: str, provider: Optional[str] = None
-    ) -> Optional[QiskitFunction]:
+    ) -> Optional[RunnableQiskitFunction]:
         """Returns program based on parameters."""
         raise NotImplementedError("get_program is not available for RayClient.")

client/qiskit_serverless/core/clients/serverless_client.py

@@ -54,7 +54,12 @@
     Job,
     Configuration,
 )
-from qiskit_serverless.core.function import QiskitFunction
+from qiskit_serverless.core.function import (
+    QiskitFunction,
+    RunService,
+    RunnableQiskitFunction,
+)
+
 from qiskit_serverless.exception import QiskitServerlessException
 from qiskit_serverless.utils.json import (
     safe_json_request_as_dict,
@@ -316,7 +321,7 @@ def filtered_logs(self, job_id: str, **kwargs):
     ####### Functions #######
     #########################
 
-    def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
+    def upload(self, program: QiskitFunction) -> Optional[RunnableQiskitFunction]:
         tracer = trace.get_tracer("client.tracer")
         with tracer.start_as_current_span("job.run") as span:
             span.set_attribute("program", program.title)
@@ -325,12 +330,12 @@ def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
             if program.image is not None:
                 # upload function with custom image
                 function_uploaded = _upload_with_docker_image(
-                    program=program, url=url, token=self.token, span=span
+                    program=program, url=url, token=self.token, span=span, client=self
                 )
             elif program.entrypoint is not None:
                 # upload funciton with artifact
                 function_uploaded = _upload_with_artifact(
-                    program=program, url=url, token=self.token, span=span
+                    program=program, url=url, token=self.token, span=span, client=self
                 )
             else:
                 raise QiskitServerlessException(
@@ -339,7 +344,7 @@ def upload(self, program: QiskitFunction) -> Optional[QiskitFunction]:
 
         return function_uploaded
 
-    def functions(self, **kwargs) -> List[QiskitFunction]:
+    def functions(self, **kwargs) -> List[RunnableQiskitFunction]:
         """Returns list of available programs."""
         tracer = trace.get_tracer("client.tracer")
         with tracer.start_as_current_span("program.list"):
@@ -353,19 +358,19 @@ def functions(self, **kwargs) -> List[QiskitFunction]:
             )
 
         return [
-            QiskitFunction(
-                program.get("title"),
+            RunnableQiskitFunction(
+                client=self,
+                title=program.get("title"),
                 provider=program.get("provider", None),
                 raw_data=program,
-                client=self,
                 description=program.get("description"),
             )
             for program in response_data
         ]
 
     def function(
         self, title: str, provider: Optional[str] = None
-    ) -> Optional[QiskitFunction]:
+    ) -> Optional[RunnableQiskitFunction]:
         """Returns program based on parameters."""
         provider, title = format_provider_name_and_title(
             request_provider=provider, title=title
@@ -381,11 +386,11 @@ def function(
                     timeout=REQUESTS_TIMEOUT,
                 )
             )
-            return QiskitFunction(
-                response_data.get("title"),
+            return RunnableQiskitFunction(
+                client=self,
+                title=response_data.get("title"),
                 provider=response_data.get("provider", None),
                 raw_data=response_data,
-                client=self,
             )
 
     #####################
@@ -484,8 +489,8 @@ def save_account(
 
 
 def _upload_with_docker_image(
-    program: QiskitFunction, url: str, token: str, span: Any
-) -> QiskitFunction:
+    program: QiskitFunction, url: str, token: str, span: Any, client: RunService
+) -> RunnableQiskitFunction:
     """Uploads function with custom docker image.
 
     Args:
@@ -517,12 +522,13 @@ def _upload_with_docker_image(
     program_provider = response_data.get("provider", "na")
     span.set_attribute("program.title", program_title)
     span.set_attribute("program.provider", program_provider)
-    return QiskitFunction.from_json(response_data)
+    response_data["client"] = client
+    return RunnableQiskitFunction.from_json(response_data)
 
 
 def _upload_with_artifact(
-    program: QiskitFunction, url: str, token: str, span: Any
-) -> QiskitFunction:
+    program: QiskitFunction, url: str, token: str, span: Any, client: RunService
+) -> RunnableQiskitFunction:
     """Uploads function with artifact.
 
     Args:
@@ -582,11 +588,10 @@ def _upload_with_artifact(
                     timeout=REQUESTS_TIMEOUT,
                 )
             )
-            program_title = response_data.get("title", "na")
-            program_provider = response_data.get("provider", "na")
-            span.set_attribute("program.title", program_title)
-            span.set_attribute("program.provider", program_provider)
-            response_function = QiskitFunction.from_json(response_data)
+            span.set_attribute("program.title", response_data.get("title", "na"))
+            span.set_attribute("program.provider", response_data.get("provider", "na"))
+            response_data["client"] = client
+            response_function = RunnableQiskitFunction.from_json(response_data)
     except Exception as error:  # pylint: disable=broad-exception-caught
         raise QiskitServerlessException from error
     finally: