lkr-dev-cli 0.0.24__tar.gz → 0.0.26__tar.gz

lkr_dev_cli-0.0.26/.vscode/launch.json

@@ -0,0 +1,41 @@
+{
+    // Use IntelliSense to learn about possible attributes.
+    // Hover to view descriptions of existing attributes.
+    // For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
+    "version": "0.2.0",
+    "configurations": [
+        {
+            "name": "lkr auth login",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/.venv/bin/lkr",
+            "args": ["auth", "login"],
+            "justMyCode": false,
+            "env": {
+                "PYTHONPATH": "${workspaceFolder}"
+            }
+        },
+        {
+            "name": "lkr mcp run",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/.venv/bin/lkr",
+            "args": ["mcp", "run", "--debug"],
+            "justMyCode": false,
+            "env": {
+                "PYTHONPATH": "${workspaceFolder}"
+            }
+        },
+        {
+            "name": "lkr tools access token updater",
+            "type": "debugpy",
+            "request": "launch",
+            "program": "${workspaceFolder}/.venv/bin/lkr",
+            "args": ["tools", "access-token-updater"],
+            "justMyCode": false,
+            "env": {
+                "PYTHONPATH": "${workspaceFolder}"
+            }
+        }
+    ]
+}

{lkr_dev_cli-0.0.24 → lkr_dev_cli-0.0.26}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lkr-dev-cli
-Version: 0.0.24
+Version: 0.0.26
 Summary: lkr: a command line interface for looker
 Author: bwebs
 License-Expression: MIT
@@ -9,8 +9,8 @@ Requires-Python: >=3.12
 Requires-Dist: cryptography>=42.0.0
 Requires-Dist: duckdb>=1.2.2
 Requires-Dist: fastapi>=0.115.12
-Requires-Dist: fastmcp>=2.3.5
 Requires-Dist: looker-sdk>=25.4.0
+Requires-Dist: mcp[cli]>=1.9.2
 Requires-Dist: pydantic>=2.11.4
 Requires-Dist: pydash>=8.0.5
 Requires-Dist: questionary>=2.1.0
@@ -178,26 +178,33 @@ For example:
 
 ### Cloud Run + GCP Health Check example
 
-One of the simplest ways to launch the health check is the `lkr-cli` public docker image, Cloud Run, and the GCP health check service. Here's an example; make sure to change your region, models, user_attributes, and external_user_id.
+One of the simplest ways to launch the health check is the `lkr-cli` public docker image, Cloud Run, and the GCP health check service. Here's an example; make sure to change your region and project. HEALTH_URL is an example of how to structure the query parameters for the health check.
 
 ```bash
 export REGION=<your region>
 export PROJECT=<your project id>
 
-export HEALTH_URL="/health?dashboard_id=1&external_user_id=embed-user-abc&models=thelook&user_attributes={\"store_id\":\"1\"}"
+export HEALTH_URL="/health?dashboard_id=1&external_user_id=observability-embed-user&models=thelook&user_attributes={\"store_id\":\"1\"}"
 
 gcloud run deploy lkr-observability \
   --image us-central1-docker.pkg.dev/lkr-dev-production/lkr-cli/cli:latest \
   --command lkr \
   --args observability,embed \
   --platform managed \
-  --region $REGION \ 
+  --region $REGION \
   --project $PROJECT \
   --cpu 2 \
   --memory 4Gi \
-  --liveness-probe httpGet.path=$HEALTH_URL,timeoutSeconds=20,periodSeconds=20 \
   --set-env-vars LOOKERSDK_CLIENT_ID=<your client id>,LOOKERSDK_CLIENT_SECRET=<your client secret>,LOOKERSDK_BASE_URL=<your instance url> 
 
+gcloud monitoring uptime create lkr-observability-health-check \
+  --protocol https \
+  --project $PROJECT \
+  --resource-type="cloud-run-revision" \
+  --resource-labels="project_id=${PROJECT},service_name=lkr-observability,location=${REGION}" \
+  --path="${HEALTH_URL}" \
+  --period="15" \
+  --timeout="60"
 
 ```
 

{lkr_dev_cli-0.0.24 → lkr_dev_cli-0.0.26}/README.md

@@ -156,26 +156,33 @@ For example:
 
 ### Cloud Run + GCP Health Check example
 
-One of the simplest ways to launch the health check is the `lkr-cli` public docker image, Cloud Run, and the GCP health check service. Here's an example; make sure to change your region, models, user_attributes, and external_user_id.
+One of the simplest ways to launch the health check is the `lkr-cli` public docker image, Cloud Run, and the GCP health check service. Here's an example; make sure to change your region and project. HEALTH_URL is an example of how to structure the query parameters for the health check.
 
 ```bash
 export REGION=<your region>
 export PROJECT=<your project id>
 
-export HEALTH_URL="/health?dashboard_id=1&external_user_id=embed-user-abc&models=thelook&user_attributes={\"store_id\":\"1\"}"
+export HEALTH_URL="/health?dashboard_id=1&external_user_id=observability-embed-user&models=thelook&user_attributes={\"store_id\":\"1\"}"
 
 gcloud run deploy lkr-observability \
   --image us-central1-docker.pkg.dev/lkr-dev-production/lkr-cli/cli:latest \
   --command lkr \
   --args observability,embed \
   --platform managed \
-  --region $REGION \ 
+  --region $REGION \
   --project $PROJECT \
   --cpu 2 \
   --memory 4Gi \
-  --liveness-probe httpGet.path=$HEALTH_URL,timeoutSeconds=20,periodSeconds=20 \
   --set-env-vars LOOKERSDK_CLIENT_ID=<your client id>,LOOKERSDK_CLIENT_SECRET=<your client secret>,LOOKERSDK_BASE_URL=<your instance url> 
 
+gcloud monitoring uptime create lkr-observability-health-check \
+  --protocol https \
+  --project $PROJECT \
+  --resource-type="cloud-run-revision" \
+  --resource-labels="project_id=${PROJECT},service_name=lkr-observability,location=${REGION}" \
+  --path="${HEALTH_URL}" \
+  --period="15" \
+  --timeout="60"
 
 ```
 

