synapse-sdk 1.0.0a4__tar.gz → 1.0.0a5__tar.gz

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a4
+Version: 1.0.0a5
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT
@@ -10,11 +10,13 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: boto3
 Requires-Dist: click
+Requires-Dist: cookiecutter
 Requires-Dist: requests
 Requires-Dist: tqdm
 Requires-Dist: python-dotenv
 Requires-Dist: pyyaml
 Requires-Dist: pydantic
-Requires-Dist: ray[all]
+Provides-Extra: all
+Requires-Dist: ray[all]; extra == "all"
 
 This is the SDK to develop synapse plugins

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/pyproject.toml

@@ -15,15 +15,23 @@ classifiers = [
 dependencies = [
     'boto3',
     'click',
+    'cookiecutter',
     'requests',
     'tqdm',
     'python-dotenv',
     'pyyaml',
     'pydantic',
-    'ray[all]'
 ]
 dynamic = ['version']
 
+[project.optional-dependencies]
+all = [
+    'ray[all]'
+]
+
+[project.scripts]
+synapse = 'synapse_sdk.cli:cli'
+
 [tool.setuptools_scm]
 
 [tool.ruff]

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/requirements.txt

@@ -1,5 +1,6 @@
 boto3
 click
+cookiecutter
 requests
 tqdm
 pyyaml

synapse_sdk-1.0.0a5/synapse_sdk/cli/__init__.py

@@ -0,0 +1,11 @@
+import click
+
+from synapse_sdk.cli.create_plugin import create_plugin
+
+
+@click.group()
+def cli():
+    pass
+
+
+cli.add_command(create_plugin)

synapse_sdk-1.0.0a5/synapse_sdk/cli/create_plugin.py

@@ -0,0 +1,10 @@
+from pathlib import Path
+
+import click
+from cookiecutter.main import cookiecutter
+
+
+@click.command()
+def create_plugin():
+    project_root = Path(__file__).parent.parent
+    cookiecutter(str(project_root / 'plugins/templates'))

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/categories/base.py

@@ -11,10 +11,9 @@ from ray.dashboard.modules.job.sdk import JobSubmissionClient
 from synapse_sdk.clients.backend import BackendClient
 from synapse_sdk.loggers import ConsoleLogger, BackendLogger
 from synapse_sdk.plugins.enums import RunMethod
+from synapse_sdk.plugins.models import PluginRelease
 from synapse_sdk.plugins.upload import build_and_upload, archive_and_upload, download_and_upload
-from synapse_sdk.plugins.utils import get_plugin_checksum
 from synapse_sdk.utils.module_loading import import_string
-from synapse_sdk.utils.storage import get_storage
 
 
 class Action:
@@ -23,6 +22,7 @@ class Action:
     method = None
     params = None
     plugin_config = None
+    plugin_release = None
     config = None
     client = None
     logger = None
@@ -42,23 +42,14 @@ class Action:
     def __init__(self, params, plugin_config, envs=None, job_id=None, direct=False, debug=False):
         self.params = params
         self.plugin_config = plugin_config
-        self.config = plugin_config['actions'][self.name]
+        self.plugin_release = PluginRelease(config=plugin_config)
+        self.config = self.plugin_release.get_action_config(self.name)
         self.job_id = job_id
         self.direct = direct
         self.debug = debug
         self.envs = {**envs, **self.get_default_envs()} if envs else self.get_default_envs()
         self.set_logger()
 
-    @cached_property
-    def plugin_id(self):
-        code = self.plugin_config['code']
-        version = self.plugin_config['version']
-        return f'{code}@{version}'
-
-    @cached_property
-    def plugin_checksum(self):
-        return get_plugin_checksum(self.plugin_id)
-
     @cached_property
     def entrypoint(self):
         return import_string(self.config['entrypoint'])
@@ -79,8 +70,7 @@ class Action:
                 plugin_url = archive_and_upload(plugin_path, self.plugin_storage_url)
             self.envs['SYNAPSE_DEBUG_PLUGIN_PATH'] = plugin_url
             return plugin_url
-        storage = get_storage(self.plugin_storage_url)
-        return storage.get_url(f'{self.plugin_checksum}.zip')
+        return self.plugin_release.get_url(self.plugin_storage_url)
 
     @property
     def debug_modules(self):
@@ -115,7 +105,7 @@ class Action:
         }
 
         if self.debug:
