gibson-cli 0.8.3__py3-none-any.whl → 0.8.4__py3-none-any.whl

gibson/services/mcp/server.py

@@ -18,13 +18,19 @@ def error_handler(e: HTTPError) -> Dict:
     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."
+        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 GibsonAI projects"""
+    """
+    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()
@@ -34,7 +40,16 @@ def get_projects() -> List[Dict]:
 
 @mcp.tool()
 def create_project() -> Dict:
-    """Create a new GibsonAI project"""
+    """
+    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()
@@ -44,7 +59,12 @@ def create_project() -> Dict:
 
 @mcp.tool()
 def get_project_details(uuid: str) -> Dict:
-    """Get a GibsonAI project's details"""
+    """
+    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)
@@ -55,8 +75,10 @@ def get_project_details(uuid: str) -> Dict:
 @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
+    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:
@@ -68,9 +90,9 @@ def get_project_hosted_api_details(uuid: str) -> str:
 @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
+    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:
@@ -82,9 +104,13 @@ def update_project(uuid: str, project_name: str) -> Dict:
 @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
+    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:
@@ -96,8 +122,10 @@ def submit_data_modeling_request(uuid: str, data_modeling_request: str) -> Dict:
 @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
+    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:
@@ -109,8 +137,10 @@ def deploy_project(uuid: str) -> None:
 @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
+    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:
@@ -122,8 +152,10 @@ def get_project_schema(uuid: str) -> str:
 @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
+    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:
@@ -133,13 +165,18 @@ def get_deployed_schema(uuid: str) -> str:
 
 
 @mcp.tool()
-def query_database(api_key: str, query: str) -> List[Dict] | None | Dict:
+def query_database(api_key: str, query: str) -> List[Dict] | str | Dict:
     """
-    Query a GibsonAI project's hosted database using SQL
-    Note: the environment-specific API key must be provided
+    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:
-        return data_api.query(query=query)
+        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.dist-info → gibson_cli-0.8.4.dist-info}/METADATA

@@ -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.dist-info → gibson_cli-0.8.4.dist-info}/RECORD

@@ -106,7 +106,7 @@ gibson/services/code/customization/CustomizationManager.py,sha256=M2gz98Yo2WTnnh
 gibson/services/code/customization/Index.py,sha256=4Thf0gZM6VErZJS97w748PRNmHi8QvsyblOLCw1Y_XE,364
 gibson/services/code/customization/tests/test_code_customization_Authenticator.py,sha256=kKExkLfKPpRA2NQH3fvRCuBEMhCGhR-IvNJqXuyBz3c,1949
 gibson/services/code/customization/tests/test_code_customization_BaseCustomization.py,sha256=jaEwxxoU7d9ziOtfF21NPmZX2qSRpa-kz_8Ju9BKGts,412
-gibson/services/mcp/server.py,sha256=pmJrZ2A1U33Y_UMCA6kgvIoqUZju3m8fLLfOjaAo_oM,4533
+gibson/services/mcp/server.py,sha256=nwDEO7k6vroiEr9zPLeSWUsC86h_r7kyEmwYPTBqcow,6684
 gibson/structure/Entity.py,sha256=N_Tx8RTs9ySMMgAoR9rVuMcsRgNA7zvNvJBScJLfYE4,675
 gibson/structure/mysql/Entity.py,sha256=zolt3N_F3WlQtlOqrHflwsJeJ6r6A3MN4LxCzeAbU_k,3693
 gibson/structure/mysql/testing.py,sha256=al4LI6e3bhjopsR0qTAmaOJyCQXF0_inVQ4xv7VQ6qo,9149
@@ -129,8 +129,8 @@ gibson/tests/test_Env.py,sha256=DPWmP0-aEelducq9bAwv7rKoY2NjWXUeCrzfJDQkn2M,369
 gibson/tests/test_Memory.py,sha256=YP7owToABAk_-s7fD5UG0HTc4lamDjdA39JUlLnk3Fg,2574
 gibson/tests/test_utils.py,sha256=r_y-EG05YTCNtL8MWiAK1KmPsmeoMgypKsQC_lVgOtM,559
 venv/bin/activate_this.py,sha256=E1T7r3559tBsyqFpdcQW0HbY7gDvNiIv5Pc6HQ4bpoA,2383
-gibson_cli-0.8.3.dist-info/METADATA,sha256=NJ34YJ9UIbKeQMbn2dkE4jXy1gykNNsNZ3l0_kR67tU,14592
-gibson_cli-0.8.3.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
-gibson_cli-0.8.3.dist-info/entry_points.txt,sha256=j5VUvq3AzL21xPvVC24zMoXFt-I5lUWulr66nL3OAPM,50
-gibson_cli-0.8.3.dist-info/top_level.txt,sha256=fSV3vegbdbSDwiB6n5z3FCeYwkIonzFrx4ek3F_OSdI,16
-gibson_cli-0.8.3.dist-info/RECORD,,
+gibson_cli-0.8.4.dist-info/METADATA,sha256=aW2ZihtrumK5XY9vi1wv1LVdC--xCYNTyjVYsP7LI8Q,14592
+gibson_cli-0.8.4.dist-info/WHEEL,sha256=CmyFI0kx5cdEMTLiONQRbGQwjIoR1aIYB7eCAQ4KPJ0,91
+gibson_cli-0.8.4.dist-info/entry_points.txt,sha256=j5VUvq3AzL21xPvVC24zMoXFt-I5lUWulr66nL3OAPM,50
+gibson_cli-0.8.4.dist-info/top_level.txt,sha256=fSV3vegbdbSDwiB6n5z3FCeYwkIonzFrx4ek3F_OSdI,16
+gibson_cli-0.8.4.dist-info/RECORD,,