synapse-sdk 1.0.0a8__py3-none-any.whl → 1.0.0a9__py3-none-any.whl

synapse_sdk/clients/agent/__init__.py

@@ -1,21 +1,23 @@
+from synapse_sdk.clients.agent.core import CoreClientMixin
 from synapse_sdk.clients.agent.service import ServiceClientMixin
 
 
-class AgentClient(ServiceClientMixin):
+class AgentClient(CoreClientMixin, ServiceClientMixin):
     name = 'Agent'
     agent_token = None
     user_token = None
     tenant = None
 
-    def __init__(self, base_url, agent_token, user_token, tenant):
+    def __init__(self, base_url, agent_token, user_token=None, tenant=None):
         super().__init__(base_url)
         self.agent_token = agent_token
         self.user_token = user_token
         self.tenant = tenant
 
     def _get_headers(self):
-        return {
-            'Authorization': self.agent_token,
-            'SYNAPSE-User': f'Token {self.user_token}',
-            'SYNAPSE-Tenant': f'Token {self.tenant}',
-        }
+        headers = {'Authorization': self.agent_token}
+        if self.user_token:
+            headers['SYNAPSE-User'] = f'Token {self.user_token}'
+        if self.tenant:
+            headers['SYNAPSE-Tenant'] = f'Token {self.tenant}'
+        return headers

synapse_sdk/clients/agent/core.py

@@ -0,0 +1,7 @@
+from synapse_sdk.clients.base import BaseClient
+
+
+class CoreClientMixin(BaseClient):
+    def health_check(self):
+        path = 'health/'
+        return self._get(path)

synapse_sdk/clients/backend/__init__.py