-            runtime_env['pip'] += self.debug_modules
+            runtime_env['pip'] = self.debug_modules + runtime_env['pip']
 
         # 맨 마지막에 진행되어야 함
         runtime_env['env_vars'] = self.envs
@@ -183,7 +173,7 @@ class Action:
         path = self.params.pop('path', '')
         method = self.params.pop('method')
 
-        url = os.path.join(self.envs['RAY_SERVE_ADDRESS'], self.plugin_checksum, path)
+        url = self.plugin_release.get_serve_url(self.envs['RAY_SERVE_ADDRESS'], path)
         response = getattr(requests, method)(url, **self.params)
         # TODO ok response가 아닌 경우 대응하기
         return response.json()

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/categories/neural_net/actions/deployment.py

@@ -19,5 +19,5 @@ class DeploymentAction(Action):
 
     def run(self):
         deployment = self.get_deployment()
-        serve.delete(self.plugin_id)
-        serve.run(deployment.bind(), name=self.plugin_id, route_prefix=f'/{self.plugin_checksum}')
+        serve.delete(self.plugin_release.code)
+        serve.run(deployment.bind(), name=self.plugin_release.code, route_prefix=f'/{self.plugin_release.checksum}')

synapse_sdk-1.0.0a5/synapse_sdk/plugins/categories/neural_net/templates/config.yaml

@@ -0,0 +1,12 @@
+actions:
+  train:
+    dataset: datamaker
+    entrypoint: plugin.train.train
+  deployment:
+    entrypoint: plugin.inference.InferenceDeployment
+  inference:
+    method: restapi
+    endpoints:
+      - method: get
+  test:
+    entrypoint: plugin.test.test

synapse_sdk-1.0.0a5/synapse_sdk/plugins/categories/neural_net/templates/plugin/inference.py

@@ -0,0 +1,4 @@
+class InferenceDeployment:
+    def __call__(self):
+        message = 'hello datamaker'
+        return {'message': message}

synapse_sdk-1.0.0a5/synapse_sdk/plugins/categories/neural_net/templates/plugin/test.py

@@ -0,0 +1,2 @@
+def test(run):
+    return 'this is neural net test!'

synapse_sdk-1.0.0a5/synapse_sdk/plugins/categories/neural_net/templates/plugin/train.py

@@ -0,0 +1,14 @@
+import time
+
+
+def train(run, dataset, hyperparameter, checkpoint=None):
+    count_iterations = hyperparameter['iterations']
+
+    for i in range(1, count_iterations + 1):
+        time.sleep(0.5)
+        loss = float(round((count_iterations - i) / count_iterations, 2))
+        miou = 1 - loss
+        run.log_metric('iteration', i, loss=loss, miou=miou)
+        run.set_progress(i, count_iterations, category='iteration')
+
+    return {'weight': '/tmp/agent/test/a.txt', 'config': '/tmp/agent/test/b.txt'}

synapse_sdk-1.0.0a5/synapse_sdk/plugins/categories/templates.py

