olas-operate-middleware 0.6.2__tar.gz → 0.6.3__tar.gz

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: olas-operate-middleware
-Version: 0.6.2
+Version: 0.6.3
 Summary: 
 Author: David Vilela
 Author-email: dvilelaf@gmail.com
@@ -23,7 +23,7 @@ Requires-Dist: eth-keys (==0.4.0)
 Requires-Dist: eth-rlp (==0.3.0)
 Requires-Dist: eth-typing (==3.5.2)
 Requires-Dist: eth-utils (==2.3.1)
-Requires-Dist: fastapi (==0.110.0)
+Requires-Dist: fastapi (==0.110.3)
 Requires-Dist: frozenlist (==1.4.1)
 Requires-Dist: halo (==0.0.31)
 Requires-Dist: hexbytes (==0.3.1)
@@ -38,7 +38,7 @@ Requires-Dist: open-autonomy (>=0.19.11,<0.20.0)
 Requires-Dist: psutil (>=5.9.8,<6.0.0)
 Requires-Dist: pyinstaller (>=6.8.0,<7.0.0)
 Requires-Dist: requests-toolbelt (==1.0.0)
-Requires-Dist: starlette (==0.36.3)
+Requires-Dist: starlette (==0.37.2)
 Requires-Dist: twikit (==2.2.0)
 Requires-Dist: uvicorn (==0.27.0)
 Requires-Dist: web3 (==6.1.0)

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/cli.py

@@ -56,6 +56,7 @@ from operate.migration import MigrationManager
 from operate.operate_types import Chain, DeploymentStatus, LedgerType
 from operate.quickstart.analyse_logs import analyse_logs
 from operate.quickstart.claim_staking_rewards import claim_staking_rewards
+from operate.quickstart.reset_configs import reset_configs
 from operate.quickstart.reset_password import reset_password
 from operate.quickstart.reset_staking import reset_staking
 from operate.quickstart.run_service import run_service
