mech-client 0.1.5__tar.gz → 0.1.7__tar.gz

{mech_client-0.1.5 → mech_client-0.1.7}/PKG-INFO

@@ -1,10 +1,12 @@
 Metadata-Version: 2.1
 Name: mech-client
-Version: 0.1.5
+Version: 0.1.7
 Summary: Basic client to interact with a mech
+License: Apache-2.0
 Author: David Minarsch
 Author-email: david.minarsch@googlemail.com
 Requires-Python: >=3.10,<4.0
+Classifier: License :: OSI Approved :: Apache Software License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11

mech_client-0.1.7/mech_client/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.7"

{mech_client-0.1.5 → mech_client-0.1.7}/mech_client/cli.py

@@ -4,6 +4,7 @@ from mech_client import __version__
 from mech_client.interact import interact as interact_
 from mech_client.prompt_to_ipfs import main as prompt_to_ipfs_main
 from mech_client.push_to_ipfs import main as push_to_ipfs_main
+from mech_client.to_png import main as to_png_main
 
 
 @click.group(name="mechx")  # type: ignore
@@ -35,9 +36,19 @@ def push_to_ipfs(file_path: str) -> None:
     push_to_ipfs_main(file_path=file_path)
 
 
+@click.command()
+@click.argument("ipfs_hash")
+@click.argument("path")
+@click.argument("request_id")
+def to_png(ipfs_hash: str, path: str, request_id: str) -> None:
+    """Convert a stability AI API's diffusion model output into a PNG format."""
+    to_png_main(ipfs_hash, path, request_id)
+
+
 cli.add_command(interact)
 cli.add_command(prompt_to_ipfs)
 cli.add_command(push_to_ipfs)
+cli.add_command(to_png)
 
 
 if __name__ == "__main__":

{mech_client-0.1.5 → mech_client-0.1.7}/mech_client/interact.py

@@ -72,6 +72,10 @@ def check_for_tools(tool: str) -> Optional[int]:
             "openai-text-davinci-003",
             "openai-gpt-3.5-turbo",
             "openai-gpt-4",
+            "stabilityai-stable-diffusion-v1-5",
+            "stabilityai-stable-diffusion-xl-beta-v2-2-2",
+            "stabilityai-stable-diffusion-512-v2-1",
+            "stabilityai-stable-diffusion-768-v2-1",
         ]
         else None
     )

mech_client-0.1.7/mech_client/to_png.py

@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# ------------------------------------------------------------------------------
+#
+#   Copyright 2023 Valory AG
+#
+#   Licensed under the Apache License, Version 2.0 (the "License");
+#   you may not use this file except in compliance with the License.
+#   You may obtain a copy of the License at
+#
+#       http://www.apache.org/licenses/LICENSE-2.0
+#
+#   Unless required by applicable law or agreed to in writing, software
+#   distributed under the License is distributed on an "AS IS" BASIS,
+#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#   See the License for the specific language governing permissions and
+#   limitations under the License.
+#
+# ------------------------------------------------------------------------------
+
+"""
+This script facilitates the conversion of a stability AI API's diffusion model output into a PNG format.
+
+Usage:
+
+python push_to_ipfs.py <ipfs_hash> <path>
+"""
+import json
+import os.path
+from base64 import b64decode
+from tempfile import gettempdir
+
+from aea_cli_ipfs.ipfs_utils import IPFSTool
+
+
+def to_png(data: dict, path: str) -> None:
+    """Stores a stability AI API's diffusion model output into a PNG formatted file."""
+    for i, image in enumerate(data["artifacts"]):
+        with open(path, "wb") as f:
+            f.write(b64decode(image["base64"]))
+
+    print(f"Successfully created {path}.")
+
+
+def get_from_ipfs(ipfs_hash: str, request_id: str) -> dict:
+    """Get data from IPFS."""
+    temp_dir = gettempdir()
+    IPFSTool().client.get(cid=ipfs_hash, target=temp_dir)
+    stored_data = os.path.join(temp_dir, ipfs_hash)
+
+    with open(os.path.join(stored_data, request_id)) as f:
+        data = json.loads(f.read()).get("result", {})
+
+    if not isinstance(data, dict):
+        raise ValueError("Data do not have the expected format!")
+
+    return data
+
+
+def main(ipfs_hash: str, path: str, request_id: str) -> None:
+    data_ = get_from_ipfs(ipfs_hash, request_id)
+    to_png(data_, path)