@@ -0,0 +1,32 @@
+import shutil
+from pathlib import Path
+
+import yaml
+
+
+def copy_project_category_template(category):
+    copy_plugin(category)
+    merge_config(category)
+
+
+def copy_plugin(category):
+    template_path = Path(__file__).parent / category / 'templates' / 'plugin'
+    if not template_path.exists():
+        return
+
+    output_path = Path('plugin')
+    shutil.copytree(template_path, output_path, dirs_exist_ok=True)
+
+
+def merge_config(category):
+    config_path = Path(__file__).parent / category / 'templates' / 'config.yaml'
+    if not config_path.exists():
+        return
+
+    config_base_path = Path('config.yaml')
+
+    config_base = yaml.safe_load(config_base_path.read_text(encoding='utf-8'))
+    config = yaml.safe_load(config_path.read_text(encoding='utf-8'))
+
+    config_base.update(config)
+    config_base_path.write_text(yaml.dump(config_base, sort_keys=False), encoding='utf-8')

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/cli/publish.py

@@ -4,8 +4,8 @@ from pathlib import Path
 import click
 
 from synapse_sdk.clients.backend import BackendClient
+from synapse_sdk.plugins.models import PluginRelease
 from synapse_sdk.plugins.upload import archive
-from synapse_sdk.plugins.utils import read_config
 
 
 @click.command()
@@ -17,20 +17,20 @@ from synapse_sdk.plugins.utils import read_config
 def publish(ctx, host, user_token, tenant, debug_modules):
     debug = ctx.obj['DEBUG']
 
-    config = read_config()
+    plugin_release = PluginRelease()
 
     source_path = Path('./')
     archive_path = source_path / 'dist' / 'archive.zip'
     archive(source_path, archive_path)
 
-    data = {'plugin': config['code'], 'file': str(archive_path), 'debug': debug}
+    data = {'plugin': plugin_release.plugin, 'file': str(archive_path), 'debug': debug}
     if debug:
         data['debug_meta'] = json.dumps({'modules': debug_modules.split(',')})
 
     client = BackendClient(host, user_token, tenant=tenant)
     client.create_plugin_release(data)
     click.secho(
-        f'Successfully published "{config["name"]}" ({config["code"]}@{config["version"]}) to synapse backend!',
+        f'Successfully published "{plugin_release.name}" ({plugin_release.code}) to synapse backend!',
         fg='green',
         bold=True,
     )

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/cli/run.py

@@ -4,7 +4,8 @@ import click
 
 from synapse_sdk.clients.agent import AgentClient
 from synapse_sdk.clients.backend import BackendClient
-from synapse_sdk.plugins.utils import get_action, read_config
+from synapse_sdk.plugins.models import PluginRelease
+from synapse_sdk.plugins.utils import get_action
 
 
 @click.command()
@@ -41,7 +42,9 @@ def run_by_script(action, params, job_id, direct, debug):
 
 def run_by_agent(action, params, job_id, agent_host, agent_token, user_token, tenant, debug):
     client = AgentClient(agent_host, agent_token, user_token, tenant)