@@ -9,14 +9,15 @@ class BackendClient(AnnotationClientMixin, DatasetClientMixin, IntegrationClient
     token = None
     tenant = None
 
-    def __init__(self, base_url, token, tenant=None):
+    def __init__(self, base_url, token=None, tenant=None):
         super().__init__(base_url)
         self.token = token
-        if tenant:
-            self.tenant = tenant
+        self.tenant = tenant
 
     def _get_headers(self):
-        headers = {'Authorization': f'Token {self.token}'}
+        headers = {}
+        if self.token:
+            headers = {'Authorization': f'Token {self.token}'}
         if self.tenant:
             headers['SYNAPSE-Tenant'] = f'Token {self.tenant}'
         return headers

synapse_sdk/clients/backend/integration.py

@@ -2,6 +2,10 @@ from synapse_sdk.clients.base import BaseClient
 
 
 class IntegrationClientMixin(BaseClient):
+    def health_check_agent(self, token):
+        path = f'agents/{token}/connect/'
+        return self._post(path)
+
     def get_plugin(self, pk):
         path = f'plugins/{pk}/'
         return self._get(path)

synapse_sdk/clients/base.py

@@ -50,7 +50,10 @@ class BaseClient:
         except requests.ConnectionError:
             raise ClientError(408, f'{self.name} is not responding')
 
-        return response.json()
+        try:
+            return response.json()
+        except ValueError:
+            return response.text
 
     def _get(self, path, url_conversion=None, **kwargs):
         response = self._request('get', path, **kwargs)

synapse_sdk/plugins/categories/base.py

@@ -74,7 +74,7 @@ class Action:
     @property
     def plugin_url(self):
         if self.debug:
-            plugin_path = self.envs.get('SYNAPSE_DEBUG_PLUGIN_PATH', '.')
+            plugin_path = self.envs.get('SYNAPSE_DEBUG_PLUGIN_PATH') or '.'
             if plugin_path.startswith('https://'):  # TODO ray에서 지원하는 remote uri 형식 (https, s3, gs) 모두 지원
                 plugin_url = plugin_path
             elif plugin_path.startswith('http://'):

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

@@ -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/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/main.py

@@ -0,0 +1,4 @@
+from synapse_sdk import plugins
+
+if __name__ == '__main__':
+    plugins.init()

{synapse_sdk-1.0.0a8.dist-info → synapse_sdk-1.0.0a9.dist-info}/METADATA

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

{synapse_sdk-1.0.0a8.dist-info → synapse_sdk-1.0.0a9.dist-info}/RECORD

@@ -3,15 +3,16 @@ synapse_sdk/loggers.py,sha256=avoceBBu9VuSNS_CC3M5X8rWmAlk5xCTYEAVqxAXzBM,4008
 synapse_sdk/cli/__init__.py,sha256=WmYGW1qZEXXIGJe3SGr8QjOStY4svuZKK1Lp_aPvtPs,140
 synapse_sdk/cli/create_plugin.py,sha256=egbW_92WwxfHz50Gy4znX5Bf5fxDdQj3GFyd0l3Y3SY,228
 synapse_sdk/clients/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/clients/base.py,sha256=HfUJDP5FerRkiCJ0eaAZvs4F_kv73PtZ5FSnz_lKV0o,3069
+synapse_sdk/clients/base.py,sha256=vVt6Qi3R3lVnzCTD0WMRDYw99u2d1SMAqDXgkcvq4VM,3146
 synapse_sdk/clients/exceptions.py,sha256=ylv7x10eOp4aA3a48jwonnvqvkiYwzJYXjkVkRTAjwk,220
 synapse_sdk/clients/utils.py,sha256=8pPJTdzHiRPSbZMoQYHAgR2BAMO6u_R_jMV6a2p34iQ,392
-synapse_sdk/clients/agent/__init__.py,sha256=n4exZwc8gwuH6zqZUhstcD_sqIdm-ZoPvUcAxnCrnOQ,609
+synapse_sdk/clients/agent/__init__.py,sha256=k5cGnN-dUAYYfeQDiRW4XEFWUgwSH9mwJFcvjzgwpOc,768
+synapse_sdk/clients/agent/core.py,sha256=YicfkO0YvjDDOt1jNWoZ0mokrh8xxKibBL4qF5yOjKs,169
 synapse_sdk/clients/agent/service.py,sha256=NnRu0XPVRqry6PWN_8-3iuzdXx20zO1lPXw-yVWQbdg,1433
-synapse_sdk/clients/backend/__init__.py,sha256=850EjxNoGWXJpVZLRIaajrymBuW0MXiP3M3onRHDGbQ,800
+synapse_sdk/clients/backend/__init__.py,sha256=SVC7kDO_z_4YqpLDR-O_RrrxJFkWmMhJC8Hl_Mc5Sz4,830
 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=wYfxRI2miZIwC1n7JoVRuSeq1cw5FkNIpKMxcMjj-b0,1628
+synapse_sdk/clients/backend/integration.py,sha256=Z0Vvhp4NsbP4W9esTH_0SDrPkzHy84t1joSinvtjWgI,1744
 synapse_sdk/clients/backend/ml.py,sha256=SwA4_noRCZ1dVIzyXfrs2zHuRlQZdZt17o1We5QOxl8,1053
 synapse_sdk/plugins/__init__.py,sha256=9vsbYhxah4_ofTaG0x0qLFID_raHNkO57Y8A31Ws-lU,222
 synapse_sdk/plugins/enums.py,sha256=s59P6Oz2WAK9IX-kLVhNOvNKYJifKlWBhPpZbc9-ttE,486
@@ -20,7 +21,7 @@ synapse_sdk/plugins/models.py,sha256=pzzJl9zKL8v2XjBQC0UR-W9rG4Vyp95DzJ9MMG2W5L0
 synapse_sdk/plugins/upload.py,sha256=SSCHUR4HJ_mdLN7mUkk5UlXmL9028SKP6tpBS2rpf0A,3175
 synapse_sdk/plugins/utils.py,sha256=n3s-zFnj4hrGWFtaBTJFbaupI8qUQL6S8_5YcbxOmeY,1482
 synapse_sdk/plugins/categories/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-synapse_sdk/plugins/categories/base.py,sha256=QBC96T0VIe-Lp63itjh0E13dgRriU7FT4_xZw7-7Zw8,7060
+synapse_sdk/plugins/categories/base.py,sha256=8WhcLa406V3rR31byhueQssRKeJsoO-kFWGzpre7O20,7062
 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
@@ -71,11 +72,13 @@ synapse_sdk/plugins/cli/run.py,sha256=lw1KbsL-xTGllF4NtD2cq-Rh6HMbhi-sO862_Ds-sU
 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
 synapse_sdk/plugins/templates/hooks/pre_prompt.py,sha256=aOAMM623s0sKFGjTZaotAOYFvsNMxeii4tPyhOAFKVE,539
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.env,sha256=YlAG5YKwFEiwKm5NUbPsqvOJ9FJKoN0lbT3sxymBJpE,551
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.env.dist,sha256=YlAG5YKwFEiwKm5NUbPsqvOJ9FJKoN0lbT3sxymBJpE,551
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.gitignore,sha256=RvBmUbwmzgv0vYzYk2aQ0HpvKyPj6sCpBqsrthCuA4s,243
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/.pre-commit-config.yaml,sha256=p0yZwCUC8tYS1B0GPkjKiXYoRY9EZlq_ejFP98hB50g,154
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/README.md,sha256=ETBZv_2Ocgzn4Fe3o5Y842mZiz00ABuAalrXpNVnWU0,56
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/config.yaml,sha256=CRfKjPJ6K2UlYbcHMSAYUGzn2TGmV7kDbZmKHwNuq3w,210
+synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/main.py,sha256=fPFaIPBwskS84YvnmDGERXn3wC5z_yV_f4NbisVvBfQ,79
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/pyproject.toml,sha256=Usgd80tHZAD1Ug5MAjPfETUZxtKKgZW-xovFEAEbQDo,317
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/requirements.txt,sha256=F5UwinpTLQFfyakFGTFxgBOo4H-EKD9d4e77WKOPHhk,17
 synapse_sdk/plugins/templates/synapse-{{cookiecutter.plugin_code}}-plugin/plugin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -89,9 +92,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.0a8.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
-synapse_sdk-1.0.0a8.dist-info/METADATA,sha256=-qf5IZO1tU5DNWmiT4BMVD_26XoKjF3KaFsAfceIRx8,567
-synapse_sdk-1.0.0a8.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-synapse_sdk-1.0.0a8.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
-synapse_sdk-1.0.0a8.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
-synapse_sdk-1.0.0a8.dist-info/RECORD,,
+synapse_sdk-1.0.0a9.dist-info/LICENSE,sha256=bKzmC5YAg4V1Fhl8OO_tqY8j62hgdncAkN7VrdjmrGk,1101
+synapse_sdk-1.0.0a9.dist-info/METADATA,sha256=4pnZMrqZytZAvodVHryUt8WTbLnPwreD2iYPr7XTvvE,567
+synapse_sdk-1.0.0a9.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+synapse_sdk-1.0.0a9.dist-info/entry_points.txt,sha256=VNptJoGoNJI8yLXfBmhgUefMsmGI0m3-0YoMvrOgbxo,48
+synapse_sdk-1.0.0a9.dist-info/top_level.txt,sha256=ytgJMRK1slVOKUpgcw3LEyHHP7S34J6n_gJzdkcSsw8,12
+synapse_sdk-1.0.0a9.dist-info/RECORD,,