gibson-cli 0.8.1__tar.gz → 0.8.2__tar.gz

{gibson_cli-0.8.1/gibson_cli.egg-info → gibson_cli-0.8.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gibson-cli
-Version: 0.8.1
+Version: 0.8.2
 Summary: Gibson Command Line Interface
 Author-email: GibsonAI <noc@gibsonai.com>
 Project-URL: Homepage, https://gibsonai.com/

gibson_cli-0.8.2/gibson/api/DataApi.py

@@ -0,0 +1,18 @@
+from gibson.api.BaseApi import BaseApi
+from gibson.core.Configuration import Configuration
+
+
+class DataApi(BaseApi):
+    PREFIX = "-"
+
+    def __init__(self, configuration: Configuration, api_key: str):
+        self.configuration = configuration
+        self.api_key = api_key or self.configuration.project.api.key
+
+    def headers(self):
+        headers = super().headers()
+        headers["X-Gibson-API-Key"] = self.api_key
+        return headers
+
+    def query(self, query: str):
+        return self.post("query", {"query": query}).json()

{gibson_cli-0.8.1 → gibson_cli-0.8.2}/gibson/bin/gibson.py

@@ -1,10 +1,15 @@
 #!/usr/bin/env python3
 
+import os
+
 from gibson.core.CommandRouter import CommandRouter
 from gibson.core.Configuration import Configuration
+from gibson.display.Header import Header
 
 
 def main():
+    if os.getenv("GIBSON_CLI_DEV"):
+        print(f"{Header().render('dev mode')}\n")
     try:
         configuration = Configuration()
         if configuration.settings is None:

gibson_cli-0.8.2/gibson/services/mcp/server.py

@@ -0,0 +1,136 @@
+from typing import Dict, List
+
+from mcp.server.fastmcp import FastMCP
+from requests.exceptions import HTTPError
+
+from gibson.api.DataApi import DataApi
+from gibson.api.ProjectApi import ProjectApi
+from gibson.core.Configuration import Configuration
+
+mcp = FastMCP("GibsonAI")
+
+# Note: Resources are not yet supported by Cursor, everything must be implemented as a tool
+# See https://docs.cursor.com/context/model-context-protocol#limitations
+
+
+@mcp.tool()
+def get_projects() -> List[Dict]:
+    """Get all GibsonAI projects"""
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.list()
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def create_project() -> Dict:
+    """Create a new GibsonAI project"""
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.create()
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def get_project_details(uuid: str) -> Dict:
+    """Get a GibsonAI project's details"""
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.lookup(uuid=uuid)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def get_project_hosted_api_details(uuid: str) -> str:
+    """
+    Get a GibsonAI project's hosted API details
+    This includes necessary context for an LLM to understand and generate API calls related to fetching or modifying the project's data
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.mcp(uuid=uuid)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def update_project(uuid: str, project_name: str) -> Dict:
+    """
+    Update a GibsonAI project's details
+    This currently only updates the project's name
+    Returns the updated project details
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.update(uuid=uuid, name=project_name)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def submit_data_modeling_request(uuid: str, data_modeling_request: str) -> Dict:
+    """
+    Submit a data modeling request for a GibsonAI project
+    This tool fully handles all data modeling, you should provide the user's request as-is
+    Returns the response from the LLM
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.submit_message(uuid=uuid, message=data_modeling_request)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def deploy_project(uuid: str) -> None:
+    """
+    Deploy a GibsonAI project's hosted databases
+    This deploys both the development and production databases simultaneously and automatically handles the migrations
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.deploy(uuid=uuid)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def get_project_schema(uuid: str) -> str:
+    """
+    Get the schema for a GibsonAI project
+    This includes any changes made to the schema since the last deployment
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.schema(uuid=uuid)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def get_deployed_schema(uuid: str) -> str:
+    """
+    Get the deployed schema for a GibsonAI project
+    This is the schema that is currently live on the project's hosted databases
+    """
+    project_api = ProjectApi(Configuration())
+    try:
+        return project_api.database_schema(uuid=uuid)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}
+
+
+@mcp.tool()
+def query_database(api_key: str, query: str) -> List[Dict] | None | Dict:
+    """
+    Query a GibsonAI project's hosted database using SQL
+    Note: the environment-specific API key must be provided
+    """
+    data_api = DataApi(Configuration(), api_key=api_key)
+    try:
+        return data_api.query(query=query)
+    except HTTPError as e:
+        return {"status_code": e.response.status_code, "error": e.response.json()}

{gibson_cli-0.8.1 → gibson_cli-0.8.2/gibson_cli.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gibson-cli
-Version: 0.8.1
+Version: 0.8.2
 Summary: Gibson Command Line Interface
 Author-email: GibsonAI <noc@gibsonai.com>
 Project-URL: Homepage, https://gibsonai.com/

{gibson_cli-0.8.1 → gibson_cli-0.8.2}/gibson_cli.egg-info/SOURCES.txt

@@ -5,10 +5,10 @@ pyproject.toml
 requirements.txt
 bin/build.sh
 bin/clean.sh
-bin/gibson
 bin/release.sh
 gibson/api/BaseApi.py
 gibson/api/Cli.py
+gibson/api/DataApi.py
 gibson/api/ProjectApi.py
 gibson/bin/gibson.py
 gibson/command/BaseCommand.py

{gibson_cli-0.8.1 → gibson_cli-0.8.2}/pyproject.toml

@@ -15,7 +15,7 @@ classifiers = [
   "Intended Audience :: Developers",
   "Programming Language :: Python :: 3.10",
 ]
-version = "0.8.1"
+version = "0.8.2"
 
 [tool.setuptools.packages.find]
 where = ["."]

gibson_cli-0.8.1/bin/gibson

@@ -1,26 +0,0 @@
-#!/Library/Frameworks/Python.framework/Versions/3.9/bin/python3.9
-# -*- coding: utf-8 -*-
-
-# This file is identical to what pip creates when you install an executable package, but it adds a dev mode banner.
-# To easily switch between dev mode and installed mode, add the following functions to your .zshrc or .bashrc:
-#
-# cli_dev_on() {
-#   export PATH="$HOME/src/gibson/cli/bin:$PATH"
-#   export PYTHONPATH="$HOME/src/gibson/cli:$PYTHONPATH"
-# }
-#
-# cli_dev_off() {
-#   export PATH=${PATH//$HOME\/src\/gibson\/cli\/bin:/}
-#   export PYTHONPATH=${PYTHONPATH//$HOME\/src\/gibson\/cli:/}
-# }
-
-import re
-import sys
-
-from gibson.bin.gibson import main
-from gibson.display.Header import Header
-
-if __name__ == "__main__":
-    print(f"{Header().render('dev mode')}\n")
-    sys.argv[0] = re.sub(r"(-script\.pyw|\.exe)?$", "", sys.argv[0])
-    sys.exit(main())

gibson_cli-0.8.1/gibson/services/mcp/server.py

@@ -1,87 +0,0 @@
-from typing import Dict, List
-
-from mcp.server.fastmcp import FastMCP
-
-from gibson.api.ProjectApi import ProjectApi
-from gibson.core.Configuration import Configuration
-
-mcp = FastMCP("GibsonAI")
-
-project_api = ProjectApi(Configuration())
-
-# Note: Resources are not yet supported by Cursor, everything must be implemented as a tool
-# See https://docs.cursor.com/context/model-context-protocol#limitations
-
-
-@mcp.tool()
-def get_projects() -> List[Dict]:
-    """Get all GibsonAI projects"""
-    return project_api.list()
-
-
-@mcp.tool()
-def create_project() -> Dict:
-    """Create a new GibsonAI project"""
-    return project_api.create()
-
-
-@mcp.tool()
-def get_project_details(uuid: str) -> Dict:
-    """Get a GibsonAI project's details"""
-    return project_api.lookup(uuid=uuid)
-
-
-@mcp.tool()
-def get_project_hosted_api_details(uuid: str) -> str:
-    """
-    Get a GibsonAI project's hosted API details
-    This includes necessary context for an LLM to understand and generate API calls related to fetching or modifying the project's data
-    """
-    return project_api.mcp(uuid=uuid)
-
-
-@mcp.tool()
-def update_project(uuid: str, project_name: str) -> Dict:
-    """
-    Update a GibsonAI project's details
-    This currently only updates the project's name
-    Returns the updated project details
-    """
-    return project_api.update(uuid=uuid, name=project_name)
-
-
-@mcp.tool()
-def submit_data_modeling_request(uuid: str, data_modeling_request: str) -> Dict:
-    """
-    Submit a data modeling request for a GibsonAI project
-    This tool fully handles all data modeling, you should provide the user's request as-is
-    Returns the response from the LLM
-    """
-    return project_api.submit_message(uuid=uuid, message=data_modeling_request)
-
-
-@mcp.tool()
-def deploy_project(uuid: str) -> None:
-    """
-    Deploy a GibsonAI project's hosted databases
-    This deploys both the development and production databases simultaneously and automatically handles the migrations
-    """
-    project_api.deploy(uuid=uuid)
-
-
-@mcp.tool()
-def get_project_schema(uuid: str) -> str:
-    """
-    Get the schema for a GibsonAI project
-    This includes any changes made to the schema since the last deployment
-    """
-    return project_api.schema(uuid=uuid)
-
-
-@mcp.tool()
-def get_deployed_schema(uuid: str) -> str:
-    """
-    Get the deployed schema for a GibsonAI project
-    This is the schema that is currently live on the project's hosted databases
-    """
-    return project_api.database_schema(uuid=uuid)