{mech_client-0.1.5 → mech_client-0.1.7}/pyproject.toml

@@ -1,10 +1,11 @@
 [tool.poetry]
 name = "mech-client"
-version = "0.1.5"
+version = "0.1.7"
 description = "Basic client to interact with a mech"
 authors = ["David Minarsch <david.minarsch@googlemail.com>"]
 readme = "README.md"
 packages = [{include = "mech_client"}]
+license = "Apache-2.0"
 
 [tool.poetry.dependencies]
 python = "^3.10"

{mech_client-0.1.5 → mech_client-0.1.7}/setup.py

@@ -18,7 +18,7 @@ entry_points = \
 
 setup_kwargs = {
     'name': 'mech-client',
-    'version': '0.1.5',
+    'version': '0.1.7',
     'description': 'Basic client to interact with a mech',
     'long_description': '# mech-client\nBasic client to interact with a mech\n\n> **Warning**<br />\n> **This is a hacky alpha version of the client - don\'t rely on it as production software.**\n\n## Installation\n\n```bash\npip install mech-client\n```\n\nThen, set a websocket endpoint for Gnosis RPC like so:\n\n```bash\nexport WEBSOCKET_ENDPOINT=<YOUR ENDPOINT>\n```\n\n## CLI:\n\n```bash\nUsage: mechx [OPTIONS] COMMAND [ARGS]...\n\n  Command-line tool for interacting with mechs.\n\nOptions:\n  --version  Show the version and exit.\n  --help     Show this message and exit.\n\nCommands:\n  interact        Interact with a mech specifying a prompt and tool.\n  prompt-to-ipfs  Upload a prompt and tool to IPFS as metadata.\n  push-to-ipfs    Upload a file to IPFS.\n ```\n\n## Usage:\n\nFirst, create a private key in file `ethereum_private_key.txt` with this command:\n\n```bash\naea generate-key ethereum\n```\n\nEnsure the private key carries funds on Gnosis Chain.\n\nSecond, run the following command to instruct the mech with `<prompt>` and `<tool>`:\n\n```bash\nmechx interact <prompt> <tool>\n```\n\nExample output:\n```bash\nmechx interact "write a short poem" "openai-text-davinci-003"\nPrompt uploaded: https://gateway.autonolas.tech/ipfs/f01701220ad9e2d5698fbd6c3a4ce61f329590e68a23181772669e543e69decdae316423b\nTransaction sent: https://gnosisscan.io/tx/0xb3a17ef90da6cc7a86e008a3a91bd367d573b406eae53405a4aa981001a5eaf3\nRequest on-chain with id: 15263135923206312300456917202469137903009897852865973093832667165921851537677\nData arrived: https://gateway.autonolas.tech/ipfs/f017012205053a4ae3ef0cf4ed7eff0c2d74dbaf3479fbdeb292472560e7bfaa4cfecfcdc\nData: {\'requestId\': 15263135923206312300456917202469137903009897852865973093832667165921851537677, \'result\': "\\n\\nA sun-filled sky,\\nA soft breeze blowing by,\\nWhere the trees sway in the wind,\\nA peaceful moment I can\'t rewind."}\n```\n\n## Release guide:\n\nFinish edits, bump versions in `pyproject.toml` and `mech_client/__init__.py`, then `poetry lock`, then `rm -rf dist`, then `poetry publish --build --username=<username> --password=<password>`.',
     'author': 'David Minarsch',

mech_client-0.1.5/mech_client/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.1.5"