lkr-dev-cli 0.0.34__tar.gz → 0.0.36__tar.gz

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/Makefile

@@ -17,10 +17,10 @@ codemode-start:
 	@echo "Add this to your mcpServers config:"
 	@echo "{"
 	@echo "  \"mcpServers\": {"
-	@echo "    \"lkr-codemode\": {"
+	@echo "    \"lkr_codemode\": {"
 	@echo "      \"command\": \"uvx\","
-	@echo "      \"args\": [\"--from\", \"lkr-dev-cli[codemode]\", \"lkr\", \"code-mode\", \"run\"]"
+	@echo "      \"args\": [\"-q\", \"--from\", \"lkr-dev-cli[codemode]\", \"lkr\", \"code-mode\", \"run\"]"
 	@echo "    }"
 	@echo "  }"
 	@echo "}"
-	uv run lkr code-mode run
+	uv run -q lkr code-mode run

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lkr-dev-cli
-Version: 0.0.34
+Version: 0.0.36
 Summary: lkr: a command line interface for looker
 Author: bwebs
 License-Expression: MIT
@@ -109,9 +109,9 @@ Go to the Looker API Explorer for Register OAuth App (https://your.looker.instan
 - This only needs to be done once per instance
 
 
-## Code-Mode
+## Code Mode
 
-Execute Python code safely with full Looker SDK coverage within a secure sandbox environment. Constructed as an MCP tool, it dynamically inspects the Looker SDK for all public methods and injects them into the Monty sandbox safely. For detailed options, safe primitives transformations, and PKCE configurations, view the full [Code-Mode Docs](./codemode.md).
+Execute Python code safely with full Looker SDK coverage within a secure sandbox environment. Constructed as an MCP tool, it dynamically inspects the Looker SDK for all public methods and injects them into the Monty sandbox safely. For detailed options, safe primitives transformations, and PKCE configurations, view the full [Code Mode Docs](./codemode.md).
 
 ## MCP
 Built into the `lkr` is an MCP server. Right now its tools are based on helping you work within an IDE. To use it a tool like [Cursor](https://www.cursor.com/), add this to your mcp.json

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/README.md

@@ -73,9 +73,9 @@ Go to the Looker API Explorer for Register OAuth App (https://your.looker.instan
 - This only needs to be done once per instance
 
 
-## Code-Mode
+## Code Mode
 
-Execute Python code safely with full Looker SDK coverage within a secure sandbox environment. Constructed as an MCP tool, it dynamically inspects the Looker SDK for all public methods and injects them into the Monty sandbox safely. For detailed options, safe primitives transformations, and PKCE configurations, view the full [Code-Mode Docs](./codemode.md).
+Execute Python code safely with full Looker SDK coverage within a secure sandbox environment. Constructed as an MCP tool, it dynamically inspects the Looker SDK for all public methods and injects them into the Monty sandbox safely. For detailed options, safe primitives transformations, and PKCE configurations, view the full [Code Mode Docs](./codemode.md).
 
 ## MCP
 Built into the `lkr` is an MCP server. Right now its tools are based on helping you work within an IDE. To use it a tool like [Cursor](https://www.cursor.com/), add this to your mcp.json

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/codemode.md

@@ -1,4 +1,4 @@
-# Looker Code-Mode MCP Server
+# Looker Code Mode MCP Server
 
 `lkr code-mode` allows you to invoke a Python-based Model Context Protocol (MCP) server. It offers an AI agent the unique capacity to batched-execute Python commands securely within the Monty sandbox against your active Looker instance!
 
@@ -11,7 +11,7 @@ The tool instantiates Looker SDK natively, searches all bound methods, and passe
 ### Key Features:
 - **100% Tool Coverage:** Accesses all Looker SDK public operations smoothly without token limits.
 - **Recursive Translation:** Complex Looker models like User, Folder, Dashboard get string-converted into dictionaries immediately before ingesting them into Monty.
-- **Automatic PKCE Restarter:** Caught an invalid token? Code-Mode immediately catches `InvalidRefreshTokenError` and safely opens up your PKCE authentication browser automatically.
+- **Automatic PKCE Restarter:** Caught an invalid token? Code Mode immediately catches `InvalidRefreshTokenError` and safely opens up your PKCE authentication browser automatically.
 - **Extremely Secure:** Monty interpreter ensures isolated sandbox processing. No local filesystem accesses are exposed.
 
 ## Continuous Usage
@@ -19,7 +19,7 @@ The tool instantiates Looker SDK natively, searches all bound methods, and passe
 ### 1. Starting the Server
 To immediately trigger the stdio listener, use:
 ```bash
-uvx --from lkr-dev-cli[codemode] lkr code-mode run
+uvx -q --from lkr-dev-cli[codemode] lkr code-mode run
 ```
 
 ### 2. Client Configuration
@@ -28,9 +28,9 @@ To hook this server into Cursor or Claude Desktop natively over stdio, append th
 ```json
 {
   "mcpServers": {
-    "looker-codemode": {
+    "lkr_codemode": {
       "command": "uvx",
-      "args": ["--from", "lkr-dev-cli[codemode]", "lkr", "code-mode", "run"],
+      "args": ["-q", "--from", "lkr-dev-cli[codemode]", "lkr", "code-mode", "run"],
       "env": {
         "LOOKERSDK_BASE_URL": "https://your.looker.instance",
         "LOOKERSDK_CLIENT_ID": "your-client-id",
@@ -44,7 +44,7 @@ To hook this server into Cursor or Claude Desktop natively over stdio, append th
 ### 3. Visual Inspector
 To check things out on a web panel:
 ```bash
-npx @modelcontextprotocol/inspector uvx --from lkr-dev-cli[codemode] lkr code-mode run
+npx @modelcontextprotocol/inspector uvx -q --from lkr-dev-cli[codemode] lkr code-mode run
 ```
 
 ```

lkr_dev_cli-0.0.36/lkr/codemode/LOCAL.md

@@ -0,0 +1,56 @@
+# Local Development Configuration for lkr:codemode
+
+To run this MCP server locally from the repository (for development or testing), you can configure your MCP client (like Claude Desktop or Cursor) to use `uv run` with the `--directory` flag pointing to this project.
+
+### Claude Desktop Configuration
+
+Add the following to your `claude_desktop_config.json` (usually located at `~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
+
+```json
+{
+  "mcpServers": {
+    "lkr_codemode_dev": {
+      "command": "/opt/homebrew/bin/uv",
+      "args": [
+        "-q",
+        "run",
+        "--directory",
+        "/path/to/lkr/cli",
+        "lkr",
+        "code-mode",
+        "run"
+      ],
+      "env": {
+        "LOOKERSDK_BASE_URL": "https://your-instance.cloud.looker.com",
+        "LOOKERSDK_CLIENT_ID": "your_client_id",
+        "LOOKERSDK_CLIENT_SECRET": "your_client_secret"
+      }
+    }
+  }
+}
+```
+
+> [!WARNING]
+> Do NOT place the directory path directly after `run` without the `--directory` flag (e.g., `["run", "/path/to/project", ...]`). `uv` will attempt to execute the directory as a command, resulting in a `Permission denied (os error 13)` error.
+
+> [!TIP]
+> **macOS GUI Apps & PATH:** Claude Desktop and Cursor do not inherit your terminal's `PATH` environment variable. If you simply use `"uv"` as the command, the app may fail to start the server because it cannot find `uv`. Using the absolute path (e.g., `/opt/homebrew/bin/uv` or `~/.cargo/bin/uv`) ensures the IDE can launch the process. Find out your path by running `which uv` in your terminal.
+
+### Cursor Integration
+
+To add this local dev server to Cursor:
+
+1. Open **Settings** (`Cmd + ,`).
+2. Go to **Features** -> **MCP**.
+3. Click **+ Add New MCP Server**.
+4. **Name**: `Looker Codemode (Local)`
+5. **Type**: `command`
+6. **Command**: `uv run -q --directory /Users/bryanweber/projects/lkr/cli lkr code-mode run`
+
+### Testing Interactively via MCP Inspector
+
+You can test the server and its tools interactively in your browser by running the MCP inspector from your terminal:
+
+```bash
+npx @modelcontextprotocol/inspector uv run -q --directory /Users/bryanweber/projects/lkr/cli lkr code-mode run
+```

lkr_dev_cli-0.0.36/lkr/codemode/main.py

@@ -0,0 +1,200 @@
+import inspect
+import io
+import json
+from contextlib import redirect_stdout
+from typing import Optional
+
+import typer
+import pydantic_monty
+from mcp.server.fastmcp import FastMCP
+
+from lkr.auth_service import get_auth, is_auth_expired
+from lkr.classes import LkrCtxObj
+from lkr.logger import logger
+
+__all__ = ["group"]
+
+mcp = FastMCP("lkr:codemode")
+group = typer.Typer()
+
+
+
+def get_mcp_sdk(ctx: LkrCtxObj):
+    sdk = get_auth(ctx).get_current_sdk(prompt_refresh_invalid_token=True)
+    sdk.auth.settings.agent_tag += "-codemode"
+    return sdk
+
+
+
+def to_primitive(obj):
+    seen = set()
+
+    def _to_primitive(o):
+        if isinstance(o, (str, int, float, bool, type(None))):
+            return o
+        
+        obj_id = id(o)
+        if obj_id in seen:
+            return f"<Circular reference to {type(o).__name__}>"
+        seen.add(obj_id)
+        
+        try:
+            if isinstance(o, list):
+                return [_to_primitive(item) for item in o]
+            elif isinstance(o, dict):
+                return {k: _to_primitive(v) for k, v in o.items()}
+            else:
+                try:
+                    return _to_primitive(vars(o))
+                except TypeError:
+                    return str(o)
+                except Exception:
+                    return str(o)
+        finally:
+            seen.remove(obj_id)
+
+    return _to_primitive(obj)
+
+
+
+@mcp.tool()
+def run_python_code(code: str) -> str:
+    """
+    Execute Python code safely with access to all Looker SDK methods as global functions.
+    Capture the result. 
+    
+    AGENT HINTS:
+    - Use `dir()` and `help('method_name')` to discover available SDK methods.
+    - Do not instantiate an SDK; use global functions directly (e.g. `me()`).
+    - Returned Looker models are primitive dictionaries (use `obj["id"]`, not `obj.id`).
+    - Return your output (avoid using print() as it may pollute the stdio stream).
+    - Recursion: Use `folder_children(id)` to traverse nested folders.
+    """
+    try:
+        ctx = LkrCtxObj(force_oauth=False)
+        sdk = get_mcp_sdk(ctx)
+        
+        external_funcs = {}
+        for name, method in inspect.getmembers(sdk, predicate=inspect.ismethod):
+            if not name.startswith('_'):
+                # Wrap in a lambda to recursively convert output to primitives
+                def make_wrapper(m):
+                    def wrapper(*args, **kwargs):
+                        res = m(*args, **kwargs)
+                        return to_primitive(res)
+                    return wrapper
+                external_funcs[name] = make_wrapper(method)
+
+        # Provide helper functions for the LLM to explore the SDK
+        external_funcs['dir'] = lambda: list(external_funcs.keys())
+        
+        def _help(name: str) -> str:
+            if name in external_funcs:
+                if hasattr(sdk, name):
+                    return getattr(sdk, name).__doc__ or "No docstring available."
+                return f"{name} is a built-in helper function."
+            return f"Function '{name}' not found."
+        external_funcs['help'] = _help
+
+        m = pydantic_monty.Monty(code)
+        
+        # Redirect stdout to capture any print() statements
+        # and prevent them from corrupting the JSON-RPC stream
+        f = io.StringIO()
+        with redirect_stdout(f):
+            result = m.run(external_functions=external_funcs)
+        
+        printed_output = f.getvalue()
+        
+        # m.run() returns the evaluated result of the last expression (which is already a primitive)
+        try:
+            # Use JSON for nice formatting if it's a dict/list
+            if result is not None:
+                if isinstance(result, str):
+                    output = result
+                else:
+                    output = json.dumps(result, indent=2, default=str)
+            else:
+                output = ""
+        except Exception:
+            output = repr(result)
+            
+        if printed_output:
+            return f"PRINTED OUTPUT:\n{printed_output}\nRESULT:\n{output}"
+        return output
+    except Exception as e:
+        logger.error(f"Error executing Monty: {e}")
+        if is_auth_expired(e):
+            return "Error: Your Looker OAuth session has expired. Please run 'lkr auth login' to re-authenticate."
+        return f"Error: {str(e)}"
+
+
+@mcp.resource("looker://agent-hints")
+def get_agent_hints() -> str:
+    """Crucial hints and rules for AI agents writing Python for the Looker SDK."""
+    return """
+1. **Global Functions**: All Looker SDK methods are global. Use `me()`, not `sdk.me()`.
+2. **Dict Access**: Return values are dictionaries, not objects. Use `user["id"]`, not `user.id`.
+3. **Discovery**: Use `dir()` and `help('method')` to explore the SDK.
+4. **No Imports**: Do not `import looker_sdk`.
+5. **Output**: Return your results instead of using `print()`.
+6. **Efficiency**: Always use the `fields` parameter (e.g., `all_dashboards(fields="id,title")`) when listing many objects to prevent timeouts.
+7. **Nested Folders**: Use `folder_children(id)` to get sub-folders.
+"""
+
+
+@mcp.prompt("explore_looker_sdk")
+def explore_looker_sdk() -> str:
+    """Provide examples for how to explore the Looker SDK in code mode."""
+    return '''
+To explore the Looker SDK, you can use the injected `dir()` and `help()` helpers.
+Do not use print() as it may corrupt the MCP output stream; always return the result.
+
+Example 1: Find all dashboard-related methods
+```python
+return [m for m in dir() if 'dashboard' in m.lower()]
+```
+
+Example 2: Get the description of a specific method
+```python
+return help('search_dashboards')
+```
+'''
+
+
+@mcp.prompt("list_personal_dashboards")
+def list_personal_dashboards() -> str:
+    """Provide an example of how to recursively list dashboards in a user's personal folder."""
+    return '''
+Here is a robust example of how to traverse the folder hierarchy using the Looker SDK in code mode:
+
+```python
+def get_all_items(folder_id):
+    f = folder(folder_id)
+    items = {
+        "dashboards": f.get("dashboards", []),
+        "looks": f.get("looks", [])
+    }
+    
+    for child in folder_children(folder_id):
+        child_items = get_all_items(child["id"])
+        items["dashboards"].extend(child_items["dashboards"])
+        items["looks"].extend(child_items["looks"])
+        
+    return items
+
+me_data = me()
+return get_all_items(me_data["personal_folder_id"])
+```
+'''
+
+
+@group.command(name="run")
+def run(
+    ctx: typer.Context,
+    debug: bool = typer.Option(False, help="Debug mode"),
+):
+    mcp.run()
+
+if __name__ == "__main__":
+    mcp.run("sse")

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/lkr/logger.py

@@ -36,7 +36,7 @@ theme = Theme(
 ) if RICH_AVAILABLE else None
 
 # Create a console for logging
-console = Console(theme=theme) if RICH_AVAILABLE else None
+console = Console(theme=theme, stderr=True) if RICH_AVAILABLE else None
 
 # Configure the logging handler
 handler = RichHandler(

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/lkr/mcp/main.py

@@ -1,4 +1,5 @@
 # server.py
+import sys
 import threading
 from datetime import datetime
 from pathlib import Path
@@ -549,6 +550,7 @@ def run(
         set_log_level(LogLevel.DEBUG)
     else:
         set_log_level(LogLevel.ERROR)
+    sys.stdout = sys.stderr
     mcp.run()
 
 

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "lkr-dev-cli"
-version = "0.0.34"
+version = "0.0.36"
 description = "lkr: a command line interface for looker"
 readme = "README.md"
 license = "MIT"

{lkr_dev_cli-0.0.34 → lkr_dev_cli-0.0.36}/tests/test_codemode2.py

@@ -1,5 +1,7 @@
 from lkr.codemode.main import run_python_code
 
+# Find me all my dashboards and looks within all my personal folder and nested folders
+
 code = """
 me_obj = me()
 personal_folder_id = me_obj["personal_folder_id"]

lkr_dev_cli-0.0.34/lkr/codemode/main.py

@@ -1,104 +0,0 @@
-import inspect
-from typing import Optional
-
-import typer
-import pydantic_monty
-from mcp.server.fastmcp import FastMCP
-
-from lkr.auth_service import get_auth, is_auth_expired
-from lkr.classes import LkrCtxObj
-from lkr.logger import logger
-
-__all__ = ["group"]
-
-mcp = FastMCP("lkr:codemode")
-group = typer.Typer()
-
-
-
-def get_mcp_sdk(ctx: LkrCtxObj):
-    sdk = get_auth(ctx).get_current_sdk(prompt_refresh_invalid_token=True)
-    sdk.auth.settings.agent_tag += "-codemode"
-    return sdk
-
-
-
-def to_primitive(obj):
-    seen = set()
-
-    def _to_primitive(o):
-        if isinstance(o, (str, int, float, bool, type(None))):
-            return o
-        
-        obj_id = id(o)
-        if obj_id in seen:
-            return f"<Circular reference to {type(o).__name__}>"
-        seen.add(obj_id)
-        
-        try:
-            if isinstance(o, list):
-                return [_to_primitive(item) for item in o]
-            elif isinstance(o, dict):
-                return {k: _to_primitive(v) for k, v in o.items()}
-            else:
-                try:
-                    return _to_primitive(vars(o))
-                except TypeError:
-                    return str(o)
-                except Exception:
-                    return str(o)
-        finally:
-            seen.remove(obj_id)
-
-    return _to_primitive(obj)
-
-
-
-@mcp.tool()
-def run_python_code(code: str) -> str:
-    """
-    Execute Python code safely with access to all Looker SDK methods as global functions.
-    Capture the result and any print outputs.
-    """
-    try:
-        ctx = LkrCtxObj(force_oauth=False)
-        sdk = get_mcp_sdk(ctx)
-        
-        external_funcs = {}
-        for name, method in inspect.getmembers(sdk, predicate=inspect.ismethod):
-            if not name.startswith('_'):
-                # Wrap in a lambda to recursively convert output to primitives
-                def make_wrapper(m):
-                    def wrapper(*args, **kwargs):
-                        res = m(*args, **kwargs)
-                        return to_primitive(res)
-                    return wrapper
-                external_funcs[name] = make_wrapper(method)
-
-        m = pydantic_monty.Monty(code)
-        result = m.run(external_functions=external_funcs)
-        
-        # Monty run returns a MontyComplete or None
-        output = str(getattr(result, "output", "")) if result is not None else ""
-        
-        # Try to append captured stdout if available on the object
-        stdout = getattr(result, "stdout", None) if result is not None else None
-        if stdout:
-            return f"Output:\n{output}\nStdout:\n{stdout}"
-        return output
-    except Exception as e:
-        logger.error(f"Error executing Monty: {e}")
-        if is_auth_expired(e):
-            return "Error: Your Looker OAuth session has expired. Please run 'lkr auth login' to re-authenticate."
-        return f"Error: {str(e)}"
-
-
-@group.command(name="run")
-def run(
-    ctx: typer.Context,
-    debug: bool = typer.Option(False, help="Debug mode"),
-):
-    mcp.run()
-
-if __name__ == "__main__":
-    mcp.run("sse")