-    data = {'action': action, 'params': params, 'job_id': job_id}
+    data = {'action': action, 'params': params}
+    if job_id:
+        data['job_id'] = job_id
     if debug:
         data.update({
             'plugin_path': os.getcwd(),
@@ -49,16 +52,16 @@ def run_by_agent(action, params, job_id, agent_host, agent_token, user_token, te
         })
         result = client.run_debug_plugin_release(data=data)
     else:
-        config = read_config()
-        result = client.run_plugin_release(code=f'{config["code"]}@{config["version"]}', data=data)
+        plugin_release = PluginRelease()
+        result = client.run_plugin_release(code=plugin_release.code, data=data)
 
     click.echo(result)
 
 
 def run_by_backend(action, params, agent, host, user_token, tenant):
     client = BackendClient(host, user_token, tenant=tenant)
-    config = read_config()
-    data = {'agent': agent, 'version': config['version'], 'action': action, 'params': params}
-    result = client.run_plugin(config['code'], data=data)
+    plugin_release = PluginRelease()
+    data = {'agent': agent, 'version': plugin_release.version, 'action': action, 'params': params}
+    result = client.run_plugin(plugin_release.plugin, data=data)
 
     click.echo(result)

synapse_sdk-1.0.0a5/synapse_sdk/plugins/models.py

@@ -0,0 +1,66 @@
+import os
+from functools import cached_property
+from typing import Dict, Any
+
+from synapse_sdk.plugins.utils import read_plugin_config
+from synapse_sdk.utils.storage import get_storage
+from synapse_sdk.utils.string import hash_text
+
+
+class PluginRelease:
+    config: Dict[str, Any]
+
+    def __init__(self, config=None, plugin_path=None):
+        if config:
+            self.config = config
+        else:
+            self.config = read_plugin_config(plugin_path=plugin_path)
+
+    @cached_property
+    def plugin(self):
+        return self.config['code']
+
+    @cached_property
+    def version(self):
+        return self.config['version']
+
+    @cached_property
+    def code(self):
+        return f'{self.plugin}@{self.version}'
+
+    @cached_property
+    def category(self):
+        return self.config['category']
+
+    @cached_property
+    def name(self):
+        return self.config['name']
+
+    @cached_property
+    def checksum(self):
+        return hash_text(self.code)
+
+    @cached_property
+    def actions(self):
+        return list(self.config['actions'].keys())
+
+    def setup_runtime_env(self):
+        # TODO ray에 해당 plugin release runtime env 캐싱
+        pass
+
+    def get_action_config(self, action):
+        return self.config['actions'][action]
+
+    def get_url(self, storage_url):
+        storage = get_storage(storage_url)
+        return storage.get_url(f'{self.checksum}.zip')
+
+    def get_serve_url(self, serve_address, path):
+        return os.path.join(serve_address, self.checksum, path)
+
+
+class Job:
+    job_id = None
+
+    def __init__(self, job_id):
+        self.job_id = job_id

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/cookiecutter.json

@@ -0,0 +1,11 @@
+{
+  "category": null,
+  "plugin_name": null,
+  "plugin_code": "{{ cookiecutter.plugin_name.lower().replace(' ', '-') }}",
+  "version": "0.1.0",
+  "description": "This is {{ cookiecutter.plugin_name }} plugin",
+  "__prompts__": {
+    "project_name": "Plugin name",
+    "project_slug": "Unique identifier of your plugin. Lowercase letters, numbers, and hyphens only (e.g., 'my-plugin-name')"
+  }
+}

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/hooks/post_gen_project.py

@@ -0,0 +1,3 @@
+from synapse_sdk.plugins.categories.templates import copy_project_category_template
+
+copy_project_category_template('{{ cookiecutter.category }}')

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/hooks/pre_prompt.py

@@ -0,0 +1,20 @@
+import json
+from pathlib import Path
+from synapse_sdk.plugins.utils import get_plugin_categories
+from synapse_sdk.utils.file import get_dict_from_file
+
+
+def update_config(config):
+    config['category'] = get_plugin_categories()
+    return config
+
+
+def main():
+    cookiecutter_path = Path('cookiecutter.json')
+    config = get_dict_from_file(cookiecutter_path)
+    config = update_config(config)
+    cookiecutter_path.write_text(json.dumps(config, indent=4, ensure_ascii=False), encoding='utf-8')
+
+
+if __name__ == '__main__':
+    main()

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.env.dist

@@ -0,0 +1,24 @@
+# RAY 관련
+RAY_ADDRESS=
+RAY_DASHBOARD_ADDRESS=
+RAY_SERVE_ADDRESS=
+
+# 플러그인 관련
+SYNAPSE_PLUGIN_STORAGE=
+
+# 플러그인 디버그 관련
+SYNAPSE_DEBUG_PLUGIN_PATH=
+SYNAPSE_DEBUG_MODULES=
+
+# 플러그인 PUBLISH 관련
+SYNAPSE_PLUGIN_PUBLISH_HOST=https://synapse.datamaker.io
+SYNAPSE_PLUGIN_PUBLISH_USER_TOKEN=
+SYNAPSE_PLUGIN_PUBLISH_TENANT=
+
+# 플러그인 RUN 관련
+SYNAPSE_PLUGIN_RUN_HOST=
+SYNAPSE_PLUGIN_RUN_AGENT=
+SYNAPSE_PLUGIN_RUN_AGENT_HOST=
+SYNAPSE_PLUGIN_RUN_AGENT_TOKEN=
+SYNAPSE_PLUGIN_RUN_USER_TOKEN=
+SYNAPSE_PLUGIN_RUN_TENANT=

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.gitignore

@@ -0,0 +1,27 @@
+### Example user template template
+# virtualenv
+.venv
+
+# idea
+.DS_Store
+*.pyc
+.idea
+*.iml
+
+# http
+/http/
+
+# general things to ignore
+build/
+dist/
+*.egg-info/
+*.egg
+*.py[cod]
+__pycache__/
+*.so
+*~
+.env
+
+# due to using tox and pytest
+.tox
+.cache

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.pre-commit-config.yaml

@@ -0,0 +1,7 @@
+repos:
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.7.1
+    hooks:
+      - id: ruff
+        args: [ --fix ]
+      - id: ruff-format

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/README.md

@@ -0,0 +1,5 @@
+# 실행방법
+
+```bash
+$ python main.py run --help
+```

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/config.yaml

@@ -0,0 +1,18 @@
+name: {{ cookiecutter.plugin_name }}
+code: {{ cookiecutter.plugin_code }}
+version: {{ cookiecutter.version }}
+readme: README.md
+description: {{ cookiecutter.description }}
+category: {{ cookiecutter.category }}
+actions:
+  train:
+    dataset: datamaker
+    entrypoint: plugin.train.train
+  deployment:
+    entrypoint: plugin.inference.MockNetInference
+  inference:
+    method: restapi
+    endpoints:
+      - method: get
+  test:
+    entrypoint: plugin.test.test

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/pyproject.toml

@@ -0,0 +1,13 @@
+[tool.ruff]
+line-length = 120
+
+[tool.ruff.lint]
+select = ['E', 'F', 'Q']
+ignore = ['W191', 'E111', 'E114', 'E117', 'D206', 'D300', 'Q000', 'Q001', 'Q002', 'Q003', 'COM812', 'COM819', 'ISC001', 'ISC002']
+
+[tool.ruff.lint.pydocstyle]
+convention = 'google'
+
+[tool.ruff.format]
+quote-style = 'single'
+preview = true

synapse_sdk-1.0.0a5/synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/requirements.txt

@@ -0,0 +1 @@
+synapse-sdk

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/upload.py

@@ -1,4 +1,3 @@
-import re
 import subprocess
 import tempfile
 from pathlib import Path
@@ -17,9 +16,9 @@ def download_and_upload(source_url, url):
     storage = get_storage(url)
     with tempfile.TemporaryDirectory() as temp_path:
         file_path = str(download_file(source_url, temp_path))
-        checksum = calculate_checksum(file_path, prefix='dev')
+        checksum = calculate_checksum(file_path)
         # TODO 중복 체크
-        return storage.upload(file_path, f'{checksum}.zip')
+        return storage.upload(file_path, f'dev-{checksum}.zip')
 
 
 def archive_and_upload(source_path, url):
@@ -28,8 +27,8 @@ def archive_and_upload(source_path, url):
     archive_path = dist_path / 'archive.zip'
 
     archive(source_path, archive_path)
-    checksum = calculate_checksum(archive_path, prefix='dev')
-    checksum_archive_path = dist_path / f'{checksum}.zip'
+    checksum = calculate_checksum(archive_path)
+    checksum_archive_path = dist_path / f'dev-{checksum}.zip'
 
     if checksum_archive_path.exists():
         # TODO 실제 스토리지 있는지 확인
@@ -48,8 +47,8 @@ def build_and_upload(source_path, url, virtualenv_path='.venv'):
     archive_path = dist_path / 'archive.zip'
 
     archive(source_path, archive_path)
-    checksum = calculate_checksum(archive_path, prefix='dev')
-    checksum_archive_path = dist_path / f'{checksum}.zip'
+    checksum = calculate_checksum(archive_path)
+    checksum_archive_path = dist_path / f'dev-{checksum}.zip'
 
     if checksum_archive_path.exists():
         # TODO 실제 스토리지 있는지 확인
@@ -70,13 +69,20 @@ def build_and_upload(source_path, url, virtualenv_path='.venv'):
     )
     wheel_path = next(dist_path.glob('*.whl'), None)
 
