mech-client 0.2.16__py3-none-any.whl → 0.2.17__py3-none-any.whl

mech_client/__init__.py

@@ -1,3 +1,3 @@
 """Mech client."""
 
-__version__ = "0.2.16"
+__version__ = "0.2.17"

mech_client/cli.py

@@ -167,17 +167,39 @@ def tools_for_agents(agent_id: Optional[int], chain_config: str) -> None:
         result = get_tools_for_agents(agent_id, chain_config)
 
         if agent_id is not None:
-            headers = ["Tool Name", "Unique Identifier"]
+            headers = ["Tool Name", "Unique Identifier", "Mech Marketplace Support"]
             data: List[Tuple[str, ...]] = [
-                (str(tool["tool_name"]), str(tool["unique_identifier"]))
+                (
+                    str(tool["tool_name"]),
+                    str(tool["unique_identifier"]),
+                    "✓" if bool(tool["is_marketplace_supported"]) else "✗",
+                )
                 for tool in result["tools"]
             ]
         else:
-            headers = ["Agent ID", "Tool Name", "Unique Identifier"]
+            headers = [
+                "Agent ID",
+                "Tool Name",
+                "Unique Identifier",
+                "Mech Marketplace Support",
+            ]
+
             data = [
-                (str(agent_id), str(tool_name), f"{agent_id}-{tool_name}")
-                for agent_id, tools in result["agent_tools_map"].items()
-                for tool_name in tools
+                (
+                    str(agent_id),
+                    tool["tool_name"],
+                    tool["unique_identifier"],
+                    (
+                        "✓"
+                        if bool(
+                            tool["is_marketplace_supported"],
+                        )
+                        else "✗"
+                    ),
+                )
+                for agent_id, _ in result["agent_tools_map"].items()
+                for tool in result["all_tools_with_identifiers"]
+                if tool["unique_identifier"].startswith(f"{agent_id}-")
             ]
 
         click.echo(tabulate(data, headers=headers, tablefmt="grid"))
@@ -209,18 +231,35 @@ def tool_description(tool_id: str, chain_config: str) -> None:
 @click.argument("tool_id")
 @click.option("--chain-config", default="gnosis", help="Chain configuration to use.")
 def tool_io_schema(tool_id: str, chain_config: str) -> None:
-    """Fetch and display the input/output schema for a specific tool."""
+    """Fetch and display the tool's name and description along with the input/output schema for a specific tool."""
     try:
-        io_schema = get_tool_io_schema(tool_id, chain_config)
+        result = get_tool_io_schema(tool_id, chain_config)
+
+        name = result["name"]
+        description = result["description"]
         # Prepare data for tabulation
-        input_schema = [(key, io_schema["input"][key]) for key in io_schema["input"]]
+        input_schema = [(key, result["input"][key]) for key in result["input"]]
 
         # Handling nested output schema
         output_schema = []
-        if "properties" in io_schema["output"]["schema"]:
-            for key, value in io_schema["output"]["schema"]["properties"].items():
+        if "properties" in result["output"]["schema"]:
+            for key, value in result["output"]["schema"]["properties"].items():
                 output_schema.append((key, value["type"], value.get("description", "")))
 
+        # Display tool details in tabulated format
+        click.echo("Tool Details:")
+        click.echo(
+            tabulate(
+                [
+                    [
+                        name,
+                        description,
+                    ]
+                ],
+                headers=["Tool Name", "Tool Description"],
+                tablefmt="grid",
+            )
+        )
         # Display schemas in tabulated format
         click.echo("Input Schema:")
         click.echo(tabulate(input_schema, headers=["Field", "Value"], tablefmt="grid"))

mech_client/mech_tool_management.py

@@ -35,13 +35,14 @@ def get_total_supply(chain_config: str = "gnosis") -> int:
 
 
 def get_agent_tools(
-    agent_id: int, chain_config: str = "gnosis"
+    agent_id: int, chain_config: str = "gnosis", include_metadata: bool = False
 ) -> Optional[Union[List[str], Tuple[List[str], Dict[str, Any]]]]:
     """
     Fetch tools for a given agent ID.
 
     :param agent_id: The ID of the agent.
     :param chain_config: The chain configuration to use (default is "gnosis").
+    :param include_metadata: To include tools metadata or not (default is False)
     :return: A list of tools if successful, or a tuple of (list of tools, metadata) if metadata is included, or None if an error occurs.
     """
     try:
@@ -58,6 +59,7 @@ def get_agent_tools(
             ledger_api=ledger_api,
             agent_registry_contract=mech_config.agent_registry_contract,
             contract_abi_url=mech_config.contract_abi_url,
+            include_metadata=include_metadata,
         )
     except (requests.exceptions.RequestException, json.JSONDecodeError, KeyError) as e:
         print(f"An error occurred while fetching tools for agent {agent_id}: {e}")