@@ -537,6 +538,35 @@ def create_app(  # pylint: disable=too-many-locals, unused-argument, too-many-st
         wallet, mnemonic = manager.create(ledger_type=ledger_type)
         return JSONResponse(content={"wallet": wallet.json, "mnemonic": mnemonic})
 
+    @app.post("/api/wallet/private_key")
+    @with_retries
+    async def _get_private_key(request: Request) -> t.List[t.Dict]:
+        """Get Master EOA private key."""
+        if operate.user_account is None:
+            return JSONResponse(
+                content={
+                    "error": "Cannot retrieve private key; User account does not exist!"
+                },
+                status_code=HTTPStatus.BAD_REQUEST,
+            )
+
+        data = await request.json()
+        password = data.get("password")
+        error = None
+        if operate.password is None:
+            error = {"error": "You need to login before retrieving the private key"}
+        if operate.password != password:
+            error = {"error": "Password is not valid"}
+        if error is not None:
+            return JSONResponse(
+                content=error,
+                status_code=HTTPStatus.UNAUTHORIZED,
+            )
+
+        ledger_type = data.get("ledger_type", LedgerType.ETHEREUM.value)
+        wallet = operate.wallet_manager.load(ledger_type=LedgerType(ledger_type))
+        return JSONResponse(content={"private_key": wallet.crypto.private_key})
+
     @app.get("/api/extended/wallet")
     @with_retries
     async def _get_wallet_safe(request: Request) -> t.List[t.Dict]:
@@ -1145,6 +1175,20 @@ def qs_claim(
     claim_staking_rewards(operate=operate, config_path=config)
 
 
+@_operate.command(name="reset-configs")
+def qs_reset_configs(
+    config: Annotated[str, params.String(help="Quickstart config file path")],
+    attended: Annotated[
+        str, params.String(help="Run in attended/unattended mode (default: true")
+    ] = "true",
+) -> None:
+    """Reset configs."""
+    os.environ["ATTENDED"] = attended.lower()
+    operate = OperateApp()
+    operate.setup()
+    reset_configs(operate=operate, config_path=config)
+
+
 @_operate.command(name="reset-staking")
 def qs_reset_staking(
     config: Annotated[str, params.String(help="Quickstart config file path")],

olas_operate_middleware-0.6.3/operate/quickstart/reset_configs.py

@@ -0,0 +1,109 @@
+# -*- coding: utf-8 -*-
+# ------------------------------------------------------------------------------
+#
+#   Copyright 2023-2024 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.
+#
+# ------------------------------------------------------------------------------
+"""Reset configurations."""
+
+import json
+from typing import Callable, Optional, TYPE_CHECKING, cast
+
+from operate.quickstart.run_service import load_local_config
+from operate.quickstart.utils import (
+    ask_or_get_from_env,
+    ask_yes_or_no,
+    check_rpc,
+    print_section,
+    print_title,
+)
+
+
+if TYPE_CHECKING:
+    from operate.cli import OperateApp
+
+
+def _ask_to_change(
+    name: str,
+    env_var: str,
+    old_value: str,
+    hidden: bool = False,
+    validator: Callable[[Optional[str]], bool] = lambda x: False if x is None else True,
+) -> str:
+    """Ask user if they want to change a configuration value."""
+    old_value_str = old_value
+    if hidden:
+        if len(old_value_str) < 4:
+            old_value_str = "*" * len(old_value_str)
+        else:
+            old_value_str = "*" * len(old_value_str[:-4]) + old_value_str[-4:]
+
+    print(f"\nCurrent '{name}' is set to: {old_value_str}")
+    if ask_yes_or_no(f"Do you want to change the '{name}'?"):
+        new_value = None
+        while not validator(new_value):
+            new_value = ask_or_get_from_env(
+                prompt=f"Enter new value for '{name}' {'[hidden]' if hidden else ''}: ",
+                env_var_name=env_var,
+                is_pass=hidden,
+            )
+
+        return str(new_value)
+
+    return old_value
+
+
+def reset_configs(operate: "OperateApp", config_path: str) -> None:
+    """Reset configurations."""
+    with open(config_path, "r") as config_file:
+        template = json.load(config_file)
+
+    operate.service_manager().migrate_service_configs()
+    operate.wallet_manager.migrate_wallet_configs()
+
+    print_title(f"Reset your {template['name']} configurations")
+
+    # check if agent was started before
+    config = load_local_config(
+        operate=operate, service_name=cast(str, template["name"])
+    )
+    if not config.path.exists():
+        print("No previous agent setup found. Exiting.")
+        return
+
+    if config.rpc is None:
+        config.rpc = {}
+
+    for chain_name in config.rpc:
+        config.rpc[chain_name] = _ask_to_change(
+            name=f"{chain_name.capitalize()} RPC URL",
+            env_var=f"{chain_name.upper()}_LEDGER_RPC",
+            old_value=config.rpc[chain_name],
+            hidden=True,
+            validator=check_rpc,
+        )
+
+    if config.user_provided_args is None:
+        config.user_provided_args = {}
+
+    for env_var in config.user_provided_args:
+        config.user_provided_args[env_var] = _ask_to_change(
+            name=env_var,
+            env_var=env_var,
+            old_value=config.user_provided_args[env_var],
+        )
+
+    config.store()
+    print_section("Configurations updated")

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/quickstart/run_service.py

@@ -99,8 +99,6 @@ QS_STAKING_PROGRAMS: t.Dict[Chain, t.Dict[str, str]] = {
         "quickstart_beta_mech_marketplace_expert_8": "trader",
         "mech_marketplace": "mech",
         "marketplace_supply_alpha": "mech",
-        "marketplace_demand_alpha_1": "mech",
-        "marketplace_demand_alpha_2": "mech",
     },
     Chain.OPTIMISTIC: {
         "optimus_alpha_2": "optimus",
@@ -111,8 +109,6 @@ QS_STAKING_PROGRAMS: t.Dict[Chain, t.Dict[str, str]] = {
     Chain.BASE: {
         "meme_base_alpha_2": "memeooorr",
         "marketplace_supply_alpha": "mech",
-        "marketplace_demand_alpha_1": "mech",
-        "marketplace_demand_alpha_2": "mech",
         "agents_fun_1": "memeooorr",
         "agents_fun_2": "memeooorr",
         "agents_fun_3": "memeooorr",

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/quickstart/utils.py

@@ -188,8 +188,12 @@ def ask_yes_or_no(question: str) -> bool:
     """Ask a yes/no question."""
     if os.environ.get("ATTENDED", "true").lower() != "true":
         return True
-    response = input(f"{question} (yes/no): ").strip().lower()
-    return response in ["yes", "y"]
+    while True:
+        response = input(f"{question} (yes/no): ").strip().lower()
+        if response.lower() in ("yes", "y"):
+            return True
+        if response.lower() in ("no", "n"):
+            return False
 
 
 def ask_or_get_from_env(

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/services/manage.py

@@ -106,6 +106,9 @@ HTTP_OK = 200
 URI_HASH_POSITION = 7
 IPFS_GATEWAY = "https://gateway.autonolas.tech/ipfs/"
 DEFAULT_TOPUP_THRESHOLD = 0.5
+# At the moment, we only support running one agent per service locally on a machine.
+# If multiple agents are provided in the service.yaml file, only the 0th index config will be used.
+NUM_LOCAL_AGENT_INSTANCES = 1
 
 
 class ServiceManager:
@@ -227,7 +230,7 @@ class ServiceManager:
         if not service.keys:
             service.keys = [
                 self.keys_manager.get(self.keys_manager.create())
-                for _ in range(service.helper.config.number_of_agents)
+                for _ in range(NUM_LOCAL_AGENT_INSTANCES)
             ]
             service.store()
 
@@ -267,7 +270,7 @@ class ServiceManager:
         if not service.keys:
             service.keys = [
                 self.keys_manager.get(self.keys_manager.create())
-                for _ in range(service.helper.config.number_of_agents)
+                for _ in range(NUM_LOCAL_AGENT_INSTANCES)
             ]
             service.store()
 
@@ -437,7 +440,7 @@ class ServiceManager:
                 ocm.mint(
                     package_path=service.package_absolute_path_absolute_path,
                     agent_id=staking_params["agent_ids"][0],
-                    number_of_slots=service.helper.config.number_of_agents,
+                    number_of_slots=NUM_LOCAL_AGENT_INSTANCES,
                     cost_of_bond=(
                         staking_params["min_staking_deposit"]
                         if user_params.use_staking
@@ -709,7 +712,7 @@ class ServiceManager:
                 )
                 protocol_asset_requirements[target_staking_params["staking_token"]] = (
                     target_staking_params["min_staking_deposit"]
-                    * service.helper.config.number_of_agents
+                    * NUM_LOCAL_AGENT_INSTANCES
                 )
             else:
                 protocol_asset_requirements = {}
@@ -791,7 +794,7 @@ class ServiceManager:
                         sftxb.get_mint_tx_data(
                             package_path=service.package_absolute_path,
                             agent_id=agent_id,
-                            number_of_slots=service.helper.config.number_of_agents,
+                            number_of_slots=NUM_LOCAL_AGENT_INSTANCES,
                             cost_of_bond=(
                                 target_staking_params["min_staking_deposit"]
                                 if user_params.use_staking
@@ -841,7 +844,7 @@ class ServiceManager:
                     sftxb.get_mint_tx_data(
                         package_path=service.package_absolute_path,
                         agent_id=agent_id,
-                        number_of_slots=service.helper.config.number_of_agents,
+                        number_of_slots=NUM_LOCAL_AGENT_INSTANCES,
                         cost_of_bond=(
                             target_staking_params["min_staking_deposit"]
                             if user_params.use_staking
@@ -2504,7 +2507,7 @@ class ServiceManager:
         chain_config = service.chain_configs[chain]
         user_params = chain_config.chain_data.user_params
         ledger_config = chain_config.ledger_config
-        number_of_agents = service.helper.config.number_of_agents
+        number_of_agents = NUM_LOCAL_AGENT_INSTANCES
         os.environ["CUSTOM_CHAIN_RPC"] = ledger_config.rpc
         sftxb = self.get_eth_safe_tx_builder(ledger_config=ledger_config)
         service_asset_requirements: defaultdict = defaultdict(int)

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/services/service.py

@@ -320,13 +320,19 @@ class HostDeploymentGenerator(BaseDeploymentGenerator):
         tendermint_executable = str(
             Path(os.path.dirname(sys.executable)) / "tendermint"
         )
-        tendermint_executable = "tendermint"
+
         if platform.system() == "Windows":
             env["PATH"] = os.path.dirname(sys.executable) + ";" + os.environ["PATH"]
             tendermint_executable = str(
                 Path(os.path.dirname(sys.executable)) / "tendermint.exe"
             )
-            tendermint_executable = "tendermint.exe"
+
+        if not (getattr(sys, "frozen", False) and hasattr(sys, "_MEIPASS")):
+            # we dont run inside pyinstaller, mean DEV mode!
+            tendermint_executable = "tendermint"
+            if platform.system() == "Windows":
+                tendermint_executable = "tendermint.exe"
+
         subprocess.run(  # pylint: disable=subprocess-run-check # nosec
             args=[
                 tendermint_executable,
@@ -1251,6 +1257,8 @@ class Service(LocalResource):
                 config  # type: ignore
             )
 
+            self.chain_configs[chain].ledger_config.rpc = config["rpc"]
+
         self.store()
 
     def consume_env_variables(self) -> None:

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/operate/services/utils/mech.py

@@ -26,7 +26,7 @@ from aea_ledger_ethereum import Web3
 
 from operate.constants import MECH_MARKETPLACE_JSON_URL
 from operate.operate_types import Chain
-from operate.quickstart.utils import print_section, unit_to_wei
+from operate.quickstart.utils import print_section
 from operate.services.protocol import EthSafeTxBuilder
 from operate.services.service import Service
 from operate.utils.gnosis import SafeOperation
@@ -74,9 +74,11 @@ def deploy_mech(sftxb: EthSafeTxBuilder, service: Service) -> Tuple[str, str]:
         mech_type
     ]
 
-    # 0.01xDAI hardcoded for price
-    # better to be configurable and part of local config
-    mech_request_price = unit_to_wei(0.01)
+    mech_request_price = int(
+        service.env_variables.get("MECH_REQUEST_PRICE", {}).get(
+            "value", 10000000000000000
+        )
+    )
     contract = sftxb.ledger_api.api.eth.contract(
         address=Web3.to_checksum_address(mech_marketplace_address), abi=abi
     )

{olas_operate_middleware-0.6.2 → olas_operate_middleware-0.6.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "olas-operate-middleware"
-version = "0.6.2"
+version = "0.6.3"
 description = ""
 authors = ["David Vilela <dvilelaf@gmail.com>", "Viraj Patel <vptl185@gmail.com>"]
 readme = "README.md"
@@ -27,7 +27,7 @@ open-aea-cli-ipfs = "==1.65.0"
 clea = "==0.1.0rc4"
 cytoolz = "==0.12.3"
 docker = "6.1.2"
-fastapi = "==0.110.0"
+fastapi = "0.110.3"
 eth-hash = "==0.7.0"
 eth-account = "==0.8.0"
 eth-keyfile = "==0.6.1"
@@ -42,7 +42,7 @@ ipfshttpclient = "==0.8.0a2"
 jsonschema = "==4.3.3"
 multidict = "==6.0.5"
 requests-toolbelt = "1.0.0"
-starlette = "==0.36.3"
+starlette = "0.37.2"
 uvicorn = "==0.27.0"
 web3 = "==6.1.0"
 psutil = "^5.9.8"

olas_operate_middleware-0.6.2/operate/data/contracts/service_staking_token/contract.yaml

@@ -1,23 +0,0 @@
-name: service_staking_token
-author: valory
-version: 0.1.0
-type: contract
-description: Service staking token contract
-license: Apache-2.0
-aea_version: '>=1.0.0, <2.0.0'
-fingerprint:
-  __init__.py: bafybeid3wfzglolebuo6jrrsopswzu4lk77bm76mvw3euizlsjtnt3wmgu
-  build/ServiceStakingToken.json: bafybeiclsckyk2kkuwhrmifupugg7ixj2hpwhd3mjghwgcdlrjqbwafwym
-  contract.py: bafybeiff62bquzeisbd6iptqdjetrhlkt2ut5d7j6md2kqinrqgmbveceu
-fingerprint_ignore_patterns: []
-contracts: []
-class_name: ServiceStakingTokenContract
-contract_interface_paths:
-  ethereum: build/ServiceStakingToken.json
-dependencies:
-  open-aea-ledger-ethereum:
-    version: ==1.42.0
-  open-aea-test-autonomy:
-    version: ==0.13.6
-  web3:
-    version: <7,>=6.0.0