+    # whl 파일 버전이 동일한 이슈를 해결하기 위해 checksum으로 build명 변경
+    checksum_wheel_path = wheel_path.with_name(change_build_from_whl_name(wheel_path.name, checksum))
+    wheel_path.rename(checksum_wheel_path)
+
     archive_path.rename(checksum_archive_path)
+
     for file_path in dist_path.glob('*.zip'):
         if file_path.name != checksum_archive_path.name:
             file_path.unlink()
-    return storage.upload(str(wheel_path), wheel_path.name)
+    return storage.upload(str(checksum_wheel_path), checksum_wheel_path.name)
 
 
-def change_whl_version(whl_name, new_version):
-    pattern = r'^(?P<distribution>.+?)-(?P<version>[\d\.\w]+(\+[\w\.]+)?)(?P<rest>-.+\.whl)$'
-    return re.sub(pattern, rf'\g<distribution>-{new_version}\g<rest>', whl_name)
+def change_build_from_whl_name(whl_name, build):
+    components = whl_name.split('-')
+    version = components[1].split('+')[0]
+    components[1] = f'{version}+{build}'
+    return '-'.join(components)

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk/plugins/utils.py

@@ -1,8 +1,8 @@
 import json
+from pathlib import Path
 
 from synapse_sdk.plugins.categories.registry import _REGISTERED_ACTIONS, register_actions
 from synapse_sdk.utils.file import get_dict_from_file
