kiwi-code 0.0.447__tar.gz → 0.0.448__tar.gz

{kiwi_code-0.0.447 → kiwi_code-0.0.448}/PKG-INFO

@@ -1,8 +1,8 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.5
 Name: kiwi-code
-Version: 0.0.447
+Version: 0.0.448
 Summary: A textual-based terminal user interface application
-Project-URL: Homepage, https://meetkiwi.ai
+Project-URL: Homepage, https://kiwicode.ai
 Author-email: Anurag <anurag@meetkiwi.co>
 License: Proprietary
 Keywords: cli,terminal,textual,tui
@@ -14,11 +14,12 @@ 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: codeglass>=0.1.2
 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.12
 Requires-Dist: setproctitle>=1.3.0
 Requires-Dist: textual-dev>=1.8.0
 Requires-Dist: textual>=8.1.1

{kiwi_code-0.0.447 → kiwi_code-0.0.448}/pyproject.toml

@@ -1,11 +1,11 @@
 [project]
 name = "kiwi-code"
-version = "0.0.447"
+version = "0.0.448"
 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.12",
     "loguru>=0.7.3",
     "pydantic>=2.12.5",
     "textual>=8.1.1",
@@ -16,6 +16,7 @@ dependencies = [
     "psutil>=5.9.0",
     "setproctitle>=1.3.0",
     "wonderwords>=2.2.0",
+    "codeglass>=0.1.2",
 ]
 
 authors = [
@@ -34,7 +35,8 @@ classifiers = [
 ]
 
 [project.urls]
-Homepage = "https://meetkiwi.ai"
+Homepage = "https://kiwicode.ai"
+# Repository = "https://github.com/runkailabs/kiwi-code"
 
 [project.scripts]
 kiwi = "kiwi_tui.main:main"
@@ -54,8 +56,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.448/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.meetkiwi.ai/v1/terminal/ws/connect",
+        help="Full ws connect URI (default: wss://api.meetkiwi.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.448/src/common/banner.py

@@ -0,0 +1,57 @@
+"""Shared Kiwi Code banner assets used by kiwi-runtime and kiwi-tui."""
+
+# Block-character chevron-and-dot logo (6 lines, 11 chars wide — same height as
+# the KIWI CODE wordmark). Reads as ">." — half-blocks give the diagonal a
+# half-row step so the stroke stays continuous instead of reading as a staircase.
+LOGO = [
+    "▀██▄       ",
+    "  ▀██▄     ",
+    "    ▀██▄   ",
+    "    ▄██▀   ",
+    "  ▄██▀     ",
+    "▄██▀    █▌ ",
+]
+
+# "KIWI CODE" 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 = 11
+GAP = "  "
+TEXT_START = 0
+TAGLINE_ROW = TEXT_START + len(TEXT) + 1
+
+
+def compose_banner_lines(tagline: str | None = None) -> list[str]:
+    """Compose logo + KIWI CODE text side by side, returning a list of lines."""
+    effective_tagline = tagline if tagline is not None else TAGLINE
+    lines = []
+    # The logo is shorter than logo+tagline, so keep going past it with blanks.
+    for i in range(max(len(LOGO), TAGLINE_ROW + 1)):
+        padded = (LOGO[i] if i < len(LOGO) else "").ljust(LOGO_W)
+        text_idx = i - TEXT_START
+        if 0 <= text_idx < len(TEXT):
+            lines.append(f"{padded}{GAP}{TEXT[text_idx]}")
+        elif i == TAGLINE_ROW:
+            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.447 → kiwi_code-0.0.448}/src/kiwi_cli/__init__.py

@@ -1,4 +1,3 @@
 """Kiwi CLI - command-line interface and shared infrastructure modules."""
 
-
-__version__ = "0.0.447"
+__version__ = "0.0.448"

{kiwi_code-0.0.447 → kiwi_code-0.0.448}/src/kiwi_cli/checkpoints.py

@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 import json
 import os
 from dataclasses import dataclass

{kiwi_code-0.0.447 → kiwi_code-0.0.448}/src/kiwi_cli/cli.py

@@ -3,14 +3,14 @@ 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
 from . import commands
 from . import __version__ as KIWI_CLI_VERSION
 
-app = typer.Typer(name="kiwi", help="Kiwi — interact with the Kiwi AI platform.")
+app = typer.Typer(name="kiwi", help="Kiwi — interact with the Kiwi Code platform.")
 
 actions_app = typer.Typer(name="actions", help="Manage actions.")
 action_runs_app = typer.Typer(name="runs", help="Manage action runs (results).")

{kiwi_code-0.0.447 → kiwi_code-0.0.448}/src/kiwi_cli/client.py

@@ -6,22 +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 (
-    BodyAsyncRunActionV1ActionsIdAsyncRunPostInput,
-)
+from pykiwiai.models import DataBlock, Metadata
 from .models import AuthTokens, LoginCredentials
 
 
@@ -253,8 +251,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 = []
@@ -333,18 +331,17 @@ class AutobotsClientWrapper:
             if metadata:
                 default_metadata.update(metadata)
 
-            input_data = BodyAsyncRunActionV1ActionsIdAsyncRunPostInput()
-            input_data.additional_properties = {
-                "text": user_input,
-                "urls": urls or [],
-                "metadata": default_metadata
-            }
+            input_data = DataBlock(
+                text=user_input,
+                urls=urls or [],
+                metadata=Metadata.from_dict(default_metadata),
+            )
 
             body = BodyAsyncRunActionV1ActionsIdAsyncRunPost(input_=input_data)
             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 +461,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.447 → kiwi_code-0.0.448}/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
@@ -167,12 +179,6 @@ Files:
   /files                Show pending file uploads
   /clear-files          Clear pending file uploads
 
-Metadata (persistent across runs):
-  /metadata             Show current metadata
-  /metadata set <k> <v> Set a metadata field
-  /metadata remove <k>  Remove a metadata field
-  /metadata clear       Clear all metadata
-
 Runtime:
   /runtime status       Show runtime status
   /runtime start        Start runtime