lkr_dev_cli-0.0.26/lkr/__init__.py

@@ -0,0 +1,3 @@
+from lkr.tools.classes import UserAttributeUpdater
+
+__all__ = ["UserAttributeUpdater"]

{lkr_dev_cli-0.0.24 → lkr_dev_cli-0.0.26}/lkr/logger.py

@@ -8,6 +8,13 @@ from rich.theme import Theme
 
 from lkr.custom_types import LogLevel
 
+structlog.configure(
+    processors=[
+        structlog.processors.TimeStamper(fmt="iso"),
+        structlog.processors.JSONRenderer(),
+    ]
+)
+
 # Define a custom theme for our logging
 theme = Theme(
     {
@@ -50,6 +57,7 @@ logging.basicConfig(
 logger = logging.getLogger("lkr")
 structured_logger = structlog.get_logger("lkr.structured")
 
+
 # Configure the requests_transport logger to only show debug messages when LOG_LEVEL is DEBUG
 requests_logger = logging.getLogger("looker_sdk.rtl.requests_transport")
 if log_level != "DEBUG":

{lkr_dev_cli-0.0.24 → lkr_dev_cli-0.0.26}/lkr/main.py

@@ -9,6 +9,7 @@ from lkr.custom_types import LogLevel
 from lkr.logger import logger
 from lkr.mcp.main import group as mcp_group
 from lkr.observability.main import group as observability_group
+from lkr.tools.main import group as tools_group
 
 app = typer.Typer(
     name="lkr", help="LookML Repository CLI", add_completion=True, no_args_is_help=True
@@ -17,6 +18,7 @@ app = typer.Typer(
 app.add_typer(auth_group, name="auth")
 app.add_typer(mcp_group, name="mcp")
 app.add_typer(observability_group, name="observability")
+app.add_typer(tools_group, name="tools")
 
 
 @app.callback()

lkr_dev_cli-0.0.26/lkr/mcp/classes.py

@@ -0,0 +1,99 @@
+# server.py
+from datetime import datetime
+from typing import Annotated, Any, List
+
+import duckdb
+from pydantic import BaseModel, Field, computed_field
+
+from lkr.mcp.utils import get_database_search_file
+from lkr.observability.classes import now
+
+
+class SpectaclesResponse(BaseModel):
+    success: bool
+    result: Any | None = None
+    error: str | None = None
+    sql: str | None = None
+    share_url: str | None = None
+
+
+class SpectaclesRequest(BaseModel):
+    model: Annotated[
+        str,
+        Field(
+            description="the model to run a test query against, you can find this by the filenames in the repository, they will end with .model.lkml. You should not pass in the .model.lkml extension.",
+            default="",
+        ),
+    ]
+    explore: Annotated[
+        str,
+        Field(
+            description="the explore to run a test query against, you can find this by finding explore: <name> {} in any file in the repository",
+            default="",
+        ),
+    ]
+    fields: Annotated[
+        List[str],
+        Field(
+            description="this should be the list of fields you want to return from the test query. If the user does not provide them, use all that have changed in your current context",
+            default=[],
+        ),
+    ]
+
+
+class Connection(BaseModel):
+    connection: str
+    updated_at: datetime = Field(default_factory=now)
+
+    @computed_field(return_type=str)
+    @property
+    def fully_qualified_name(self) -> str:
+        return self.connection
+
+
+class Database(Connection):
+    database: str
+
+    @computed_field(return_type=str)
+    @property
+    def fully_qualified_name(self) -> str:
+        return f"{self.connection}.{self.database}"
+
+
+class Schema(Database):
+    database_schema_name: str
+
+    @computed_field(return_type=str)
+    @property
+    def fully_qualified_name(self) -> str:
+        return f"{self.connection}.{self.database}.{self.database_schema_name}"
+
+
+class Table(Schema):
+    database_table_name: str
+
+    @computed_field(return_type=str)
+    @property
+    def fully_qualified_name(self) -> str:
+        return f"{self.connection}.{self.database}.{self.database_schema_name}.{self.database_table_name}"
+
+
+class Row(Table):
+    database_column_name: str
+    data_type_database: str
+    data_type_looker: str
+
+    @computed_field(return_type=str)
+    @property
+    def fully_qualified_name(self) -> str:
+        return f"{self.connection}.{self.database}.{self.database_schema_name}.{self.database_table_name}.{self.database_column_name}"
+
+    def append(self, base_url: str) -> None:
+        with open(get_database_search_file(base_url), "a") as f:
+            f.write(self.model_dump_json() + "\n")
+
+    def exists(self, conn: duckdb.DuckDBPyConnection, *, base_url: str) -> bool:
+        columns = conn.execute(
+            f"SELECT * FROM read_json_auto('{get_database_search_file(base_url)}') WHERE fully_qualified_name = '{self.fully_qualified_name}'"
+        ).fetchall()
+        return len(columns) > 0