@@ -78,31 +80,51 @@ def get_tools_for_agents(
         total_supply = get_total_supply(chain_config)
 
         if agent_id is not None:
-            tools = get_agent_tools(agent_id, chain_config)
-            if isinstance(tools, list):
-                tools_with_ids = [
-                    {"tool_name": tool, "unique_identifier": f"{agent_id}-{tool}"}
-                    for tool in tools
-                ]
-            else:
-                tools_with_ids = []
-            return {"agent_id": agent_id, "tools": tools_with_ids}
+            result = get_agent_tools(agent_id, chain_config, True)
+
+            if result is not None:
+                (tools, tool_metadata) = result
+
+                if isinstance(tools, list) and isinstance(tool_metadata, dict):
+                    tools_with_ids = [
+                        {
+                            "tool_name": tool,
+                            "unique_identifier": f"{agent_id}-{tool}",
+                            "is_marketplace_supported": (
+                                tool_metadata.get(tool, {}).get(
+                                    "isMechMarketplaceSupported", None
+                                )
+                            ),
+                        }
+                        for tool in tools
+                    ]
+                else:
+                    tools_with_ids = []
+                return {"agent_id": agent_id, "tools": tools_with_ids}
 
         all_tools_with_ids = []
         agent_tools_map = {}
 
         for current_agent_id in range(1, total_supply + 1):
-            tools = get_agent_tools(current_agent_id, chain_config)
-            if isinstance(tools, list):
-                tools_with_ids = [
-                    {
-                        "tool_name": tool,
-                        "unique_identifier": f"{current_agent_id}-{tool}",
-                    }
-                    for tool in tools
-                ]
-                agent_tools_map[current_agent_id] = tools
-                all_tools_with_ids.extend(tools_with_ids)
+            result = get_agent_tools(current_agent_id, chain_config, True)
+            if result is not None:
+                (tools, tool_metadata) = result
+
+                if isinstance(tools, list) and isinstance(tool_metadata, dict):
+                    tools_with_ids = [
+                        {
+                            "tool_name": tool,
+                            "unique_identifier": f"{current_agent_id}-{tool}",
+                            "is_marketplace_supported": (
+                                tool_metadata.get(tool, {}).get(
+                                    "isMechMarketplaceSupported", None
+                                )
+                            ),
+                        }
+                        for tool in tools
+                    ]
+                    agent_tools_map[current_agent_id] = tools
+                    all_tools_with_ids.extend(tools_with_ids)
 
         return {
             "all_tools_with_identifiers": all_tools_with_ids,
@@ -148,11 +170,11 @@ def get_tool_io_schema(
     unique_identifier: str, chain_config: str = "gnosis"
 ) -> Dict[str, Any]:
     """
-    Fetch the input and output schema of a specific tool based on a unique identifier.
+    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 input and output schemas.
+    :return: Dictionary containing name, description, input and output schemas.
     """
     parts = unique_identifier.split("-")
     agent_id = int(parts[0])
@@ -174,6 +196,8 @@ def get_tool_io_schema(
         tool_info = tool_metadata.get(tool_name, {})
         if isinstance(tool_info, dict):
             return {
+                "name": tool_info.get("name", {}),
+                "description": tool_info.get("description", {}),
                 "input": tool_info.get("input", {}),
                 "output": tool_info.get("output", {}),
             }

{mech_client-0.2.16.dist-info → mech_client-0.2.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: mech-client
-Version: 0.2.16
+Version: 0.2.17
 Summary: Basic client to interact with a mech
 License: Apache-2.0
 Author: David Minarsch
@@ -226,12 +226,12 @@ You will see an output like this:
 To get the description of a specific tool, use the `tool-description` command. You need to specify the unique identifier of the tool.
 
 ```bash
-mechx tool-description --unique-identifier <unique_identifier> --chain-config <chain_config>
+mechx tool-description <unique_identifier> --chain-config <chain_config>
 ```
 Example usage:
 
 ```bash
-mechx tool-description --unique-identifier "6-claude-prediction-offline" --chain-config gnosis
+mechx tool-description "6-claude-prediction-offline" --chain-config gnosis
 ```
 You will see an output like this:
 ```bash
@@ -244,16 +244,22 @@ Description for tool 6-claude-prediction-offline: Makes a prediction using Claud
 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
-mechx tool-io-schema --unique-identifier <unique_identifier> --chain-config <chain_config>
+mechx tool-io-schema <unique_identifier> --chain-config <chain_config>
 ```
 
 Example usage:
 
 ```bash
-mechx tool-io-schema --unique-identifier "6-prediction-offline" --chain-config gnosis
+mechx tool-io-schema "6-prediction-offline" --chain-config gnosis
 ```
 You will see an output like this:
 ```bash
+Tool Details:
++---------------------------+-----------------------------------------------+
+| Tool Name                 | Tool Description                              |
++===========================+===============================================+
+| OpenAI Prediction Offline | Makes a prediction using OpenAI GPT-3.5 Turbo |
++---------------------------+-----------------------------------------------+
 Input Schema:
 +-------------+----------------------------------+
 | Field       | Value                            |

{mech_client-0.2.16.dist-info → mech_client-0.2.17.dist-info}/RECORD

@@ -1,6 +1,6 @@
-mech_client/__init__.py,sha256=82jc8W2nwk575i4bZvj6qpAoJEn_97PMQSlAM7ma8u0,43
+mech_client/__init__.py,sha256=1i5v_YJZX04Olc98bp1Cl57tALIfGyEhL9dX_Vkmd70,43
 mech_client/acn.py,sha256=Rj_jLPvJ5loDQfGbu3a_O24cJC4SwIErLceSz_zVYS8,5356
-mech_client/cli.py,sha256=87ZBQAbzWpuf6s59FnBa4YO2eyDCyTOG22V54408MiY,8377
+mech_client/cli.py,sha256=cJ3QAP29QsbsyITVZbCVxZ7378ql2VPaEjar9Iotj0U,9555
 mech_client/configs/mechs.json,sha256=IABnne1lWi62l84Hzqh3XrP659uPKme2k32yAI0HBrg,4488
 mech_client/helpers/__init__.py,sha256=f13zpwGDaKQUjML-5Iq66rRfzg8IS5UNK5I8gBr7w54,1028
 mech_client/helpers/acn/README.md,sha256=WMXR2Lk0IpWjr3vpZ8cxcTHk4gwsx4wC06UPkwj9dbQ,1641
@@ -31,14 +31,14 @@ mech_client/helpers/p2p_libp2p_client/__init__.py,sha256=-GOP3D_JnmXTDomrMLCbnRk
 mech_client/helpers/p2p_libp2p_client/connection.py,sha256=pvfHtI-NhgDbay1wLNit6m8InH4c0p00c3hQo0I2jwQ,27887
 mech_client/helpers/p2p_libp2p_client/connection.yaml,sha256=giYV5FwwugD7ha9IqFHJtvs-Oz1jC5og9rpkstrTqoc,1763
 mech_client/interact.py,sha256=DOlUQBrfJKdDlfPnQ10tC8uAFTjlC3FGHOkPI2E_BYg,20846
-mech_client/mech_tool_management.py,sha256=_nOa8FLaow2XyztxnEMT9sJSlZEhUidmG8dFdR1lfV8,6550
+mech_client/mech_tool_management.py,sha256=XWQDcFStTK66-3ZzOwKtMo5c66KDjvbeDsIXHIP4SuU,7810
 mech_client/prompt_to_ipfs.py,sha256=XqSIBko15MEkpWOQNT97fRI6jNxMF5EDBDEPOJFdhyk,2533
 mech_client/push_to_ipfs.py,sha256=IfvgaPU79N_ZmCPF9d7sPCYz2uduZH0KjT_HQ2LHXoQ,2059
 mech_client/subgraph.py,sha256=4vY6QFyUVs15gS0SvanJbvAxb3aie07IuxQnfMMnStc,4931
 mech_client/to_png.py,sha256=pjUcFJ63MJj_r73eqnfqCWMtlpsrj6H4ZmgvIEmRcFw,2581
 mech_client/wss.py,sha256=hRInQjjsyOrs8dmgBb2VpJvpNt6Tx0aEiY3OhOPQvIs,6600
-mech_client-0.2.16.dist-info/LICENSE,sha256=mdBDB-mWKV5Cz4ejBzBiKqan6Z8zVLAh9xwM64O2FW4,11339
-mech_client-0.2.16.dist-info/METADATA,sha256=RKWiEjr5xeovuc4sIDI633vmOd4M7oBStQdzjoSBPB0,17358
-mech_client-0.2.16.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
-mech_client-0.2.16.dist-info/entry_points.txt,sha256=SbRMRsayzD8XfNXhgwPuXEqQsdZ5Bw9XDPnUuaDExyY,45
-mech_client-0.2.16.dist-info/RECORD,,
+mech_client-0.2.17.dist-info/LICENSE,sha256=mdBDB-mWKV5Cz4ejBzBiKqan6Z8zVLAh9xwM64O2FW4,11339
+mech_client-0.2.17.dist-info/METADATA,sha256=DvJxvDI_5aM_8-kVHyk6BEbg_sbowSatqqUOji50jK0,17682
+mech_client-0.2.17.dist-info/WHEEL,sha256=7Z8_27uaHI_UZAc4Uox4PpBhQ9Y5_modZXWMxtUi4NU,88
+mech_client-0.2.17.dist-info/entry_points.txt,sha256=SbRMRsayzD8XfNXhgwPuXEqQsdZ5Bw9XDPnUuaDExyY,45
+mech_client-0.2.17.dist-info/RECORD,,