-from synapse_sdk.utils.string import hash_text
 
 
 def get_action(action, params_data, *args, **kwargs):
@@ -17,11 +17,11 @@ def get_action(action, params_data, *args, **kwargs):
     config_data = kwargs.pop('config', False)
     if config_data:
         if isinstance(config_data, str):
-            config = get_dict_from_file(config_data)
+            config = read_plugin_config(plugin_path=config_data)
         else:
             config = config_data
     else:
-        config = read_config()
+        config = read_plugin_config()
     category = config['category']
     return get_action_class(category, action)(params, config, *args, **kwargs)
 
@@ -36,9 +36,15 @@ def get_available_actions(category):
     return list(_REGISTERED_ACTIONS[category].keys())
 
 
-def get_plugin_checksum(plugin_id):
-    return hash_text(plugin_id)
+def get_plugin_categories():
+    register_actions()
+    return list(_REGISTERED_ACTIONS.keys())
 
 
-def read_config():
-    return get_dict_from_file('config.yaml')
+def read_plugin_config(plugin_path=None):
+    config_file_name = 'config.yaml'
+    if plugin_path:
+        config_path = Path(plugin_path) / config_file_name
+    else:
+        config_path = config_file_name
+    return get_dict_from_file(config_path)

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: synapse-sdk
-Version: 1.0.0a4
+Version: 1.0.0a5
 Summary: synapse sdk
 Author-email: datamaker <developer@datamaker.io>
 License: MIT
@@ -10,11 +10,13 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: boto3
 Requires-Dist: click
+Requires-Dist: cookiecutter
 Requires-Dist: requests
 Requires-Dist: tqdm
 Requires-Dist: python-dotenv
 Requires-Dist: pyyaml
 Requires-Dist: pydantic
-Requires-Dist: ray[all]
+Provides-Extra: all
+Requires-Dist: ray[all]; extra == "all"
 
 This is the SDK to develop synapse plugins

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk.egg-info/SOURCES.txt

