File size: 2,582 Bytes
1b7e88c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from omagent_core.engine.authorization_client import AuthorizationClient
from omagent_core.engine.configuration.configuration import Configuration
from omagent_core.engine.integration_client import IntegrationClient
from omagent_core.engine.metadata_client import MetadataClient
from omagent_core.engine.orkes.orkes_authorization_client import \
OrkesAuthorizationClient
from omagent_core.engine.orkes.orkes_integration_client import \
OrkesIntegrationClient
from omagent_core.engine.orkes.orkes_metadata_client import OrkesMetadataClient
from omagent_core.engine.orkes.orkes_prompt_client import OrkesPromptClient
from omagent_core.engine.orkes.orkes_scheduler_client import \
OrkesSchedulerClient
from omagent_core.engine.orkes.orkes_secret_client import OrkesSecretClient
from omagent_core.engine.orkes.orkes_task_client import OrkesTaskClient
from omagent_core.engine.orkes.orkes_workflow_client import OrkesWorkflowClient
from omagent_core.engine.prompt_client import PromptClient
from omagent_core.engine.scheduler_client import SchedulerClient
from omagent_core.engine.secret_client import SecretClient
from omagent_core.engine.task_client import TaskClient
from omagent_core.engine.workflow.executor.workflow_executor import \
WorkflowExecutor
from omagent_core.engine.workflow_client import WorkflowClient
from omagent_core.utils.container import container
class OrkesClients:
def __init__(self, configuration: Configuration = None):
if configuration is None:
configuration = container.conductor_config
self.configuration = configuration
def get_workflow_client(self) -> WorkflowClient:
return OrkesWorkflowClient(self.configuration)
def get_authorization_client(self) -> AuthorizationClient:
return OrkesAuthorizationClient(self.configuration)
def get_metadata_client(self) -> MetadataClient:
return OrkesMetadataClient(self.configuration)
def get_scheduler_client(self) -> SchedulerClient:
return OrkesSchedulerClient(self.configuration)
def get_secret_client(self) -> SecretClient:
return OrkesSecretClient(self.configuration)
def get_task_client(self) -> TaskClient:
return OrkesTaskClient(self.configuration)
def get_integration_client(self) -> IntegrationClient:
return OrkesIntegrationClient(self.configuration)
def get_workflow_executor(self) -> WorkflowExecutor:
return WorkflowExecutor(self.configuration)
def get_prompt_client(self) -> PromptClient:
return OrkesPromptClient(self.configuration)
|