synapse-sdk 1.0.0a13__py3-none-any.whl → 1.0.0a15__py3-none-any.whl

synapse_sdk/clients/backend/__init__.py

@@ -8,16 +8,20 @@ class BackendClient(AnnotationClientMixin, DatasetClientMixin, IntegrationClient
     name = 'Backend'
     token = None
     tenant = None
+    agent_token = None
 
-    def __init__(self, base_url, token=None, tenant=None):
+    def __init__(self, base_url, token=None, tenant=None, agent_token=None):
         super().__init__(base_url)
         self.token = token
         self.tenant = tenant
+        self.agent_token = agent_token
 
     def _get_headers(self):
         headers = {}
         if self.token:
-            headers = {'Authorization': f'Token {self.token}'}
+            headers['Authorization'] = f'Token {self.token}'
         if self.tenant:
             headers['SYNAPSE-Tenant'] = f'Token {self.tenant}'
+        if self.agent_token:
+            headers['SYNAPSE-Agent'] = f'Token {self.agent_token}'
         return headers

synapse_sdk/clients/backend/dataset.py

@@ -9,7 +9,7 @@ from synapse_sdk.clients.utils import get_batched_list
 class DatasetClientMixin(BaseClient):
     def list_dataset(self):
         path = 'datasets/'
-        return self._get(path)
+        return self._list(path)
 
     def create_data_file(self, file_path):
         path = 'data_files/'

synapse_sdk/clients/backend/integration.py

@@ -37,7 +37,7 @@ class IntegrationClientMixin(BaseClient):
 
     def list_jobs(self, params=None):
         path = 'jobs/'
-        return self._get(path, params=params)
+        return self._list(path, params=params)
 
     def update_job(self, pk, data):
         path = f'jobs/{pk}/'
@@ -66,6 +66,6 @@ class IntegrationClientMixin(BaseClient):
         path = 'serve_applications/'
         return self._post(path, data=data)
 
-    def list_serve_applications(self, params=None):
+    def list_serve_applications(self, params=None, list_all=False):
         path = 'serve_applications/'
-        return self._get(path, params=params)
+        return self._list(path, params=params, list_all=list_all)

synapse_sdk/clients/backend/ml.py

@@ -5,7 +5,7 @@ from synapse_sdk.clients.utils import get_default_url_conversion
 class MLClientMixin(BaseClient):
     def list_models(self, params=None):
         path = 'models/'
-        return self._get(path, params=params)
+        return self._list(path, params=params)
 
     def get_model(self, pk, params=None, url_conversion=None):
         path = f'models/{pk}/'

synapse_sdk/plugins/categories/base.py

@@ -36,7 +36,7 @@ class Action:
     envs = None
     run = None
 
-    default_envs = [
+    REQUIRED_ENVS = [
         'RAY_ADDRESS',
         'RAY_DASHBOARD_URL',
         'RAY_SERVE_ADDRESS',
@@ -112,7 +112,7 @@ class Action:
         return self.run_class(self.job_id, context)
 
     def get_default_envs(self):
-        return {env: os.environ[env] for env in self.default_envs if env in os.environ}
+        return {env: os.environ[env] for env in self.REQUIRED_ENVS if env in os.environ}
 
     def get_runtime_env(self):
         runtime_env = {

synapse_sdk/plugins/cli/publish.py

@@ -26,7 +26,8 @@ def publish(ctx, host, user_token, tenant, debug_modules):
 
     data = {'plugin': plugin_release.plugin, 'file': str(archive_path), 'debug': debug}
     if debug:
-        data['debug_meta'] = json.dumps({'modules': debug_modules.split(',')})
+        if debug_modules:
+            data['debug_meta'] = json.dumps({'modules': debug_modules.split(',')})
 
     client = BackendClient(host, user_token, tenant=tenant)
     client.create_plugin_release(data)

synapse_sdk/plugins/models.py

@@ -86,28 +86,26 @@ class Run:
     logger = None
     job_id = None
     context = None
+    client = None
 
     def __init__(self, job_id, context):
         self.job_id = job_id
         self.context = context
+        self.client = BackendClient(
+            self.context['envs']['SYNAPSE_PLUGIN_RUN_HOST'],
+            self.context['envs']['SYNAPSE_PLUGIN_RUN_USER_TOKEN'],
+            self.context['envs']['SYNAPSE_PLUGIN_RUN_TENANT'],
+        )
         self.set_logger()
 
     def set_logger(self):
         kwargs = {'progress_categories': self.context['progress_categories']}
+
         if self.job_id:
-            client = BackendClient(
-                self.context['envs']['SYNAPSE_PLUGIN_RUN_HOST'],
-                self.context['envs']['SYNAPSE_PLUGIN_RUN_USER_TOKEN'],
-                self.context['envs']['SYNAPSE_PLUGIN_RUN_TENANT'],
-            )
-            self.logger = BackendLogger(client, self.job_id, **kwargs)
+            self.logger = BackendLogger(self.client, self.job_id, **kwargs)
         else:
             self.logger = ConsoleLogger(**kwargs)
 
-    @property
-    def client(self):
-        return getattr(self.logger, 'client', None)
-
     def set_progress(self, current, total, category=''):
         self.logger.set_progress(current, total, category)
 

synapse_sdk/plugins/utils.py

@@ -1,8 +1,10 @@
 import json
 from pathlib import Path
 
+from synapse_sdk.i18n import gettext as _
 from synapse_sdk.plugins.categories.registry import _REGISTERED_ACTIONS, register_actions
 from synapse_sdk.plugins.enums import PluginCategory
+from synapse_sdk.plugins.exceptions import ActionError
 from synapse_sdk.utils.file import get_dict_from_file
 
 
@@ -48,3 +50,43 @@ def read_plugin_config(plugin_path=None):
     else:
         config_path = config_file_name
     return get_dict_from_file(config_path)
+
+
+def run_plugin(action, params, plugin_config=None, plugin_path=None, modules=None, envs=None, debug=False, **kwargs):
+    from synapse_sdk.plugins.models import PluginRelease
+
+    if not envs:
+        envs = {}
+
+    if debug:
+        if not plugin_path:
+            raise ActionError({'plugin_path': _('디버그 모드에서는 plugin_path는 필수입니다.')})
+
+        if plugin_path.startswith('http'):
+            if not plugin_config:
+                raise ActionError({'config': _('"plugin_path"가 url인 경우에는 "config"가 필수입니다.')})
+            plugin_release = PluginRelease(config=plugin_config)
+        else:
+            plugin_release = PluginRelease(plugin_path=plugin_path)
+            plugin_config = plugin_release.config
+
+        if action not in plugin_release.actions:
+            raise ActionError({'action': _('해당 액션은 존재하지 않습니다.')})
+
+        envs['SYNAPSE_DEBUG_PLUGIN_PATH'] = plugin_path
+        if modules:
+            envs['SYNAPSE_DEBUG_MODULES'] = ','.join(modules)
+    else:
+        if plugin_config is None:
+            raise ActionError({'config': _('플러그인 설정은 필수입니다.')})
+
+    action = get_action(
+        action,
+        params,
+        config=plugin_config,
+        envs=envs,
+        debug=debug,
+        **kwargs,
+    )
+
+    return action.run_action()

{synapse_sdk-1.0.0a13.dist-info → synapse_sdk-1.0.0a15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a13
+Version: 1.0.0a15
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT

{synapse_sdk-1.0.0a13.dist-info → synapse_sdk-1.0.0a15.dist-info}/RECORD

@@ -15,22 +15,22 @@ synapse_sdk/clients/agent/__init__.py,sha256=Pz8_iTbIbnb7ywGJ3feqoZVmO2I3mEbwpWs
 synapse_sdk/clients/agent/core.py,sha256=YicfkO0YvjDDOt1jNWoZ0mokrh8xxKibBL4qF5yOjKs,169
 synapse_sdk/clients/agent/ray.py,sha256=JrwLyVOUDG2yYsbPrxyUtWbM-FWp9B6Bl_GdDby0rt8,1559
 synapse_sdk/clients/agent/service.py,sha256=s7KuPK_DB1nr2VHrigttV1WyFonaGHNrPvU8loRxHcE,478
-synapse_sdk/clients/backend/__init__.py,sha256=SVC7kDO_z_4YqpLDR-O_RrrxJFkWmMhJC8Hl_Mc5Sz4,830
+synapse_sdk/clients/backend/__init__.py,sha256=50MW1CMWGaKnALSv2fOjhJZG55Xb3yrqlKtnSZldwcU,1004
 synapse_sdk/clients/backend/annotation.py,sha256=eZc5EidgR_RfMGwvv1r1_mLkPdRd8e52c4zuuMjMX34,979
-synapse_sdk/clients/backend/dataset.py,sha256=X8R71vff1gLytlEaOnKhBWjVQaZcIG_WYy32new4Vq0,1737
-synapse_sdk/clients/backend/integration.py,sha256=VxhQ91VWAn6eLEptxK561Vt-2bQh97kmCyuznadFMno,2123
-synapse_sdk/clients/backend/ml.py,sha256=SwA4_noRCZ1dVIzyXfrs2zHuRlQZdZt17o1We5QOxl8,1053
+synapse_sdk/clients/backend/dataset.py,sha256=a_svyCKgzF7N99l8V4u4wXD8JxiGuLW9z2EBinnz7b8,1738
+synapse_sdk/clients/backend/integration.py,sha256=Jg_8fEmbrgYXfZZcG8cDtLxR6ugPmnbNhPDyRu_Uib0,2160
+synapse_sdk/clients/backend/ml.py,sha256=lg978cCMcPiLN4ByjhrSlJHlw1_kgZQgaNZNKr9zPsI,1054
 synapse_sdk/clients/ray/__init__.py,sha256=9ZSPXVVxlJ8Wp8ku7l021ENtPjVrGgQDgqifkkVAXgM,187
 synapse_sdk/clients/ray/core.py,sha256=a4wyCocAma2HAm-BHlbZnoVbpfdR-Aad2FM0z6vPFvw,731
 synapse_sdk/clients/ray/serve.py,sha256=rbCpXZYWf0oP8XJ9faa9QFNPYU7h8dltIG8xn9ZconY,907
 synapse_sdk/plugins/__init__.py,sha256=9vsbYhxah4_ofTaG0x0qLFID_raHNkO57Y8A31Ws-lU,222
 synapse_sdk/plugins/enums.py,sha256=s59P6Oz2WAK9IX-kLVhNOvNKYJifKlWBhPpZbc9-ttE,486
 synapse_sdk/plugins/exceptions.py,sha256=Qs7qODp_RRLO9y2otU2T4ryj5LFwIZODvSIXkAh91u0,691
-synapse_sdk/plugins/models.py,sha256=LTD5fUk1Y2i2kNx5MaCYVFvXEyzyheY6cEHpNC248ls,3621
+synapse_sdk/plugins/models.py,sha256=T3ZuTw7BZwMKpz2QGaErnEPetKRD9d4z3qceJkV363o,3541
 synapse_sdk/plugins/upload.py,sha256=VJOotYMayylOH0lNoAGeGHRkLdhP7jnC_A0rFQMvQpQ,3228
-synapse_sdk/plugins/utils.py,sha256=n3s-zFnj4hrGWFtaBTJFbaupI8qUQL6S8_5YcbxOmeY,1482
+synapse_sdk/plugins/utils.py,sha256=UYkwxkmrs0-mRgQB63SkTGD1HpIQEIiHndzTcpdNaF4,2936
 synapse_sdk/plugins/categories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/base.py,sha256=2XMBdYs-feqEFfqae2wM5GqbhnINUQF1EcUMpRmodKg,8091
+synapse_sdk/plugins/categories/base.py,sha256=Q14bnUTdPLeNeFodcOBMr9Z0d8LM1VRIq-pl8vTKgTc,8093
 synapse_sdk/plugins/categories/decorators.py,sha256=Gw6T-UHwpCKrSt596X-g2sZbY_Z1zbbogowClj7Pr5Q,518
 synapse_sdk/plugins/categories/registry.py,sha256=KdQR8SUlLT-3kgYzDNWawS1uJnAhrcw2j4zFaTpilRs,636
 synapse_sdk/plugins/categories/templates.py,sha256=FF5FerhkZMeW1YcKLY5cylC0SkWSYdJODA_Qcm4OGYQ,887
@@ -76,7 +76,7 @@ synapse_sdk/plugins/categories/smart_tool/templates/config.yaml,sha256=3jxW8daip
 synapse_sdk/plugins/categories/smart_tool/templates/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 synapse_sdk/plugins/categories/smart_tool/templates/plugin/auto_label.py,sha256=eevNg0nOcYFR4z_L_R-sCvVOYoLWSAH1jwDkAf3YCjY,320
 synapse_sdk/plugins/cli/__init__.py,sha256=LPtUO0jqkhKq6xR1grpse7da2R6OoT_BeDyCNyUY0T4,380
-synapse_sdk/plugins/cli/publish.py,sha256=aJ7eWdw_c0stIoBY4jWRpeGP0h5qUJWTFUHD6jNs6aM,1200
+synapse_sdk/plugins/cli/publish.py,sha256=rhUtJsKhYyeDTQlul8mQthfMeNuQCou64faHyaP9y1c,1230
 synapse_sdk/plugins/cli/run.py,sha256=lw1KbsL-xTGllF4NtD2cq-Rh6HMbhi-sO862_Ds-sUo,2330
 synapse_sdk/plugins/templates/cookiecutter.json,sha256=NxOWk9A_v1pO0Ny4IYT9Cj5iiJ16--cIQrGC67QdR0I,396
 synapse_sdk/plugins/templates/hooks/post_gen_project.py,sha256=jqlYkY1O2TxIR-Vh3gnwILYy8k-D39Xx66d2KNQVMCs,147
@@ -103,9 +103,9 @@ synapse_sdk/utils/pydantic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJ
 synapse_sdk/utils/pydantic/config.py,sha256=1vYOcUI35GslfD1rrqhFkNXXJOXt4IDqOPSx9VWGfNE,123
 synapse_sdk/utils/pydantic/errors.py,sha256=0v0T12eQBr1KrFiEOBu6KMaPK4aPEGEC6etPJGoR5b4,1061
 synapse_sdk/utils/pydantic/validators.py,sha256=G47P8ObPhsePmd_QZDK8EdPnik2CbaYzr_N4Z6En8dc,193
-synapse_sdk-1.0.0a13.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0a13.dist-info/METADATA,sha256=KRozKnQDQqVd48GRU_jMUTMijCrWUmUuVv2YmPjPhWo,1049
-synapse_sdk-1.0.0a13.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-synapse_sdk-1.0.0a13.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0a13.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0a13.dist-info/RECORD,,
+synapse_sdk-1.0.0a15.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0a15.dist-info/METADATA,sha256=Wh0N4QJ0Hy4XMeV9YUvIHoQUGRNYklUvVnPMv9E7Bbs,1049
+synapse_sdk-1.0.0a15.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+synapse_sdk-1.0.0a15.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0a15.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0a15.dist-info/RECORD,,