@@ -11,8 +11,11 @@ synapse_sdk/loggers.py
 synapse_sdk.egg-info/PKG-INFO
 synapse_sdk.egg-info/SOURCES.txt
 synapse_sdk.egg-info/dependency_links.txt
+synapse_sdk.egg-info/entry_points.txt
 synapse_sdk.egg-info/requires.txt
 synapse_sdk.egg-info/top_level.txt
+synapse_sdk/cli/__init__.py
+synapse_sdk/cli/create_plugin.py
 synapse_sdk/clients/__init__.py
 synapse_sdk/clients/base.py
 synapse_sdk/clients/exceptions.py
@@ -26,13 +29,14 @@ synapse_sdk/clients/backend/integration.py
 synapse_sdk/clients/backend/ml.py
 synapse_sdk/plugins/__init__.py
 synapse_sdk/plugins/enums.py
-synapse_sdk/plugins/job.py
+synapse_sdk/plugins/models.py
 synapse_sdk/plugins/upload.py
 synapse_sdk/plugins/utils.py
 synapse_sdk/plugins/categories/__init__.py
 synapse_sdk/plugins/categories/base.py
 synapse_sdk/plugins/categories/decorators.py
 synapse_sdk/plugins/categories/registry.py
+synapse_sdk/plugins/categories/templates.py
 synapse_sdk/plugins/categories/data_validation/__init__.py
 synapse_sdk/plugins/categories/data_validation/actions/__init__.py
 synapse_sdk/plugins/categories/data_validation/actions/validation.py
@@ -48,6 +52,11 @@ synapse_sdk/plugins/categories/neural_net/actions/deployment.py
 synapse_sdk/plugins/categories/neural_net/actions/inference.py
 synapse_sdk/plugins/categories/neural_net/actions/test.py
 synapse_sdk/plugins/categories/neural_net/actions/train.py
+synapse_sdk/plugins/categories/neural_net/templates/config.yaml
+synapse_sdk/plugins/categories/neural_net/templates/plugin/__init__.py
+synapse_sdk/plugins/categories/neural_net/templates/plugin/inference.py
+synapse_sdk/plugins/categories/neural_net/templates/plugin/test.py
+synapse_sdk/plugins/categories/neural_net/templates/plugin/train.py
 synapse_sdk/plugins/categories/post_annotation/__init__.py
 synapse_sdk/plugins/categories/post_annotation/actions/__init__.py
 synapse_sdk/plugins/categories/post_annotation/actions/post_annotation.py
@@ -57,6 +66,17 @@ synapse_sdk/plugins/categories/pre_annotation/actions/pre_annotation.py
 synapse_sdk/plugins/cli/__init__.py
 synapse_sdk/plugins/cli/publish.py
 synapse_sdk/plugins/cli/run.py
+synapse_sdk/plugins/templates/cookiecutter.json
+synapse_sdk/plugins/templates/hooks/post_gen_project.py
+synapse_sdk/plugins/templates/hooks/pre_prompt.py
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.env.dist
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.gitignore
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.pre-commit-config.yaml
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/README.md
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/config.yaml
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/pyproject.toml
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/requirements.txt
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/plugin/__init__.py
 synapse_sdk/utils/__init__.py
 synapse_sdk/utils/debug.py
 synapse_sdk/utils/file.py

synapse_sdk-1.0.0a5/synapse_sdk.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+synapse = synapse_sdk.cli:cli

{synapse_sdk-1.0.0a4 → synapse_sdk-1.0.0a5}/synapse_sdk.egg-info/requires.txt

@@ -1,8 +1,11 @@
 boto3
 click
+cookiecutter
 requests
 tqdm
 python-dotenv
 pyyaml
 pydantic
+
+[all]
 ray[all]

synapse_sdk-1.0.0a4/synapse_sdk/plugins/job.py

@@ -1,5 +0,0 @@
-class Job:
-    job_id = None
-
-    def __init__(self, job_id):
-        self.job_id = job_id