mech-client 0.7.0__tar.gz → 0.8.0__tar.gz

{mech_client-0.7.0 → mech_client-0.8.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mech-client
-Version: 0.7.0
+Version: 0.8.0
 Summary: Basic client to interact with a mech
 License: Apache-2.0
 Author: David Minarsch
@@ -251,8 +251,9 @@ mechx interact --prompts <prompt-1> --prompts <prompt-2> --priority-mech <priori
 ```
 
 
-### List tools available for agents
+### List tools available for legacy mechs and marketplace mechs
 
+#### For legacy mechs
 To list the tools available for a specific agent or for all agents, use the `tools-for-agents` command. You can specify an agent ID to get tools for a specific agent, or omit it to list tools for all agents.
 
 ```bash
@@ -293,8 +294,28 @@ You will see an output like this:
 +---------------------------------------------+-----------------------------------------------+
 ```
 
+#### For marketplace mechs
+To list the tools available for a specific marketplace mech, use the `tools-for-marketplace-mech` command. You can specify a service ID to get tools for a specific mech.
+
+```bash
+mechx tools-for-marketplace-mech 1722 --chain-config gnosis
+```
+```bash
+You will see an output like this:
++---------------------------------------------+-----------------------------------------------+
+| Tool Name                                   | Unique Identifier                             |
++=============================================+===============================================+
+| claude-prediction-offline                   | 1722-claude-prediction-offline                |
++---------------------------------------------+-----------------------------------------------+
+| claude-prediction-online                    | 1722-claude-prediction-online                 |
++---------------------------------------------+-----------------------------------------------+
+| deepmind-optimization                       | 1722-deepmind-optimization                    |
++---------------------------------------------+-----------------------------------------------+
+```
+
 ### Get Tool Description
 
+#### For legacy mechs
 To get the description of a specific tool, use the `tool-description` command. You need to specify the unique identifier of the tool.
 
 ```bash
@@ -310,9 +331,26 @@ You will see an output like this:
 Description for tool 6-claude-prediction-offline: Makes a prediction using Claude
 ```
 
+#### For marketplace mechs
+To get the description of a specific tool, use the ` tool-description-for-marketplace-mech` command. You need to specify the unique identifier of the tool.
+
+```bash
+mechx  tool-description-for-marketplace-mech <unique_identifier> --chain-config <chain_config>
+```
+Example usage:
+
+```bash
+mechx  tool-description-for-marketplace-mech 1722-openai-gpt-4 --chain-config gnosis
+```
+You will see an output like this:
+```bash
+Description for tool 1722-openai-gpt-4: Performs a request to OpenAI's GPT-4 model.
+```
+
 
 ### Get Tool Input/Output Schema
 
+#### For legacy mechs
 To get the input/output schema of a specific tool, use the `tool_io_schema` command. You need to specify the unique identifier of the tool.
 
 ```bash
@@ -352,6 +390,47 @@ Output Schema:
 +-----------+---------+-----------------------------------------------+
 ```
 
+#### For marketplace mechs
+To get the input/output schema of a specific tool, use the `tool-io-schema-for-marketplace-mech` command. You need to specify the unique identifier of the tool.
+
+```bash
+mechx tool-io-schema-for-marketplace-mech <unique_identifier> --chain-config <chain_config>
+```
+
+Example usage:
+
+```bash
+mechx tool-io-schema-for-marketplace-mech 1722-openai-gpt-4 --chain-config gnosis
+```
+You will see an output like this:
+```bash
+Tool Details:
+Tool Details:
++------------------------+---------------------------------------------+
+| Tool Name              | Tool Description                            |
++========================+=============================================+
+| OpenAI Request (GPT-4) | Performs a request to OpenAI's GPT-4 model. |
++------------------------+---------------------------------------------+
+Input Schema:
++-------------+-----------------------------------------------+
+| Field       | Value                                         |
++=============+===============================================+
+| type        | text                                          |
++-------------+-----------------------------------------------+
+| description | The request to relay to OpenAI's GPT-4 model. |
++-------------+-----------------------------------------------+
+Output Schema:
++-----------+---------+-----------------------------------+
+| Field     | Type    | Description                       |
++===========+=========+===================================+
+| requestId | integer | Unique identifier for the request |
++-----------+---------+-----------------------------------+
+| result    | string  | Response from OpenAI              |
++-----------+---------+-----------------------------------+
+| prompt    | string  | User prompt to send to OpenAI     |
++-----------+---------+-----------------------------------+
+```
+
 > **:pencil2: Note** <br />
 > **If you encounter an "Out of gas" error when executing the Mech Client, you will need to increase the gas limit, e.g.,**
 >

{mech_client-0.7.0 → mech_client-0.8.0}/README.md

@@ -228,8 +228,9 @@ mechx interact --prompts <prompt-1> --prompts <prompt-2> --priority-mech <priori
 ```
 
 
-### List tools available for agents
+### List tools available for legacy mechs and marketplace mechs
 
+#### For legacy mechs
 To list the tools available for a specific agent or for all agents, use the `tools-for-agents` command. You can specify an agent ID to get tools for a specific agent, or omit it to list tools for all agents.
 
 ```bash
@@ -270,8 +271,28 @@ You will see an output like this:
 +---------------------------------------------+-----------------------------------------------+
 ```
 
+#### For marketplace mechs
+To list the tools available for a specific marketplace mech, use the `tools-for-marketplace-mech` command. You can specify a service ID to get tools for a specific mech.
+
+```bash
+mechx tools-for-marketplace-mech 1722 --chain-config gnosis
+```
+```bash
+You will see an output like this:
++---------------------------------------------+-----------------------------------------------+
+| Tool Name                                   | Unique Identifier                             |
++=============================================+===============================================+
+| claude-prediction-offline                   | 1722-claude-prediction-offline                |
++---------------------------------------------+-----------------------------------------------+
+| claude-prediction-online                    | 1722-claude-prediction-online                 |
++---------------------------------------------+-----------------------------------------------+
+| deepmind-optimization                       | 1722-deepmind-optimization                    |
++---------------------------------------------+-----------------------------------------------+
+```
+
 ### Get Tool Description
 
+#### For legacy mechs
 To get the description of a specific tool, use the `tool-description` command. You need to specify the unique identifier of the tool.
 
 ```bash
@@ -287,9 +308,26 @@ You will see an output like this:
 Description for tool 6-claude-prediction-offline: Makes a prediction using Claude
 ```
 
+#### For marketplace mechs
+To get the description of a specific tool, use the ` tool-description-for-marketplace-mech` command. You need to specify the unique identifier of the tool.
+
+```bash
+mechx  tool-description-for-marketplace-mech <unique_identifier> --chain-config <chain_config>
+```
+Example usage:
+
+```bash
+mechx  tool-description-for-marketplace-mech 1722-openai-gpt-4 --chain-config gnosis
+```
+You will see an output like this:
+```bash
+Description for tool 1722-openai-gpt-4: Performs a request to OpenAI's GPT-4 model.
+```
+
 
 ### Get Tool Input/Output Schema
 
+#### For legacy mechs
 To get the input/output schema of a specific tool, use the `tool_io_schema` command. You need to specify the unique identifier of the tool.
 
 ```bash
@@ -329,6 +367,47 @@ Output Schema:
 +-----------+---------+-----------------------------------------------+
 ```
 
+#### For marketplace mechs
+To get the input/output schema of a specific tool, use the `tool-io-schema-for-marketplace-mech` command. You need to specify the unique identifier of the tool.
+
+```bash
+mechx tool-io-schema-for-marketplace-mech <unique_identifier> --chain-config <chain_config>
+```
+
+Example usage:
+
+```bash
+mechx tool-io-schema-for-marketplace-mech 1722-openai-gpt-4 --chain-config gnosis
+```
+You will see an output like this:
+```bash
+Tool Details:
+Tool Details:
++------------------------+---------------------------------------------+
+| Tool Name              | Tool Description                            |
++========================+=============================================+
+| OpenAI Request (GPT-4) | Performs a request to OpenAI's GPT-4 model. |
++------------------------+---------------------------------------------+
+Input Schema:
++-------------+-----------------------------------------------+
+| Field       | Value                                         |
++=============+===============================================+
+| type        | text                                          |
++-------------+-----------------------------------------------+
+| description | The request to relay to OpenAI's GPT-4 model. |
++-------------+-----------------------------------------------+
+Output Schema:
++-----------+---------+-----------------------------------+
+| Field     | Type    | Description                       |
++===========+=========+===================================+
+| requestId | integer | Unique identifier for the request |
++-----------+---------+-----------------------------------+
+| result    | string  | Response from OpenAI              |
++-----------+---------+-----------------------------------+
+| prompt    | string  | User prompt to send to OpenAI     |
++-----------+---------+-----------------------------------+
+```
+
 > **:pencil2: Note** <br />
 > **If you encounter an "Out of gas" error when executing the Mech Client, you will need to increase the gas limit, e.g.,**
 >

mech_client-0.8.0/mech_client/__init__.py

@@ -0,0 +1,3 @@
+"""Mech client."""
+
+__version__ = "0.8.0"

mech_client-0.8.0/mech_client/abis/ComplementaryServiceMetadata.json

@@ -0,0 +1,185 @@
+[
+    {
+        "inputs": [
+            {
+                "internalType": "address",
+                "name": "_serviceRegistry",
+                "type": "address"
+            }
+        ],
+        "stateMutability": "nonpayable",
+        "type": "constructor"
+    },
+    {
+        "inputs": [],
+        "name": "ReentrancyGuard",
+        "type": "error"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "address",
+                "name": "account",
+                "type": "address"
+            }
+        ],
+        "name": "UnauthorizedAccount",
+        "type": "error"
+    },
+    {
+        "inputs": [],
+        "name": "ZeroAddress",
+        "type": "error"
+    },
+    {
+        "anonymous": false,
+        "inputs": [
+            {
+                "indexed": true,
+                "internalType": "uint256",
+                "name": "serviceId",
+                "type": "uint256"
+            },
+            {
+                "indexed": true,
+                "internalType": "bytes32",
+                "name": "hash",
+                "type": "bytes32"
+            }
+        ],
+        "name": "ComplementaryMetadataUpdated",
+        "type": "event"
+    },
+    {
+        "inputs": [],
+        "name": "CID_PREFIX",
+        "outputs": [
+            {
+                "internalType": "string",
+                "name": "",
+                "type": "string"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [],
+        "name": "VERSION",
+        "outputs": [
+            {
+                "internalType": "string",
+                "name": "",
+                "type": "string"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [],
+        "name": "baseURI",
+        "outputs": [
+            {
+                "internalType": "string",
+                "name": "",
+                "type": "string"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "uint256",
+                "name": "serviceId",
+                "type": "uint256"
+            },
+            {
+                "internalType": "bytes32",
+                "name": "hash",
+                "type": "bytes32"
+            }
+        ],
+        "name": "changeHash",
+        "outputs": [],
+        "stateMutability": "nonpayable",
+        "type": "function"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "address",
+                "name": "account",
+                "type": "address"
+            },
+            {
+                "internalType": "uint256",
+                "name": "serviceId",
+                "type": "uint256"
+            }
+        ],
+        "name": "isAbleChangeHash",
+        "outputs": [
+            {
+                "internalType": "bool",
+                "name": "",
+                "type": "bool"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "uint256",
+                "name": "",
+                "type": "uint256"
+            }
+        ],
+        "name": "mapServiceHashes",
+        "outputs": [
+            {
+                "internalType": "bytes32",
+                "name": "",
+                "type": "bytes32"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [],
+        "name": "serviceRegistry",
+        "outputs": [
+            {
+                "internalType": "address",
+                "name": "",
+                "type": "address"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    },
+    {
+        "inputs": [
+            {
+                "internalType": "uint256",
+                "name": "serviceId",
+                "type": "uint256"
+            }
+        ],
+        "name": "tokenURI",
+        "outputs": [
+            {
+                "internalType": "string",
+                "name": "",
+                "type": "string"
+            }
+        ],
+        "stateMutability": "view",
+        "type": "function"
+    }
+]

{mech_client-0.7.0 → mech_client-0.8.0}/mech_client/cli.py

@@ -31,6 +31,17 @@ from mech_client.interact import interact as interact_
 from mech_client.marketplace_interact import (
     marketplace_interact as marketplace_interact_,
 )
+from mech_client.mech_marketplace_tool_management import (
+    extract_input_schema,
+    extract_output_schema,
+)
+from mech_client.mech_marketplace_tool_management import (
+    get_tool_description as get_tool_description_for_marketplace_mech,
+)
+from mech_client.mech_marketplace_tool_management import (
+    get_tool_io_schema as get_tool_io_schema_for_marketplace_mech,
+)
+from mech_client.mech_marketplace_tool_management import get_tools_for_marketplace_mech
 from mech_client.mech_tool_management import (
     get_tool_description,
     get_tool_io_schema,
@@ -353,6 +364,84 @@ def tool_io_schema(tool_id: str, chain_config: str) -> None:
         click.echo(f"Network or I/O error: {str(e)}")
 
 
+@click.command(name="tools-for-marketplace-mech")
+@click.argument(
+    "service-id",
+    type=int,
+)
+@click.option("--chain-config", default="gnosis", help="Chain configuration to use.")
+def tools_for_marketplace_mech(service_id: int, chain_config: str) -> None:
+    """Fetch and display tools for marketplace mechs."""
+    try:
+        result = get_tools_for_marketplace_mech(service_id, chain_config)
+
+        headers = ["Tool Name", "Unique Identifier"]
+        data: List[Tuple[str, ...]] = [
+            (
+                str(tool.tool_name),
+                str(tool.unique_identifier),
+            )
+            for tool in result.tools
+        ]
+
+        click.echo(tabulate(data, headers=headers, tablefmt="grid"))
+
+    except (KeyError, TypeError) as e:
+        click.echo(f"Error processing tool data: {str(e)}")
+    except IOError as e:
+        click.echo(f"Network or I/O error: {str(e)}")
+
+
+@click.command(name="tool-description-for-marketplace-mech")
+@click.argument("tool_id")
+@click.option("--chain-config", default="gnosis", help="Chain configuration to use.")
+def tool_description_for_marketplace_mech(tool_id: str, chain_config: str) -> None:
+    """Fetch and display the description of a specific tool for marketplace mechs."""
+    try:
+        description = get_tool_description_for_marketplace_mech(tool_id, chain_config)
+        click.echo(f"Description for tool {tool_id}: {description}")
+    except KeyError as e:
+        click.echo(f"Tool not found or missing description: {str(e)}")
+    except IOError as e:
+        click.echo(f"Network or I/O error: {str(e)}")
+
+
+@click.command(name="tool-io-schema-for-marketplace-mech")
+@click.argument("tool_id")
+@click.option("--chain-config", default="gnosis", help="Chain configuration to use.")
+def tool_io_schema_for_marketplace_mech(tool_id: str, chain_config: str) -> None:
+    """Fetch and display the tool's name and description along with the input/output schema for a specific tool for marketplace mechs."""
+    try:
+        result = get_tool_io_schema_for_marketplace_mech(tool_id, chain_config)
+
+        name = result["name"]
+        description = result["description"]
+        input_schema = extract_input_schema(result["input"])
+        output_schema = extract_output_schema(result["output"])
+
+        click.echo("Tool Details:")
+        click.echo(
+            tabulate(
+                [[name, description]],
+                headers=["Tool Name", "Tool Description"],
+                tablefmt="grid",
+            )
+        )
+
+        click.echo("Input Schema:")
+        click.echo(tabulate(input_schema, headers=["Field", "Value"], tablefmt="grid"))
+        click.echo("Output Schema:")
+        click.echo(
+            tabulate(
+                output_schema, headers=["Field", "Type", "Description"], tablefmt="grid"
+            )
+        )
+    except KeyError as e:
+        click.echo(f"Error accessing schema data: {str(e)}")
+    except IOError as e:
+        click.echo(f"Network or I/O error: {str(e)}")
+
+
 @click.command(name="deposit-native")
 @click.argument("amount_to_deposit")
 @click.option(
@@ -423,8 +512,11 @@ cli.add_command(prompt_to_ipfs)
 cli.add_command(push_to_ipfs)
 cli.add_command(to_png)
 cli.add_command(tools_for_agents)
+cli.add_command(tools_for_marketplace_mech)
 cli.add_command(tool_io_schema)
+cli.add_command(tool_io_schema_for_marketplace_mech)
 cli.add_command(tool_description)
+cli.add_command(tool_description_for_marketplace_mech)
 cli.add_command(deposit_native)
 cli.add_command(deposit_token)
 cli.add_command(nvm_subscribe)

{mech_client-0.7.0 → mech_client-0.8.0}/mech_client/configs/mechs.json

@@ -3,6 +3,7 @@
         "agent_registry_contract": "0xE49CB081e8d96920C38aA7AB90cb0294ab4Bc8EA",
         "service_registry_contract": "0x9338b5153AE39BB89f50468E608eD9d764B755fD",
         "mech_marketplace_contract": "0x735FAAb1c4Ec41128c367AFb5c3baC73509f70bB",
+        "complementary_metadata_hash_address": "0x0598081D48FB80B0A7E52FAD2905AE9beCd6fC69",
         "rpc_url": "https://rpc.eu-central-2.gateway.fm/v4/gnosis/non-archival/mainnet",
         "wss_endpoint": "wss://rpc.gnosischain.com/wss",
         "ledger_config": {
@@ -21,6 +22,7 @@
         "agent_registry_contract": "0xa4799B083E0068732456EF45ff9fe5c683658327",
         "service_registry_contract": "0xE3607b00E75f6405248323A9417ff6b39B244b50",
         "mech_marketplace_contract": "0x0000000000000000000000000000000000000000",
+        "complementary_metadata_hash_address": "0x0000000000000000000000000000000000000000",
         "rpc_url": "https://arbitrum.llamarpc.com",
         "wss_endpoint": "wss://arbitrum.gateway.tenderly.co",
         "ledger_config": {
@@ -39,6 +41,7 @@
         "agent_registry_contract": "0x984cf72FDe8B5aA910e9e508aC5e007ae5BDcC9C",
         "service_registry_contract": "0xE3607b00E75f6405248323A9417ff6b39B244b50",
         "mech_marketplace_contract": "0x0000000000000000000000000000000000000000",
+        "complementary_metadata_hash_address": "0x0000000000000000000000000000000000000000",
         "rpc_url": "https://polygon-bor-rpc.publicnode.com",
         "wss_endpoint": "wss://polygon.gateway.tenderly.co",
         "ledger_config": {
@@ -57,6 +60,7 @@
         "agent_registry_contract": "0x88DE734655184a09B70700aE4F72364d1ad23728",
         "service_registry_contract": "0x3C1fF68f5aa342D296d4DEe4Bb1cACCA912D95fE",
         "mech_marketplace_contract": "0xf24eE42edA0fc9b33B7D41B06Ee8ccD2Ef7C5020",
+        "complementary_metadata_hash_address": "0x28C1edC7CEd549F7f80B732fDC19f0370160707d",
         "rpc_url": "https://base.llamarpc.com",
         "wss_endpoint": "wss://base.gateway.tenderly.co",
         "ledger_config": {
@@ -75,6 +79,7 @@
         "agent_registry_contract": "0xE49CB081e8d96920C38aA7AB90cb0294ab4Bc8EA",
         "service_registry_contract": "0xE3607b00E75f6405248323A9417ff6b39B244b50",
         "mech_marketplace_contract": "0x0000000000000000000000000000000000000000",
+        "complementary_metadata_hash_address": "0x0000000000000000000000000000000000000000",
         "rpc_url": "https://forno.celo.org",
         "wss_endpoint": "wss://forno.celo.org/ws",
         "ledger_config": {
@@ -93,6 +98,7 @@
         "agent_registry_contract": "0x75D529FAe220bC8db714F0202193726b46881B76",
         "service_registry_contract": "0x3d77596beb0f130a4415df3D2D8232B3d3D31e44",
         "mech_marketplace_contract": "0x0000000000000000000000000000000000000000",
+        "complementary_metadata_hash_address": "0x0000000000000000000000000000000000000000",
         "rpc_url": "https://mainnet.optimism.io",
         "wss_endpoint": "wss://optimism.gateway.tenderly.co",
         "ledger_config": {

{mech_client-0.7.0 → mech_client-0.8.0}/mech_client/interact.py

@@ -120,6 +120,7 @@ class MechConfig:  # pylint: disable=too-many-instance-attributes
 
     agent_registry_contract: str
     service_registry_contract: str
+    complementary_metadata_hash_address: str
     rpc_url: str
     wss_endpoint: str
     ledger_config: LedgerConfig

mech_client-0.8.0/mech_client/mech_marketplace_tool_management.py

@@ -0,0 +1,223 @@
+"""Module for managing mechanical tools and their interactions with blockchain."""
+
+import json
+from dataclasses import asdict, dataclass
+from pathlib import Path
+from typing import Any, Dict, List, Optional, Tuple
+
+import requests
+from aea_ledger_ethereum import EthereumApi
+
+from mech_client.marketplace_interact import ADDRESS_ZERO, get_contract, get_mech_config
+
+
+ABI_DIR_PATH = Path(__file__).parent / "abis"
+COMPLEMENTARY_METADATA_HASH_ABI_PATH = (
+    ABI_DIR_PATH / "ComplementaryServiceMetadata.json"
+)
+ENCODING = "utf-8"
+DEFAULT_TIMEOUT = 10
+TOOLS = "tools"
+TOOL_METADATA = "toolMetadata"
+DEFAULT_CONFIG = "gnosis"
+
+
+@dataclass
+class ToolInfo:
+    """Tool info"""
+
+    tool_name: str
+    unique_identifier: str
+
+
+@dataclass
+class ToolsForMarketplaceMech:
+    """Tools info list"""
+
+    service_id: int
+    tools: List[ToolInfo]
+
+
+def fetch_tools(
+    service_id: int,
+    ledger_api: EthereumApi,
+    complementary_metadata_hash_address: str,
+    contract_abi_path: Path,
+) -> Dict[str, Any]:
+    """Fetch tools for specified mech's service ID."""
+    with open(contract_abi_path, encoding=ENCODING) as f:
+        abi = json.load(f)
+
+    metadata_contract = get_contract(
+        contract_address=complementary_metadata_hash_address,
+        abi=abi,
+        ledger_api=ledger_api,
+    )
+    metadata_uri = metadata_contract.functions.tokenURI(service_id).call()
+    return requests.get(metadata_uri, timeout=DEFAULT_TIMEOUT).json()
+
+
+def get_mech_tools(
+    service_id: int, chain_config: str = DEFAULT_CONFIG
+) -> Optional[Dict[str, Any]]:
+    """
+    Fetch tools for a given mech's service ID.
+
+    :param service_id: The service ID of the mech.
+    :param chain_config: The chain configuration to use (default is "gnosis").
+    :return: A dictionary containing the JSON response from the `tokenURI` contract call, typically including tools and metadata.
+    """
+    # Get the mech configuration
+    mech_config = get_mech_config(chain_config)
+    ledger_config = mech_config.ledger_config
+
+    # Setup Ethereum API
+    ledger_api = EthereumApi(**asdict(ledger_config))
+
+    if mech_config.complementary_metadata_hash_address == ADDRESS_ZERO:
+        print(f"Metadata hash not yet implemented on {chain_config}")
+        return None
+
+    try:
+        # Fetch tools for the given mech's service ID
+        return fetch_tools(
+            service_id=service_id,
+            ledger_api=ledger_api,
+            complementary_metadata_hash_address=mech_config.complementary_metadata_hash_address,
+            contract_abi_path=COMPLEMENTARY_METADATA_HASH_ABI_PATH,
+        )
+    except (json.JSONDecodeError, NotImplementedError) as e:
+        print(f"An error occurred while fetching tools for mech with {service_id}: {e}")
+        return None
+
+
+def get_tools_for_marketplace_mech(
+    service_id: int, chain_config: str = DEFAULT_CONFIG
+) -> ToolsForMarketplaceMech:
+    """
+    Retrieve tools for specified mech's service id.
+
+    :param service_id: specific mech's service ID to fetch tools for.
+    :param chain_config: The chain configuration to use.
+    :return: Dictionary of tools with identifiers or a mapping of service IDs to tools.
+    """
+    empty_response = ToolsForMarketplaceMech(service_id=service_id, tools=[])
+
+    try:
+        result = get_mech_tools(service_id, chain_config)
+        if result is None:
+            return empty_response
+
+        tools = result.get(TOOLS, [])
+        tool_metadata = result.get(TOOL_METADATA, {})
+        if not isinstance(tools, list) or not isinstance(tool_metadata, dict):
+            return empty_response
+
+        tools_with_ids = [
+            ToolInfo(tool_name=tool, unique_identifier=f"{service_id}-{tool}")
+            for tool in tools
+        ]
+        return ToolsForMarketplaceMech(service_id=service_id, tools=tools_with_ids)
+
+    except (json.JSONDecodeError, NotImplementedError) as e:
+        print(f"Error in get_tools_for_marketplace_mech: {str(e)}")
+        raise
+
+
+def get_tool_description(
+    unique_identifier: str, chain_config: str = DEFAULT_CONFIG
+) -> str:
+    """
+    Fetch the description of a specific tool based on a unique identifier.
+
+    :param unique_identifier: The unique identifier for the tool.
+    :param chain_config: The chain configuration to use.
+    :return: Description of the tool or a default message if not available.
+    """
+    default_response = "Description not available"
+
+    _, tool_info = _get_tool_metadata(unique_identifier, chain_config)
+    return (
+        tool_info.get("description", default_response)
+        if tool_info
+        else default_response
+    )
+
+
+def get_tool_io_schema(
+    unique_identifier: str, chain_config: str = DEFAULT_CONFIG
+) -> Dict[str, Any]:
+    """
+    Fetch the input and output schema along with tool name and description of a specific tool based on a unique identifier.
+
+    :param unique_identifier: The unique identifier for the tool.
+    :param chain_config: The chain configuration to use.
+    :return: Dictionary containing name, description, input and output schemas.
+    """
+    _, tool_info = _get_tool_metadata(unique_identifier, chain_config)
+    if tool_info:
+        return {
+            "name": tool_info.get("name", {}),
+            "description": tool_info.get("description", {}),
+            "input": tool_info.get("input", {}),
+            "output": tool_info.get("output", {}),
+        }
+
+    return {"input": {}, "output": {}}
+
+
+def _get_tool_metadata(
+    unique_identifier: str, chain_config: str = DEFAULT_CONFIG
+) -> Tuple[str, Optional[Dict[str, Any]]]:
+    """
+    Helper function to extract tool metadata from the chain config and unique identifier.
+
+    :param unique_identifier: The unique identifier in the format "<service_id>-<tool_name>".
+    :param chain_config: The chain configuration to use.
+    :return: Tuple of tool name and its metadata dictionary (or None if not found).
+    """
+    service_id_str, *tool_parts = unique_identifier.split("-")
+    try:
+        service_id = int(service_id_str)
+    except ValueError as exc:
+        raise ValueError(
+            f"Unexpected unique identifier format: {unique_identifier}"
+        ) from exc
+    tool_name = "-".join(tool_parts)
+
+    result = get_mech_tools(service_id, chain_config)
+
+    if isinstance(result, dict):
+        tool_metadata = result.get(TOOL_METADATA, {})
+        tool_info = tool_metadata.get(tool_name)
+        if isinstance(tool_info, dict):
+            return tool_name, tool_info
+
+    return tool_name, None
+
+
+def extract_input_schema(input_data: Dict[str, Any]) -> List[Tuple[str, Any]]:
+    """
+    Extract the schema from input data.
+
+    :param input_data: A dictionary representing the input data.
+    :return: A list of key-value pairs representing the input schema.
+    """
+    return [(key, input_data[key]) for key in input_data]
+
+
+def extract_output_schema(output_data: Dict[str, Any]) -> List[Tuple[str, str, str]]:
+    """
+    Extract the output schema from the output data.
+
+    :param output_data: A dictionary representing the output data.
+    :return: A list of list of tuples representing the output schema.
+    """
+    schema = output_data.get("schema", {})
+    if "properties" not in schema:
+        return []
+
+    return [
+        (key, value.get("type", ""), value.get("description", ""))
+        for key, value in schema["properties"].items()
+    ]

{mech_client-0.7.0 → mech_client-0.8.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mech-client"
-version = "0.7.0"
+version = "0.8.0"
 description = "Basic client to interact with a mech"
 authors = ["David Minarsch <david.minarsch@googlemail.com>"]
 readme = "README.md"

mech_client-0.7.0/mech_client/__init__.py

@@ -1,3 +0,0 @@
-"""Mech client."""
-
-__version__ = "0.7.0"