mobster-cli 0.4.3__py3-none-any.whl

mobile_agent/chat_here.py

@@ -0,0 +1,189 @@
+"""`mobster chat` when no Mobster answers: the conversation and its task run in this process."""
+
+import json
+import os
+import sys
+
+LOCAL = "terminal"
+NOT_RUNNING_HERE = ("Mobster isn't running. Open Mobster for Mac, or run `mobster` for a conversation in this "
+                    "terminal.")
+
+
+def service_running(url=None):
+    from .threads.client import Client, NoService, ServiceError
+    try:
+        Client.connect(url, origin="cli")
+    except NoService:
+        return False
+    except ServiceError:
+        return True  # it answers but refuses; threads/cli says why
+    return True
+
+
+def maybe_here(args):
+    if getattr(args, "url", None) or getattr(args, "workflow", None) or service_running():
+        return None  # a saved workflow lives in the running Mobster
+    text = " ".join(getattr(args, "message", None) or []).strip()
+    as_json = bool(getattr(args, "json", False))
+    if getattr(args, "wait", None) is not None or getattr(args, "keep_waiting", False):
+        message = "--wait and --keep-waiting need the Mobster app or `mobster serve` running. Open one, then try again."
+        if as_json:
+            print(json.dumps({"status": "couldnt_run", "error": message, "code": "not_running"}), flush=True)
+        else:
+            print(f"mobster: {message}", file=sys.stderr)
+        return 3
+    if getattr(args, "device", None):
+        from .cli_run import NoDevice, device_target
+        try:
+            args.wda_url = device_target(args.device)["wdaUrl"]
+        except NoDevice as error:
+            if as_json:
+                print(json.dumps({"status": "couldnt_run", "error": str(error), "code": "no_device"}), flush=True)
+            else:
+                print(f"mobster: {error}", file=sys.stderr)
+            return 3
+    if not text:
+        if sys.stdin.isatty() and sys.stdout.isatty() and not as_json:
+            from .tui import main as tui_main
+            for name, value in (("continue_", not getattr(args, "new", False)), ("demo", False), ("resume", None),
+                                ("app", None), ("no_bell", False), ("task", None), ("wda_url", None),
+                                ("env_file", getattr(args, "env_file", None))):
+                if not hasattr(args, name) or name == "continue_":
+                    setattr(args, name, value)
+            args.thread_id = getattr(args, "thread", None)
+            return tui_main(args)
+        if as_json:
+            print(json.dumps({"status": "couldnt_run", "error": NOT_RUNNING_HERE, "code": "not_running"}), flush=True)
+        else:
+            print(f"mobster: {NOT_RUNNING_HERE}", file=sys.stderr)
+        return 3
+    return run_here(args, text, as_json)
+
+
+def run_here(args, text, as_json):
+    from .engines import smart_key
+    from .smart_run import NOBODY_TO_ASK, Approver
+    from .threads import chat_once, recent
+    from .tui.session import Session, error_text
+    if not smart_key():
+        from .cli_run import NO_KEY
+        if as_json:
+            print(json.dumps({"status": "couldnt_run", "error": NO_KEY, "code": "no_key"}), flush=True)
+        else:
+            print(f"mobster: {NO_KEY}", file=sys.stderr)
+        return chat_once.EXIT_COULDNT
+    try:
+        session = Session(wda_url=getattr(args, "wda_url", None) or os.environ.get("MOBSTER_WDA_URL"),
+                          env_file=getattr(args, "env_file", None), origin="cli")
+    except Exception as error:  # noqa: BLE001 -- another terminal UI holds the history, or the runtime can't start
+        message = f"{error_text(error)} {NOT_RUNNING_HERE}"
+        if as_json:
+            print(json.dumps({"status": "couldnt_run", "error": message, "code": "not_running"}), flush=True)
+        else:
+            print(f"mobster: {message}", file=sys.stderr)
+        return chat_once.EXIT_COULDNT
+    try:
+        thread_id = None
+        try:
+            if session.threads() is None:
+                raise ValueError(NOT_RUNNING_HERE)
+            thread_id = _thread(session, args)
+            routed, run, response = session.send(thread_id, text)
+        except Exception as error:  # noqa: BLE001 -- the service's own sentence
+            message = error_text(error)
+            if as_json:
+                print(json.dumps({"status": "couldnt_run", "error": message,
+                                  **({"threadId": thread_id} if thread_id else {})}), flush=True)
+            else:
+                print(f"mobster: {message}", file=sys.stderr)
+            return chat_once.EXIT_COULDNT
+        recent.remember_thread(LOCAL, thread_id)
+        if routed == "remember":
+            return _offer(response, thread_id, as_json)
+        if run is None:
+            print(json.dumps({"threadId": thread_id, "status": routed}) if as_json else "Mobster has your message.")
+            return chat_once.EXIT_DONE
+        from .cli_run import event_printer, output
+        emit = output if as_json else event_printer(args)
+        if emit is not output:
+            from .engines import model_words, smart_model
+            emit.start(text, f"Mobster's agent, {model_words(smart_model())}")
+        approver = None if as_json else Approver(session, run, emit)
+
+        def on_event(event):
+            kind = event.get("event")
+            if emit is not output and kind != "run_finished":
+                emit(event)
+            if kind == "approval_requested":
+                if approver is not None:
+                    approver.ask(event)
+                else:
+                    session.answer(event.get("approval_id"), False, run=run)
+        try:
+            session.follow(run, on_event)
+        except KeyboardInterrupt:
+            session.stop(run)
+            return 130
+        finally:
+            if emit is not output:
+                emit.close()
+        public = run.public(include_events=False)
+        result = chat_once.outcome(public, thread_id)
+        if as_json:
+            if public.get("status") == "approval_denied":
+                result["note"] = NOBODY_TO_ASK
+            print(json.dumps(result, ensure_ascii=False, allow_nan=False), flush=True)
+        return chat_once.EXIT_DONE if result["status"] in chat_once.DONE else chat_once.EXIT_NOT_DONE
+    finally:
+        session.close()
+
+
+def _thread(session, args):
+    from .threads import recent
+    live = session.threads()
+    wanted = (getattr(args, "thread", None) or "").strip().lower()
+    if wanted:
+        live.thread(wanted)
+        return wanted
+    if not getattr(args, "new", False):
+        last = recent.last_thread(LOCAL)
+        previous = live.store.get(last) if last else None
+        if previous and not previous.get("archived"):
+            return last
+    return session.new_thread()["id"]
+
+
+def _offer(response, thread_id, as_json):
+    proposals = response.get("proposals") or []
+    if as_json:
+        print(json.dumps({"threadId": thread_id, "status": "remember", "proposals": proposals}), flush=True)
+        return 0
+    from .memory.store import default_store
+    from .style import palette
+    paint = palette(sys.stdout)
+    store = default_store()
+    if not proposals:
+        print("Mobster already has this.")
+        return 0
+    for proposal in proposals:
+        print(paint("Remember this?", "bold") + f"  “{proposal['text']}”")
+        print("  " + paint("Mobster will use it in tasks it fits. It stays on this Mac.", "tertiary"))
+        if not sys.stdin.isatty():
+            print("  " + paint("Answer it in `mobster` or Mobster for Mac.", "tertiary"))
+            continue
+        print("  " + paint("[y]", "accent", "bold") + " Remember   " + paint("[n]", "bold") + " Not now")
+        while True:
+            try:
+                word = input("  Type y or n, then Return: ").strip().lower()
+            except (EOFError, KeyboardInterrupt):
+                print()
+                word = "n"
+            if word in ("y", "n"):
+                break
+        try:
+            store.resolve_proposal(proposal["id"], word == "y")
+        except Exception as error:  # noqa: BLE001 -- the store's own sentence
+            print(f"mobster: {error}", file=sys.stderr)
+            return 1
+        print(paint("✓ ", "green") + "Mobster will remember this." if word == "y" else "Not remembered.")
+    return 0

mobile_agent/cli_constants.py

@@ -0,0 +1,5 @@
+"""What the `mobster` command and its argument parser (cli_parser.py) both need, so neither imports the other."""
+
+DEFAULT_WDA_URL = "http://127.0.0.1:8100"
+# Exit codes: 0 done, 1 the task or check did not succeed, 2 usage, 3 couldn't run (no phone, no key).
+EXIT_OK, EXIT_FAILED, EXIT_USAGE, EXIT_COULDNT = 0, 1, 2, 3