kiwi-code 0.0.445__tar.gz → 0.0.446__tar.gz

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/PKG-INFO

@@ -1,10 +1,9 @@
 Metadata-Version: 2.4
 Name: kiwi-code
-Version: 0.0.445
+Version: 0.0.446
 Summary: A textual-based terminal user interface application
 Project-URL: Homepage, https://meetkiwi.ai
-Project-URL: Repository, https://github.com/jetoslabs/kiwi-code
-Author-email: Anurag Jha <anurag@meetkiwi.co>
+Author-email: Anurag <anurag@meetkiwi.co>
 License: Proprietary
 Keywords: cli,terminal,textual,tui
 Classifier: Development Status :: 3 - Alpha
@@ -15,11 +14,11 @@ Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Python: <4.0,>=3.11
-Requires-Dist: autobots-client==0.1.1
 Requires-Dist: httpx>=0.25.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: psutil>=5.9.0
 Requires-Dist: pydantic>=2.12.5
+Requires-Dist: pykiwiai==0.0.11
 Requires-Dist: setproctitle>=1.3.0
 Requires-Dist: textual-dev>=1.8.0
 Requires-Dist: textual>=8.1.1

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "kiwi-code"
-version = "0.0.445"
+version = "0.0.446"
 description = "A textual-based terminal user interface application"
 readme = {file = "README.md", content-type = "text/markdown"}
 requires-python = ">=3.11,<4.0"
 dependencies = [
-    "autobots-client==0.1.1",
+    "pykiwiai==0.0.11",
     "loguru>=0.7.3",
     "pydantic>=2.12.5",
     "textual>=8.1.1",
@@ -19,7 +19,7 @@ dependencies = [
 ]
 
 authors = [
-    { name = "Anurag Jha", email = "anurag@meetkiwi.co" }
+    { name = "Anurag", email = "anurag@meetkiwi.co" }
 ]
 license = { text = "Proprietary" }
 keywords = ["tui", "textual", "terminal", "cli"]
@@ -35,7 +35,6 @@ classifiers = [
 
 [project.urls]
 Homepage = "https://meetkiwi.ai"
-Repository = "https://github.com/jetoslabs/kiwi-code"
 
 [project.scripts]
 kiwi = "kiwi_tui.main:main"
@@ -55,8 +54,9 @@ filterwarnings = [
     "ignore::DeprecationWarning",
 ]
 
+
 [tool.hatch.build.targets.wheel]
-packages = ["src/kiwi_cli", "src/kiwi_tui", "src/kiwi_runtime"]
+packages = ["src/common", "src/kiwi_cli", "src/kiwi_tui", "src/kiwi_runtime"]
 
 [build-system]
 requires = ["hatchling"]

kiwi_code-0.0.446/scripts/ws_auth_check.py

@@ -0,0 +1,76 @@
+"""Quick WS handshake + auth check.
+
+Usage:
+  uv run python scripts/ws_auth_check.py --token "$ACCESS_TOKEN"
+
+Notes:
+- Pass the *raw* access token. If you paste "Bearer <token>", we will strip the prefix.
+- This script does NOT print the token.
+"""
+
+import argparse
+import asyncio
+import json
+import os
+import sys
+import uuid
+
+import websockets
+
+
+def _normalize_token(token: str) -> str:
+    t = token.strip()
+    if t.lower().startswith("bearer "):
+        t = t.split(" ", 1)[1].strip()
+    return t
+
+
+async def main() -> None:
+    ap = argparse.ArgumentParser()
+    ap.add_argument("--token", required=True, help="Access token (raw JWT; 'Bearer ' prefix is ok)")
+    ap.add_argument(
+        "--ws-uri",
+        default="wss://api.runkai.ai/v1/terminal/ws/connect",
+        help="Full ws connect URI (default: wss://api.runkai.ai/v1/terminal/ws/connect)",
+    )
+    ap.add_argument(
+        "--mode",
+        default="restricted",
+        choices=["restricted", "full"],
+        help="Mode to send in auth payload (matches kiwi-runtime)",
+    )
+    args = ap.parse_args()
+
+    token = _normalize_token(args.token)
+
+    async with websockets.connect(
+        args.ws_uri,
+        ping_interval=None,
+        open_timeout=8,
+        close_timeout=3,
+        max_queue=8,
+    ) as ws:
+        msg = {
+            "type": "auth",
+            "token": token,
+            "agent_id": str(uuid.uuid4()),
+            "platform": sys.platform,
+            "mode": args.mode,
+            "allowed_dirs": [os.getcwd()],
+        }
+        await ws.send(json.dumps(msg))
+
+        resp_raw = await asyncio.wait_for(ws.recv(), timeout=10)
+        resp = json.loads(resp_raw)
+
+        print("AUTH_RESP_TYPE", resp.get("type"))
+        print("AUTH_RESP_KEYS", sorted(resp.keys()))
+        if resp.get("type") == "error":
+            print("AUTH_ERROR_MESSAGE", (resp.get("message") or "").strip()[:300])
+        else:
+            safe = {k: resp.get(k) for k in ("type", "agent_id", "mode", "consumer_id", "message") if k in resp}
+            print("AUTH_RESP_SAFE", safe)
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

kiwi_code-0.0.446/src/common/banner.py

@@ -0,0 +1,61 @@
+"""Shared Kiwi AI banner assets used by kiwi-runtime and kiwi-tui."""
+
+# Block-character kiwi bird logo (16 lines, 33 chars wide).
+LOGO = [
+    "              ▄██▄              ",
+    "    ▄██▄▄     ████     ▄▄██▄    ",
+    "    ▀█████▄   █▀▀█   ▄█████▀   ",
+    "     ▀███▀▀█  ████  █▀▀███▀    ",
+    "       ▀█▄  █ ████ █  ▄█▀      ",
+    "          ▀▀▄██████▄▀▀         ",
+    "▄██████▀▀█▄██████████▄█▀▀██████▄",
+    "▀██████▄▄█▀██████████▀█▄▄██████▀",
+    "          ▄▄▀██████▀▄▄         ",
+    "       ▄█▀  █ ████ █  ▀█▄      ",
+    "     ▄███▄▄█  ████  █▄▄███▄    ",
+    "    ▄█████▀   █▄▄█   ▀█████▄   ",
+    "    ▀██▀▀     ████     ▀▀██▀   ",
+    "              ▀██▀              ",
+]
+
+# "KIWI AI" block text (6 lines tall).
+TEXT = [
+    "██╗  ██╗██╗██╗    ██╗██╗     █████╗ ██╗",
+    "██║ ██╔╝██║██║    ██║██║    ██╔══██╗██║",
+    "█████╔╝ ██║██║ █╗ ██║██║    ███████║██║",
+    "██╔═██╗ ██║██║███╗██║██║    ██╔══██║██║",
+    "██║  ██╗██║╚███╔███╔╝██║    ██║  ██║██║",
+    "╚═╝  ╚═╝╚═╝ ╚══╝╚══╝ ╚═╝    ╚═╝  ╚═╝╚═╝",
+]
+
+TAGLINE = "Think fast. Code smarter. Automate everything."
+
+# Brand color — darker cyan
+CYAN = "rgb(0,168,184)"
+CYAN_HEX = "#00a8b8"
+
+# Layout constants (matching kiwi-runtime banner).
+LOGO_W = 33
+GAP = "  "
+TEXT_START = 5
+
+
+def compose_banner_lines(tagline: str | None = None) -> list[str]:
+    """Compose logo + KIWI AI text side by side, returning a list of lines."""
+    effective_tagline = tagline if tagline is not None else TAGLINE
+    lines = []
+    for i, logo_line in enumerate(LOGO):
+        padded = logo_line.ljust(LOGO_W)
+        text_idx = i - TEXT_START
+        if 0 <= text_idx < len(TEXT):
+            lines.append(f"{padded}{GAP}{TEXT[text_idx]}")
+        elif text_idx == len(TEXT) + 1:
+            lines.append(f"{padded}{GAP}{effective_tagline}")
+        else:
+            lines.append(padded)
+    return lines
+
+
+def compose_banner_text(tagline: str | None = None) -> str:
+    """Return the full banner as a single multi-line string."""
+    return "\n".join(compose_banner_lines(tagline=tagline))

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/__init__.py

@@ -1,3 +1,3 @@
 """Kiwi CLI - command-line interface and shared infrastructure modules."""
 
-__version__ = "0.0.445"
+__version__ = "0.0.446"

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/cli.py

@@ -3,7 +3,7 @@ from typing import Optional, Callable
 from loguru import logger
 logger.remove()  # Suppress loguru console output in CLI mode
 import typer
-from autobots_client import AuthenticatedClient
+from pykiwiai import AuthenticatedClient
 from .auth import TokenManager
 from .models import AppConfig
 from .server import http_url_from_server

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/client.py

@@ -6,20 +6,20 @@ import json
 import codecs
 import httpx
 from loguru import logger
-from autobots_client import Client, AuthenticatedClient
-from autobots_client.api.hello import hello_v1_hello_get
-from autobots_client.api.auth import (
+from pykiwiai import Client, AuthenticatedClient
+from pykiwiai.api.hello import hello_v1_hello_get
+from pykiwiai.api.auth import (
     return_user_and_session_v1_auth_post,
 )
-from autobots_client.api.actions import async_run_action_v1_actions_id_async_run_post
-from autobots_client.api.action_results import get_action_result_v1_action_results_id_get
-from autobots_client.models.body_return_user_and_session_v1_auth_post import (
+from pykiwiai.api.actions import async_run_action_v1_actions_id_async_run_post
+from pykiwiai.api.action_results import get_action_result_v1_action_results_id_get
+from pykiwiai.models.body_return_user_and_session_v1_auth_post import (
     BodyReturnUserAndSessionV1AuthPost,
 )
-from autobots_client.models.body_async_run_action_v1_actions_id_async_run_post import (
+from pykiwiai.models.body_async_run_action_v1_actions_id_async_run_post import (
     BodyAsyncRunActionV1ActionsIdAsyncRunPost,
 )
-from autobots_client.models.body_async_run_action_v1_actions_id_async_run_post_input import (
+from pykiwiai.models.body_async_run_action_v1_actions_id_async_run_post_input import (
     BodyAsyncRunActionV1ActionsIdAsyncRunPostInput,
 )
 from .models import AuthTokens, LoginCredentials
@@ -253,8 +253,8 @@ class AutobotsClientWrapper:
 
             import mimetypes
             from pathlib import Path
-            from autobots_client.api.files import upload_files_v1_files_post
-            from autobots_client.models.body_upload_files_v1_files_post import BodyUploadFilesV1FilesPost
+            from pykiwiai.api.files import upload_files_v1_files_post
+            from pykiwiai.models.body_upload_files_v1_files_post import BodyUploadFilesV1FilesPost
 
             # Validate all files exist first
             paths = []
@@ -344,7 +344,7 @@ class AutobotsClientWrapper:
             logger.info(f"Request body: {body.to_dict()}")
 
             # Import UNSET for optional parameter
-            from autobots_client.types import UNSET
+            from pykiwiai.types import UNSET
 
             response = async_run_action_v1_actions_id_async_run_post.sync_detailed(
                 id=action_id,
@@ -464,7 +464,8 @@ class AutobotsClientWrapper:
                 logger.info(f"Result structure keys: {result_dict.keys()}")
                 if "result" in result_dict and isinstance(result_dict["result"], dict):
                     logger.info(f"Result.result keys: {result_dict['result'].keys()}")
-
+                    result = result_dict['result']
+                    logger.info(f"Result's results : {result['results']}")
                 return True, result_dict, "Result retrieved"
             else:
                 error_msg = f"Failed to get result: status {response.status_code}"

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/commands.py

@@ -5,17 +5,17 @@ and returns a list of strings (output lines).
 """
 
 from typing import Optional
-from autobots_client import AuthenticatedClient
-from autobots_client.api.actions import (
+from pykiwiai import AuthenticatedClient
+from pykiwiai.api.actions import (
     list_actions_v1_actions_get,
     get_action_v1_actions_id_get,
 )
-from autobots_client.api.action_results import (
+from pykiwiai.api.action_results import (
     list_action_result_v1_action_results_get,
     get_action_result_v1_action_results_id_get,
 )
 
-from autobots_client.types import UNSET
+from pykiwiai.types import UNSET
 
 
 # Helpers
@@ -35,6 +35,18 @@ def _val(v) -> str:
     return str(v)
 
 
+def _enum(v) -> str:
+    """Return a stable string for either Enum-like objects or plain strings.
+
+    pykiwiai <=0.0.1 used Enum types (value attribute).
+    pykiwiai >=0.0.11 uses Literal[str] so fields are plain strings.
+    """
+    if v is None or isinstance(v, type(UNSET)):
+        return ""
+    return str(getattr(v, "value", v))
+
+
+
 # actions
 
 def actions_list(client: AuthenticatedClient, *, name: Optional[str] = None, limit: int = 20, offset: int = 0) -> list[str]:
@@ -57,7 +69,7 @@ def actions_list(client: AuthenticatedClient, *, name: Optional[str] = None, lim
     ]
     for doc in result.docs:
         lines.append(
-            f"{doc.field_id:<28} {_val(doc.name):<30} {_val(doc.type_.value):<25} "
+            f"{doc.field_id:<28} {_val(doc.name):<30} {_val(_enum(doc.type_)):<25} "
             f"{_val(doc.version):<6} {_fmt_date(doc.created_at)}"
         )
     return lines
@@ -73,7 +85,7 @@ def actions_get(client: AuthenticatedClient, *, id: str) -> list[str]:
     return [
         f"ID:          {doc.field_id}",
         f"Name:        {doc.name}",
-        f"Type:        {doc.type_.value}",
+        f"Type:        {_enum(doc.type_)}",
         f"Version:     {_val(doc.version)}",
         f"Description: {_val(doc.description)}",
         f"Published:   {_val(doc.is_published)}",
@@ -88,13 +100,13 @@ def actions_get(client: AuthenticatedClient, *, id: str) -> list[str]:
 def runs_list(client: AuthenticatedClient, *, action_id: Optional[str] = None, action_name: Optional[str] = None,
               status: Optional[str] = None, limit: int = 20, offset: int = 0) -> list[str]:
     """List action runs. Returns output lines."""
-    from autobots_client.models.event_result_status import EventResultStatus
+    from pykiwiai.models.event_result_status import check_event_result_status
 
     status_enum = UNSET
     if status:
         try:
-            status_enum = EventResultStatus(status.lower())
-        except ValueError:
+            status_enum = check_event_result_status(status.lower())
+        except TypeError:
             return [f"Invalid status: {status}. Valid: processing, success, error, stuck, waiting"]
 
     resp = list_action_result_v1_action_results_get.sync_detailed(
@@ -117,7 +129,7 @@ def runs_list(client: AuthenticatedClient, *, action_id: Optional[str] = None, a
     ]
     for doc in result.docs:
         lines.append(
-            f"{doc.field_id:<28} {doc.status.value:<12} {_val(doc.name):<30} "
+            f"{doc.field_id:<28} {_enum(doc.status):<12} {_val(doc.name):<30} "
             f"{_fmt_date(doc.created_at):<18} {_fmt_date(doc.updated_at)}"
         )
     return lines
@@ -132,9 +144,9 @@ def runs_get(client: AuthenticatedClient, *, id: str) -> list[str]:
     doc = resp.parsed
     lines = [
         f"ID:          {doc.field_id}",
-        f"Status:      {doc.status.value}",
+        f"Status:      {_enum(doc.status)}",
         f"Name:        {_val(doc.name)}",
-        f"Type:        {doc.type_.value}",
+        f"Type:        {_enum(doc.type_)}",
         f"Saved:       {_val(doc.is_saved)}",
         f"Created:     {_fmt_date(doc.created_at)}",
         f"Updated:     {_fmt_date(doc.updated_at)}",
@@ -144,7 +156,7 @@ def runs_get(client: AuthenticatedClient, *, id: str) -> list[str]:
             "",
             "Action:",
             f"  Name:    {doc.result.name}",
-            f"  Type:    {doc.result.type_.value}",
+            f"  Type:    {_enum(doc.result.type_)}",
             f"  ID:      {doc.result.field_id}",
         ]
     return lines

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/models.py

@@ -115,7 +115,7 @@ class LoginCredentials(BaseModel):
 
 class AppConfig(BaseModel):
     """Application configuration."""
-    backend_url: str = "https://api.meetkiwi.ai"
+    backend_url: str = "https://api.runkai.ai"
     api_key: Optional[str] = None
     log_level: str = "INFO"
     theme: str = "dark"

{kiwi_code-0.0.445 → kiwi_code-0.0.446}/src/kiwi_cli/server.py

@@ -11,7 +11,7 @@ The value can be:
 This module intentionally contains *no persistence*. It is runtime-only.
 """
 SERVER_PRESETS: dict[str, dict[str, str]] = {
-    "app": {"ws": "wss://api.meetkiwi.ai", "http": "https://api.meetkiwi.ai"},
+    "app": {"ws": "wss://api.runkai.ai", "http": "https://api.runkai.ai"},
     "dev": {"ws": "wss://dev.api.myautobots.com", "http": "https://dev.api.myautobots.com"},
     "local": {"ws": "ws://localhost:8000", "http": "http://localhost:8000"},
 }