codewords-client 0.1.0__tar.gz

codewords_client-0.1.0/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.1
+Name: codewords_client
+Version: 0.1.0
+Author: Sam Coope
+Author-email: sam@agemo.ai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Requires-Dist: aiohttp
+Requires-Dist: requests
+Requires-Dist: websockets
+Requires-Dist: codewords_core

codewords_client-0.1.0/codewords_client/__init__.py

@@ -0,0 +1,117 @@
+
+import json
+import os
+import sys
+
+import requests
+import websockets
+
+# add ../ to sys.path
+from codewords_core.io_metadata import InputOutputMetadata
+from codewords_core.file_type_utils import download_files_and_update_variables, upload_files_and_update_variables
+
+try:
+    WS_URL = os.environ['CWR_WS_URL']
+    HTTPS_URL = os.environ['CWR_HTTPS_URL']
+except KeyError as e:
+    print(f"Environment variable {e} not set")
+    sys.exit(1)
+
+
+def cw_function(
+    function_id: str,
+    version: str,
+    auth_data: dict,
+    runner_id: str,
+    **init_kwargs
+) -> 'CWFunction':
+    resp = requests.post(
+        HTTPS_URL,
+        json={
+            "action": "get_info",
+            "function_id": function_id,
+            "version": version,
+            "auth_data": auth_data,
+            "runner_id": runner_id,
+        },
+    )
+    resp.raise_for_status()
+    definition_dict = resp.json()
+
+    inputs_metadata = [InputOutputMetadata.parse_obj(input_data) for input_data in definition_dict['inputs']]
+    outputs_metadata = [InputOutputMetadata.parse_obj(output_data) for output_data in definition_dict['outputs']]
+    return CWFunction(
+        definition_dict,
+        inputs_metadata,
+        outputs_metadata,
+        auth_data=auth_data,
+        runner_id=runner_id,
+        **init_kwargs
+    )
+
+
+class CWFunction():
+    def __init__(
+        self,
+        definition_dict,
+        inputs_metadata,
+        outputs_metadata,
+        *,
+        working_directory='',
+        auth_data: dict,
+        runner_id: str,
+        verbose=False
+    ):
+        self.definition_dict = definition_dict
+        self.inputs_metadata = inputs_metadata
+        self.outputs_metadata = outputs_metadata
+        self.working_directory = working_directory
+        self.auth_data = auth_data
+        self.runner_id = runner_id
+        self._print = print if verbose else lambda *args, **kwargs: None
+
+    async def run(self, inputs: Dict[str, Any]):
+        # find all the file type fields and upload them
+        inputs = await upload_files_and_update_variables(
+            variables=inputs,
+            variables_metadata=self.inputs_metadata,
+            working_directory=self.working_directory,
+        )
+
+        # call the lambda at WS_URL with an event
+        event = {
+            'action': 'run',
+            'function_id': self.definition_dict['function_id'],
+            'version': self.definition_dict['version'],
+            'inputs': inputs,
+            'runner_id': self.runner_id,
+            'auth_data': self.auth_data,
+        }
+        async with websockets.connect(WS_URL) as ws:
+            await ws.send(json.dumps(event))
+
+            # TODO - handle intermediate messages
+            response = await ws.recv()
+            response = json.loads(response)
+
+        if not response['type'] == 'run_complete':
+            raise ValueError(f"run didnt complete successfully: {response}")
+
+        outputs = response['outputs']
+        # download any files that were uploaded
+        outputs = await download_files_and_update_variables(
+            variables=outputs,
+            variables_metadata=self.outputs_metadata,
+            working_directory=self.working_directory,
+        )
+
+        return outputs, response['run_info']
+
+    async def __call__(self, **inputs):
+        outputs, run_info = await self.run(inputs)
+        # if there's an error throw that
+        if len(run_info['error_messages']) > 0:
+            raise Exception(f"Run failed with error messages: {run_info['error_messages']}")
+
+        return outputs
+

codewords_client-0.1.0/codewords_client.egg-info/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.1
+Name: codewords_client
+Version: 0.1.0
+Author: Sam Coope
+Author-email: sam@agemo.ai
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Requires-Dist: aiohttp
+Requires-Dist: requests
+Requires-Dist: websockets
+Requires-Dist: codewords_core

codewords_client-0.1.0/codewords_client.egg-info/SOURCES.txt

@@ -0,0 +1,7 @@
+setup.py
+codewords_client/__init__.py
+codewords_client.egg-info/PKG-INFO
+codewords_client.egg-info/SOURCES.txt
+codewords_client.egg-info/dependency_links.txt
+codewords_client.egg-info/requires.txt
+codewords_client.egg-info/top_level.txt

codewords_client-0.1.0/codewords_client.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

codewords_client-0.1.0/codewords_client.egg-info/requires.txt

@@ -0,0 +1,4 @@
+aiohttp
+requests
+websockets
+codewords_core

codewords_client-0.1.0/codewords_client.egg-info/top_level.txt

@@ -0,0 +1 @@
+codewords_client

codewords_client-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

codewords_client-0.1.0/setup.py

@@ -0,0 +1,24 @@
+from setuptools import setup, find_packages
+
+with open('requirements.txt') as f:
+    required = f.read().splitlines()
+
+setup(
+    name='codewords_client',
+    version='0.1.0',
+    author='Sam Coope',
+    author_email='sam@agemo.ai',
+    packages=find_packages(),
+    install_requires=required,
+    classifiers=[
+        'Development Status :: 3 - Alpha',
+        'Intended Audience :: Developers',
+        'License :: OSI Approved :: MIT License',  # Replace with your license
+        'Programming Language :: Python :: 3',
+        'Programming Language :: Python :: 3.9',
+        'Programming Language :: Python :: 3.10',
+        'Programming Language :: Python :: 3.11',
+        'Programming Language :: Python :: 3.12',
+    ],
+    python_requires='>=3.10'
+)