gibson-cli 0.8.3__tar.gz → 0.8.4__tar.gz

{gibson_cli-0.8.3/gibson_cli.egg-info → gibson_cli-0.8.4}/PKG-INFO

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

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

@@ -0,0 +1,182 @@
+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
+
+
+def error_handler(e: HTTPError) -> Dict:
+    """Handle HTTP errors from the API"""
+    status_code = e.response.status_code
+    message = e.response.json()
+    if status_code == 401:
+        message = "Authentication required. Instruct the user to run `uvx --from gibson-cli@latest gibson auth login` to authenticate then try again."
+    return {"status_code": status_code, "error": message}
+
+
+@mcp.tool()
+def get_projects() -> List[Dict]:
+    """
+    Get all of the user's existing GibsonAI projects.
+    <IMPORTANT>
+    If the user mentions a project by name and you don't have the project UUID, call this tool.
+    If there's a .gibsonai file in the project or workspace root directory, assume that the user wants to use that project and don't call this tool unless there's a good reason to do so.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.list()
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def create_project() -> Dict:
+    """
+    Create a new GibsonAI project.
+    <IMPORTANT>
+    Before assuming that the user wants to create a new project, look for a .gibsonai file in the project or workspace root directory.
+    If the .gibsonai file exists and contains a project UUID, call get_project_details to get the project details and ask the user if they want to use the existing project.
+    If the .gibsonai file doesn't exist, but the user mentions a project name, call get_projects to check if a project with a similar name already exists and ask the user if they want to use the existing project.
+    If the user explicitly states they want to create a new project, call this tool without confirming.
+    If you call this tool, ask the user if they want to update the .gibsonai file (or create it if it doesn't exist) with the new project UUID.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.create()
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def get_project_details(uuid: str) -> Dict:
+    """
+    Get a GibsonAI project's details.
+    <IMPORTANT>
+    If there's a .gibsonai file in the project or workspace root directory, assume that the user wants to use that project unless they explicitly state otherwise.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.lookup(uuid=uuid)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def get_project_hosted_api_details(uuid: str) -> str:
+    """
+    Get a GibsonAI project's hosted API details.
+    <IMPORTANT>
+    This includes necessary context for an LLM to understand and generate code related to fetching or modifying the project's data via the hosted REST API.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.mcp(uuid=uuid)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@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(interactive=False))
+    try:
+        return project_api.update(uuid=uuid, name=project_name)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def submit_data_modeling_request(uuid: str, data_modeling_request: str) -> Dict:
+    """
+    Submit a data modeling request for a GibsonAI project.
+    <IMPORTANT>
+    This tool fully handles all data modeling.
+    If the user describes a specific data modeling task, call this tool with the user's request as-is.
+    If the user's request is not clear, ask for clarification before calling this tool.
+    </IMPORTANT>
+    Returns the response from GibsonAI's LLM.
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.submit_message(uuid=uuid, message=data_modeling_request)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def deploy_project(uuid: str) -> None:
+    """
+    Deploy a GibsonAI project's hosted databases.
+    <IMPORTANT>
+    This updates both the development and production database schemas simultaneously by automatically handling necessary schema migrations.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.deploy(uuid=uuid)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def get_project_schema(uuid: str) -> str:
+    """
+    Get the current schema for a GibsonAI project.
+    <IMPORTANT>
+    This includes any changes made to the schema since the last deployment.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.schema(uuid=uuid)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def get_deployed_schema(uuid: str) -> str:
+    """
+    Get the deployed schema for a GibsonAI project.
+    <IMPORTANT>
+    This is the database schema that is currently live on the project's hosted databases.
+    </IMPORTANT>
+    """
+    project_api = ProjectApi(Configuration(interactive=False))
+    try:
+        return project_api.database_schema(uuid=uuid)
+    except HTTPError as e:
+        return error_handler(e)
+
+
+@mcp.tool()
+def query_database(api_key: str, query: str) -> List[Dict] | str | Dict:
+    """
+    Query a GibsonAI project's hosted database using SQL. The environment-specific API key must be provided.
+    <IMPORTANT>
+    If you're not sure which environment to use, ask the user for clarification.
+    Always use the correct syntax for the database dialect (found in the project details).
+    Always wrap identifiers in the dialect appropriate quotes (backticks for MySQL, double quotes for PostgreSQL).
+    </IMPORTANT>
+    """
+    data_api = DataApi(Configuration(interactive=False), api_key=api_key)
+    try:
+        response = data_api.query(query=query)
+        return response or "Query executed successfully"
+    except HTTPError as e:
+        return error_handler(e)

{gibson_cli-0.8.3 → gibson_cli-0.8.4/gibson_cli.egg-info}/PKG-INFO

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

{gibson_cli-0.8.3 → gibson_cli-0.8.4}/pyproject.toml

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

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

@@ -1,145 +0,0 @@
-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
-
-
-def error_handler(e: HTTPError) -> Dict:
-    """Handle HTTP errors from the API"""
-    status_code = e.response.status_code
-    message = e.response.json()
-    if status_code == 401:
-        message = "Authentication required. Run `uvx --from gibson-cli@latest gibson auth login` to authenticate then try again."
-    return {"status_code": status_code, "error": message}
-
-
-@mcp.tool()
-def get_projects() -> List[Dict]:
-    """Get all GibsonAI projects"""
-    project_api = ProjectApi(Configuration(interactive=False))
-    try:
-        return project_api.list()
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@mcp.tool()
-def create_project() -> Dict:
-    """Create a new GibsonAI project"""
-    project_api = ProjectApi(Configuration(interactive=False))
-    try:
-        return project_api.create()
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@mcp.tool()
-def get_project_details(uuid: str) -> Dict:
-    """Get a GibsonAI project's details"""
-    project_api = ProjectApi(Configuration(interactive=False))
-    try:
-        return project_api.lookup(uuid=uuid)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.mcp(uuid=uuid)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.update(uuid=uuid, name=project_name)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.submit_message(uuid=uuid, message=data_modeling_request)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.deploy(uuid=uuid)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.schema(uuid=uuid)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False))
-    try:
-        return project_api.database_schema(uuid=uuid)
-    except HTTPError as e:
-        return error_handler(e)
-
-
-@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(interactive=False), api_key=api_key)
-    try:
-        return data_api.query(query=query)
-    except HTTPError as e:
-        return error_handler(e)