utim-cli 1.44.9__tar.gz → 1.45.2__tar.gz

{utim_cli-1.44.9/utim_cli.egg-info → utim_cli-1.45.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: utim-cli
-Version: 1.44.9
+Version: 1.45.2
 Summary: UTIM – Universal Terminal Intelligence Manager. A powerful agentic AI coding assistant for your terminal.
 License: MIT
 Project-URL: Homepage, https://utim.dev
@@ -30,10 +30,6 @@ Requires-Dist: chardet==4.0.0
 Requires-Dist: typer>=0.9.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: prompt_toolkit>=3.0.0
-Requires-Dist: tree-sitter>=0.21.0
-Requires-Dist: tree-sitter-python>=0.21.0
-Requires-Dist: tree-sitter-javascript>=0.21.0
-Requires-Dist: tree-sitter-typescript>=0.21.0
 Requires-Dist: mcp>=0.1.0
 Requires-Dist: nest-asyncio>=1.5.0
 Requires-Dist: python-multipart>=0.0.7
@@ -45,11 +41,20 @@ Requires-Dist: scrapy-playwright>=0.0.34; extra == "search"
 Requires-Dist: beautifulsoup4>=4.12.0; extra == "search"
 Provides-Extra: images
 Requires-Dist: pillow>=9.0.0; extra == "images"
+Provides-Extra: parsers
+Requires-Dist: tree-sitter>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-python>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-javascript>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-typescript>=0.21.0; extra == "parsers"
 Provides-Extra: full
 Requires-Dist: scrapy>=2.11.0; extra == "full"
 Requires-Dist: scrapy-playwright>=0.0.34; extra == "full"
 Requires-Dist: beautifulsoup4>=4.12.0; extra == "full"
 Requires-Dist: pillow>=9.0.0; extra == "full"
+Requires-Dist: tree-sitter>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-python>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-javascript>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-typescript>=0.21.0; extra == "full"
 Dynamic: license-file
 
 # UTIM CLI Agent: Enterprise Coder Assistant

{utim_cli-1.44.9 → utim_cli-1.45.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "utim-cli"
-version = "1.44.9"
+version = "1.45.2"
 description = "UTIM – Universal Terminal Intelligence Manager. A powerful agentic AI coding assistant for your terminal."
 readme = "README.md"
 requires-python = ">=3.9"
@@ -32,10 +32,6 @@ dependencies = [
     "typer>=0.9.0",
     "rich>=13.0.0",
     "prompt_toolkit>=3.0.0",
-    "tree-sitter>=0.21.0",
-    "tree-sitter-python>=0.21.0",
-    "tree-sitter-javascript>=0.21.0",
-    "tree-sitter-typescript>=0.21.0",
     "mcp>=0.1.0",
     "nest-asyncio>=1.5.0",
     "python-multipart>=0.0.7",
@@ -59,11 +55,21 @@ search = [
 images = [
     "pillow>=9.0.0",
 ]
+parsers = [
+    "tree-sitter>=0.21.0",
+    "tree-sitter-python>=0.21.0",
+    "tree-sitter-javascript>=0.21.0",
+    "tree-sitter-typescript>=0.21.0",
+]
 full = [
     "scrapy>=2.11.0",
     "scrapy-playwright>=0.0.34",
     "beautifulsoup4>=4.12.0",
     "pillow>=9.0.0",
+    "tree-sitter>=0.21.0",
+    "tree-sitter-python>=0.21.0",
+    "tree-sitter-javascript>=0.21.0",
+    "tree-sitter-typescript>=0.21.0",
 ]
 
 [project.scripts]

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli/doctor.py

@@ -22,7 +22,7 @@ def run_diagnostics(console: Console = None):
     console.print("\n  [bold #42bcf5]Core Dependencies:[/bold #42bcf5]")
     core_dependencies = [
         "openai", "requests", "aiohttp", "typer", "rich", 
-        "prompt_toolkit", "tree_sitter", "mcp", "nest_asyncio"
+        "prompt_toolkit", "mcp", "nest_asyncio"
     ]
     for dep in core_dependencies:
         try:
@@ -34,6 +34,7 @@ def run_diagnostics(console: Console = None):
 
     console.print("\n  [bold #42bcf5]Optional Feature Dependencies:[/bold #42bcf5]")
     optional_dependencies = {
+        "tree_sitter": "AST-based Knowledge Graph / dependency mapping",
         "chromadb": "RAG Vector DB feature",
         "sentence_transformers": "Dynamic embedding extraction",
         "scrapy": "Scrapy search crawler",

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli/server/db.py

@@ -396,14 +396,22 @@ class EmailTracking(Base):
 
 # ── Bootstrap ─────────────────────────────────────────────────────────────────
 
-def init_db() -> None:
+def init_db(silent: bool = True) -> None:
     """Create all tables and ensure the schema is up to date."""
     import logging as _logging
     _mig_log = _logging.getLogger("utim.db.migration")
+
+    def _print(msg: str):
+        if not silent:
+            print(msg)
     
+    _print("[DB INIT] init_db() starting...")
     try:
+        _print(f"[DB INIT] Connecting to DB: {engine.url.drivername} (host: {engine.url.host})")
         Base.metadata.create_all(bind=engine)
+        _print("[DB INIT] create_all() completed.")
     except Exception as exc:
+        _print(f"[DB INIT] create_all() failed: {exc}")
         _mig_log.warning(f"Skipping table creation (database might be unavailable): {exc}")
         return
     
@@ -412,70 +420,62 @@ def init_db() -> None:
     from sqlalchemy import text, inspect as sa_inspect
     is_postgres = DATABASE_URL.startswith("postgresql")
 
-    def _run_migration(sql: str, label: str):
+    try:
+        inspector = sa_inspect(engine)
+        existing_tables = inspector.get_table_names()
+    except Exception as inspect_exc:
+        _print(f"[DB INIT] DB inspection failed (might be normal on new DB): {inspect_exc}")
+        existing_tables = []
+
+    def _col_exists(table_name: str, col_name: str) -> bool:
+        if table_name not in existing_tables:
+            return False
+        try:
+            columns = [c["name"] for c in inspector.get_columns(table_name)]
+            return col_name in columns
+        except Exception:
+            return False
+
+    def _run_migration(table_name: str, col_name: str, sql: str, label: str):
+        if _col_exists(table_name, col_name):
+            _print(f"[DB INIT] Column {table_name}.{col_name} already exists. Skipping.")
+            return
+        
+        _print(f"[DB INIT] Running migration: {label}")
         try:
             with engine.begin() as _conn:
+                if is_postgres:
+                    # Never hang startup for lock queues: fail fast if blocked for > 3s
+                    _conn.execute(text("SET lock_timeout = 3000"))
                 _conn.execute(text(sql))
+            _print(f"[DB INIT] Migration OK: {label}")
             _mig_log.info(f"Migration OK: {label}")
         except Exception as _exc:
-            # "already exists" errors are expected and safe to ignore
+            _print(f"[DB INIT] Migration skipped/failed ({label}): {_exc}")
             _mig_log.debug(f"Migration skipped ({label}): {_exc}")
 
     if is_postgres:
-        # PostgreSQL supports IF NOT EXISTS – fully idempotent
-        _run_migration(
-            "ALTER TABLE user_subscriptions ADD COLUMN IF NOT EXISTS refills_processed INTEGER DEFAULT 0 NOT NULL",
-            "add refills_processed"
-        )
-        _run_migration(
-            "ALTER TABLE user_subscriptions ADD COLUMN IF NOT EXISTS last_refill_at TIMESTAMP",
-            "add last_refill_at"
-        )
-        _run_migration(
-            "ALTER TABLE user_subscriptions ADD COLUMN IF NOT EXISTS current_cycle_used DOUBLE PRECISION DEFAULT 0.0 NOT NULL",
-            "add current_cycle_used"
-        )
-        _run_migration(
-            "ALTER TABLE credits ADD COLUMN IF NOT EXISTS bonus_balance DOUBLE PRECISION DEFAULT 0.0 NOT NULL",
-            "add bonus_balance"
-        )
-        _run_migration(
-            "ALTER TABLE credits ADD COLUMN IF NOT EXISTS bonus_limit DOUBLE PRECISION DEFAULT 0.0 NOT NULL",
-            "add bonus_limit"
-        )
-        _run_migration(
-            "ALTER TABLE email_tracking ADD COLUMN IF NOT EXISTS low_quota_email_sent_at TIMESTAMP",
-            "add low_quota_email_sent_at"
-        )
-        _run_migration(
-            "ALTER TABLE email_tracking ADD COLUMN IF NOT EXISTS exhausted_email_sent_at TIMESTAMP",
-            "add exhausted_email_sent_at"
-        )
-        _run_migration(
-            "ALTER TABLE email_tracking ADD COLUMN IF NOT EXISTS bonus_email_sent_at TIMESTAMP",
-            "add bonus_email_sent_at"
-        )
-        _run_migration(
-            "ALTER TABLE users ADD COLUMN IF NOT EXISTS last_project_folder TEXT",
-            "add last_project_folder"
-        )
+        _print("[DB INIT] Postgres detected. Bypassing ALTER TABLE migrations (already applied in production).")
     else:
         # SQLite: no IF NOT EXISTS on ADD COLUMN, swallow duplicate errors
-        _run_migration("ALTER TABLE user_subscriptions ADD COLUMN refills_processed INTEGER DEFAULT 0 NOT NULL", "sqlite add refills_processed")
-        _run_migration("ALTER TABLE user_subscriptions ADD COLUMN last_refill_at TIMESTAMP", "sqlite add last_refill_at")
-        _run_migration("ALTER TABLE user_subscriptions ADD COLUMN current_cycle_used FLOAT DEFAULT 0.0 NOT NULL", "sqlite add current_cycle_used")
-        _run_migration("ALTER TABLE credits ADD COLUMN bonus_balance FLOAT DEFAULT 0.0 NOT NULL", "sqlite add bonus_balance")
-        _run_migration("ALTER TABLE credits ADD COLUMN bonus_limit FLOAT DEFAULT 0.0 NOT NULL", "sqlite add bonus_limit")
-        _run_migration("ALTER TABLE email_tracking ADD COLUMN low_quota_email_sent_at TIMESTAMP", "sqlite add low_quota_email_sent_at")
-        _run_migration("ALTER TABLE email_tracking ADD COLUMN exhausted_email_sent_at TIMESTAMP", "sqlite add exhausted_email_sent_at")
-        _run_migration("ALTER TABLE email_tracking ADD COLUMN bonus_email_sent_at TIMESTAMP", "sqlite add bonus_email_sent_at")
-        _run_migration("ALTER TABLE users ADD COLUMN last_project_folder TEXT", "sqlite add last_project_folder")
+        _run_migration("user_subscriptions", "refills_processed", "ALTER TABLE user_subscriptions ADD COLUMN refills_processed INTEGER DEFAULT 0 NOT NULL", "sqlite add refills_processed")
+        _run_migration("user_subscriptions", "last_refill_at", "ALTER TABLE user_subscriptions ADD COLUMN last_refill_at TIMESTAMP", "sqlite add last_refill_at")
+        _run_migration("user_subscriptions", "current_cycle_used", "ALTER TABLE user_subscriptions ADD COLUMN current_cycle_used FLOAT DEFAULT 0.0 NOT NULL", "sqlite add current_cycle_used")
+        _run_migration("credits", "bonus_balance", "ALTER TABLE credits ADD COLUMN bonus_balance FLOAT DEFAULT 0.0 NOT NULL", "sqlite add bonus_balance")
+        _run_migration("credits", "bonus_limit", "ALTER TABLE credits ADD COLUMN bonus_limit FLOAT DEFAULT 0.0 NOT NULL", "sqlite add bonus_limit")
+        _run_migration("email_tracking", "low_quota_email_sent_at", "ALTER TABLE email_tracking ADD COLUMN low_quota_email_sent_at TIMESTAMP", "sqlite add low_quota_email_sent_at")
+        _run_migration("email_tracking", "exhausted_email_sent_at", "ALTER TABLE email_tracking ADD COLUMN exhausted_email_sent_at TIMESTAMP", "sqlite add exhausted_email_sent_at")
+        _run_migration("email_tracking", "bonus_email_sent_at", "ALTER TABLE email_tracking ADD COLUMN bonus_email_sent_at TIMESTAMP", "sqlite add bonus_email_sent_at")
+        _run_migration("users", "last_project_folder", "ALTER TABLE users ADD COLUMN last_project_folder TEXT", "sqlite add last_project_folder")
         
     # Seed the plans if they don't exist
+    _print("[DB INIT] Seeding plans checking...")
     db = SessionLocal()
     try:
         free_plan = db.query(Plan).filter(Plan.name == "free").first()
+        _print(f"[DB INIT] Free plan query done. Exists: {bool(free_plan)}")
         if not free_plan:
+            _print("[DB INIT] Creating new plans...")
             plans = [
                 Plan(
                     id="free",
@@ -594,6 +594,7 @@ def init_db() -> None:
         pass
     finally:
         db.close()
+        _print("[DB INIT] init_db() completed successfully.")
 
 
 def get_db():

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli/server/router.py

@@ -37,7 +37,7 @@ logger = logging.getLogger("utim.server")
 
 # ── Database init ─────────────────────────────────────────────────────────────
 
-init_db()
+init_db(silent=False)
 
 # ── App ───────────────────────────────────────────────────────────────────────
 

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli/server/routes/completion_routes.py

@@ -178,6 +178,10 @@ async def completions(
 
     client = _get_client()
 
+    # Commit transaction to release all with_for_update locks on UserSubscription
+    # before returning the StreamingResponse, preventing deadlocks in the generator's SessionLocal updates.
+    db.commit()
+
     async def _stream() -> AsyncGenerator[str, None]:
         content_buf = ""
         tool_calls_map: Dict[int, Dict[str, Any]] = {}

utim_cli-1.45.2/utim_cli/server/server.py

@@ -0,0 +1,56 @@
+"""
+UTIM Production Server — Entry Point
+Run locally:   python -m utim_cli.server.server
+Railway runs:  uvicorn utim_cli.server.router:app --host 0.0.0.0 --port $PORT
+"""
+from __future__ import annotations
+
+import sys
+import os
+
+# Ensure parent and grandparent directories are in sys.path to import 'utim_cli'
+_dir = os.path.dirname(os.path.abspath(__file__))
+for _pd in [os.path.dirname(os.path.dirname(_dir)), os.path.dirname(_dir)]:
+    if _pd and _pd not in sys.path:
+        sys.path.insert(0, _pd)
+
+import uvicorn
+
+
+def main():
+    host = os.environ.get("HOST", "0.0.0.0")
+    port = int(os.environ.get("PORT", 8080))
+
+    # Warn about missing env vars but DO NOT exit — server must start
+    required = {
+        "DATABASE_URL": os.environ.get("DATABASE_URL"),
+        "OPENROUTER_API_KEY": os.environ.get("OPENROUTER_API_KEY"),
+        "UTIM_MASTER_KEY": os.environ.get("UTIM_MASTER_KEY"),
+    }
+    missing = [k for k, v in required.items() if not v]
+    if missing:
+        print(f"[WARNING] Missing env vars: {', '.join(missing)} — some features will be degraded.")
+        print("          Set these in Railway → Service → Variables")
+
+    print(f"🚀  UTIM Production Server")
+    print(f"    http://{host}:{port}")
+    print(f"    Docs:   http://{host}:{port}/docs")
+
+    try:
+        uvicorn.run(
+            "utim_cli.server.router:app",
+            host=host,
+            port=port,
+            reload=False,
+            workers=1,
+            # Do NOT pass log_config=None — keep default logging so errors are visible
+        )
+    except Exception as exc:
+        print(f"[FATAL] uvicorn failed to start: {exc}", file=sys.stderr)
+        import traceback
+        traceback.print_exc()
+        sys.exit(1)
+
+
+if __name__ == "__main__":
+    main()

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli/utim.py

@@ -585,7 +585,7 @@ def _print_animated_banner():
         # Speed up for spaces to make it feel more like typing
         time.sleep(0.002 if char == " " else 0.005)
             
-    console.print(f"\n [bold white]U Think I Make[/bold white] [dim]v1.44.9[/dim]")
+    console.print(f"\n [bold white]U Think I Make[/bold white] [dim]v1.45.2[/dim]")
     user_email = config.email
     user_type = "UTIM Community"
     api_key = config.get("api_key")
@@ -2691,7 +2691,7 @@ def check_and_update_background():
                 return
 
             latest_ver = resp.json().get("version", "")
-            current_ver = "1.44.9"
+            current_ver = "1.45.2"
 
             def parse_ver(v):
                 return [int(x) for x in v.split(".") if x.isdigit()]
@@ -2887,7 +2887,7 @@ app = typer.Typer()
 def version_callback(value: bool):
     if value:
         import typer
-        print("U Think I Make v1.44.9")
+        print("U Think I Make v1.45.2")
         raise typer.Exit()
 
 @app.callback(invoke_without_command=True)

{utim_cli-1.44.9 → utim_cli-1.45.2/utim_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: utim-cli
-Version: 1.44.9
+Version: 1.45.2
 Summary: UTIM – Universal Terminal Intelligence Manager. A powerful agentic AI coding assistant for your terminal.
 License: MIT
 Project-URL: Homepage, https://utim.dev
@@ -30,10 +30,6 @@ Requires-Dist: chardet==4.0.0
 Requires-Dist: typer>=0.9.0
 Requires-Dist: rich>=13.0.0
 Requires-Dist: prompt_toolkit>=3.0.0
-Requires-Dist: tree-sitter>=0.21.0
-Requires-Dist: tree-sitter-python>=0.21.0
-Requires-Dist: tree-sitter-javascript>=0.21.0
-Requires-Dist: tree-sitter-typescript>=0.21.0
 Requires-Dist: mcp>=0.1.0
 Requires-Dist: nest-asyncio>=1.5.0
 Requires-Dist: python-multipart>=0.0.7
@@ -45,11 +41,20 @@ Requires-Dist: scrapy-playwright>=0.0.34; extra == "search"
 Requires-Dist: beautifulsoup4>=4.12.0; extra == "search"
 Provides-Extra: images
 Requires-Dist: pillow>=9.0.0; extra == "images"
+Provides-Extra: parsers
+Requires-Dist: tree-sitter>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-python>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-javascript>=0.21.0; extra == "parsers"
+Requires-Dist: tree-sitter-typescript>=0.21.0; extra == "parsers"
 Provides-Extra: full
 Requires-Dist: scrapy>=2.11.0; extra == "full"
 Requires-Dist: scrapy-playwright>=0.0.34; extra == "full"
 Requires-Dist: beautifulsoup4>=4.12.0; extra == "full"
 Requires-Dist: pillow>=9.0.0; extra == "full"
+Requires-Dist: tree-sitter>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-python>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-javascript>=0.21.0; extra == "full"
+Requires-Dist: tree-sitter-typescript>=0.21.0; extra == "full"
 Dynamic: license-file
 
 # UTIM CLI Agent: Enterprise Coder Assistant

{utim_cli-1.44.9 → utim_cli-1.45.2}/utim_cli.egg-info/requires.txt

@@ -8,10 +8,6 @@ chardet==4.0.0
 typer>=0.9.0
 rich>=13.0.0
 prompt_toolkit>=3.0.0
-tree-sitter>=0.21.0
-tree-sitter-python>=0.21.0
-tree-sitter-javascript>=0.21.0
-tree-sitter-typescript>=0.21.0
 mcp>=0.1.0
 nest-asyncio>=1.5.0
 python-multipart>=0.0.7
@@ -23,10 +19,20 @@ scrapy>=2.11.0
 scrapy-playwright>=0.0.34
 beautifulsoup4>=4.12.0
 pillow>=9.0.0
+tree-sitter>=0.21.0
+tree-sitter-python>=0.21.0
+tree-sitter-javascript>=0.21.0
+tree-sitter-typescript>=0.21.0
 
 [images]
 pillow>=9.0.0
 
+[parsers]
+tree-sitter>=0.21.0
+tree-sitter-python>=0.21.0
+tree-sitter-javascript>=0.21.0
+tree-sitter-typescript>=0.21.0
+
 [search]
 scrapy>=2.11.0
 scrapy-playwright>=0.0.34

utim_cli-1.44.9/utim_cli/server/server.py

@@ -1,56 +0,0 @@
-"""
-UTIM Production Server — Entry Point
-Run locally:   python -m utim_cli.server.server
-Railway runs:  uvicorn utim_cli.server.router:app --host 0.0.0.0 --port $PORT
-"""
-from __future__ import annotations
-
-import sys
-import os
-
-# Ensure parent and grandparent directories are in sys.path to import 'utim_cli'
-_dir = os.path.dirname(os.path.abspath(__file__))
-for _pd in [os.path.dirname(os.path.dirname(_dir)), os.path.dirname(_dir)]:
-    if _pd and _pd not in sys.path:
-        sys.path.insert(0, _pd)
-
-import argparse
-import uvicorn
-
-
-def main():
-    parser = argparse.ArgumentParser(description="Start the UTIM Production Server")
-    parser.add_argument("--host", default="0.0.0.0")
-    parser.add_argument("--port", type=int, default=int(os.environ.get("PORT", 8000)))
-    parser.add_argument("--reload", action="store_true")
-    parser.add_argument("--workers", type=int, default=1)
-    args = parser.parse_args()
-
-    required = {
-        "DATABASE_URL": os.environ.get("DATABASE_URL"),
-        "OPENROUTER_API_KEY": os.environ.get("OPENROUTER_API_KEY"),
-        "UTIM_MASTER_KEY": os.environ.get("UTIM_MASTER_KEY"),
-    }
-    missing = [k for k, v in required.items() if not v]
-    if missing:
-        print(f"[ERROR] Missing required env vars: {', '.join(missing)}")
-        print("        Set these in Railway → Service → Variables")
-        sys.exit(1)
-
-    print(f"🚀  UTIM Production Server")
-    print(f"    http://{args.host}:{args.port}")
-    print(f"    Docs:   http://{args.host}:{args.port}/docs")
-
-    uvicorn.run(
-        "utim_cli.server.router:app",
-        host=args.host,
-        port=args.port,
-        reload=args.reload,
-        workers=args.workers if not args.reload else 1,
-        log_config=None,   # we handle logging ourselves
-        access_log=False,  # our middleware does structured access logs
-    )
-
-
-if __name__ == "__main__":
-    main()