google-cloud-db-context-engineering 0.7.0__tar.gz → 0.7.1__tar.gz

{google_cloud_db_context_engineering-0.7.0/src/google_cloud_db_context_engineering.egg-info → google_cloud_db_context_engineering-0.7.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-db-context-engineering
-Version: 0.7.0
+Version: 0.7.1
 Summary: A FastMCP server for generating natural language to SQL templates from database schemas.
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -95,6 +95,18 @@ You can provide business artifacts directly to the agent from local filesystems
 
 ---
 
+## Installing via a compatible Agent Plugins client
+
+This repository ships a valid [Agent Plugins](https://github.com/agentplugins/agent-plugins-spec) (v1) plugin under [`plugin/`](plugin/), bundling both agent skills and its MCP servers. Any [Agent Plugins–compatible client](https://agent-plugins.org/compatible-clients) (VS Code, Cursor, GitHub Copilot, Kiro, and others) can install it directly using its own built-in plugin command — no extra tooling required — by pointing at this repository:
+
+```
+https://github.com/GoogleCloudPlatform/db-context-enrichment
+```
+
+See your agent's documentation for its exact install command.
+
+---
+
 ## How to Use
 
 Launch your agent harness (Gemini CLI, Claude Code, or Antigravity) in your workspace directory and interact in natural language:

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/README.md

@@ -77,6 +77,18 @@ You can provide business artifacts directly to the agent from local filesystems
 
 ---
 
+## Installing via a compatible Agent Plugins client
+
+This repository ships a valid [Agent Plugins](https://github.com/agentplugins/agent-plugins-spec) (v1) plugin under [`plugin/`](plugin/), bundling both agent skills and its MCP servers. Any [Agent Plugins–compatible client](https://agent-plugins.org/compatible-clients) (VS Code, Cursor, GitHub Copilot, Kiro, and others) can install it directly using its own built-in plugin command — no extra tooling required — by pointing at this repository:
+
+```
+https://github.com/GoogleCloudPlatform/db-context-enrichment
+```
+
+See your agent's documentation for its exact install command.
+
+---
+
 ## How to Use
 
 Launch your agent harness (Gemini CLI, Claude Code, or Antigravity) in your workspace directory and interact in natural language:

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "google-cloud-db-context-engineering"
-version = "0.7.0"
+version = "0.7.1"
 description = "A FastMCP server for generating natural language to SQL templates from database schemas."
 readme = "README.md"
 requires-python = ">=3.12"
@@ -46,7 +46,7 @@ dev = [
 
 [tool.db-context-engineering]
 toolbox_version = "1.4.0"
-evalbench_version = "1.9.0"
+evalbench_version = "1.12.0"
 
 [tool.ruff]
 line-length = 88

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/src/google/cloud/db_context_enrichment/evaluate/db_generators/alloydb.py

@@ -1,6 +1,5 @@
 from typing import Any
 
-import google.cloud.geminidataanalytics_v1beta as gda
 import yaml
 
 from .base import BaseDBConfigGenerator
@@ -52,21 +51,20 @@ class AlloyDBConfigGenerator(BaseDBConfigGenerator):
             db_config, sort_keys=False, default_flow_style=False
         ).strip()
 
-    def build_datasource_reference(
-        self, context_set_id: str
-    ) -> gda.DatasourceReferences:
-        datasource_ref = gda.DatasourceReferences()
-
-        datasource_ref.alloydb = gda.AlloyDbReference(
-            database_reference=gda.AlloyDbDatabaseReference(
-                project_id=self.project,
-                region=self.region,
-                cluster_id=self.cluster,
-                instance_id=self.instance,
-                database_id=self.database,
-            ),
-            agent_context_reference=gda.AgentContextReference(
-                context_set_id=context_set_id
-            ),
-        )
-        return datasource_ref
+    def build_datasource_reference(self, context_set_id: str) -> dict[str, Any]:
+        ref: dict[str, Any] = {
+            "alloydb": {
+                "database_reference": {
+                    "project_id": self.project,
+                    "region": self.region,
+                    "cluster_id": self.cluster,
+                    "instance_id": self.instance,
+                    "database_id": self.database,
+                }
+            }
+        }
+        if context_set_id:
+            ref["alloydb"]["agent_context_reference"] = {
+                "context_set_id": context_set_id
+            }
+        return ref

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/src/google/cloud/db_context_enrichment/evaluate/db_generators/base.py

@@ -1,16 +1,14 @@
 from abc import ABC, abstractmethod
 from typing import Any
 
-import google.cloud.geminidataanalytics_v1beta as gda
 import yaml
-from google.protobuf.json_format import MessageToDict
 
 
 class BaseDBConfigGenerator(ABC):
     """
     Abstract Base Class enforcing the construction contract for Evalbench database topologies.
     Each distinct DB type (Spanner, Postgres, AlloyDB, MySQL) must inherit and implement
-    the mappings required by both the standard EvalBench framework and the GDA SDK model.
+    the mappings required by both the standard EvalBench framework and the GDA REST API.
     """
 
     SOURCE_TYPE = "unknown"
@@ -29,11 +27,9 @@ class BaseDBConfigGenerator(ABC):
         raise NotImplementedError("Subclasses must implement generate_db_config")
 
     @abstractmethod
-    def build_datasource_reference(
-        self, context_set_id: str
-    ) -> gda.DatasourceReferences:
+    def build_datasource_reference(self, context_set_id: str) -> dict[str, Any]:
         """
-        Constructs the strict Protocol Buffer DatasourceReference required by the QueryDataAPI
+        Constructs the REST datasource_references dictionary required by the QueryDataAPI
         context generation flow.
         """
         raise NotImplementedError(
@@ -54,23 +50,21 @@ class BaseDBConfigGenerator(ABC):
 
     def generate_model_config(self, context_set_id: str) -> str:
         """
-        Standardized Model Builder converting the strictly typed GDA object into an EvalBench model dict.
+        Standardized Model Builder converting the datasource reference dict into an EvalBench model dict.
         """
         datasource_ref = self.build_datasource_reference(context_set_id)
 
-        query_context = gda.QueryDataContext(datasource_references=datasource_ref)
-
-        query_context_dict = MessageToDict(
-            query_context._pb, preserving_proto_field_name=True
-        )
-
         model_config = {
             "generator": "query_data_api",
             "project_id": self.params.get("project"),
             "location": self.params.get("region") or "global",
-            "context": query_context_dict,
+            "use_rest_api": True,
+            "context": {"datasource_references": datasource_ref},
         }
 
+        if self.params.get("api_endpoint"):
+            model_config["api_endpoint"] = self.params.get("api_endpoint")
+
         return yaml.safe_dump(
             model_config, sort_keys=False, default_flow_style=False
         ).strip()

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/src/google/cloud/db_context_enrichment/evaluate/db_generators/mysql.py

@@ -1,6 +1,5 @@
 from typing import Any
 
-import google.cloud.geminidataanalytics_v1beta as gda
 import yaml
 
 from .base import BaseDBConfigGenerator
@@ -49,21 +48,20 @@ class MySQLConfigGenerator(BaseDBConfigGenerator):
             db_config, sort_keys=False, default_flow_style=False
         ).strip()
 
-    def build_datasource_reference(
-        self, context_set_id: str
-    ) -> gda.DatasourceReferences:
-        datasource_ref = gda.DatasourceReferences()
-
-        datasource_ref.cloud_sql_reference = gda.CloudSqlReference(
-            database_reference=gda.CloudSqlDatabaseReference(
-                engine=gda.CloudSqlDatabaseReference.Engine.MYSQL,
-                project_id=self.project,
-                region=self.region,
-                instance_id=self.instance,
-                database_id=self.database,
-            ),
-            agent_context_reference=gda.AgentContextReference(
-                context_set_id=context_set_id
-            ),
-        )
-        return datasource_ref
+    def build_datasource_reference(self, context_set_id: str) -> dict[str, Any]:
+        ref: dict[str, Any] = {
+            "cloud_sql_reference": {
+                "database_reference": {
+                    "engine": "MYSQL",
+                    "project_id": self.project,
+                    "region": self.region,
+                    "instance_id": self.instance,
+                    "database_id": self.database,
+                }
+            }
+        }
+        if context_set_id:
+            ref["cloud_sql_reference"]["agent_context_reference"] = {
+                "context_set_id": context_set_id
+            }
+        return ref

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/src/google/cloud/db_context_enrichment/evaluate/db_generators/postgres.py

@@ -1,6 +1,5 @@
 from typing import Any
 
-import google.cloud.geminidataanalytics_v1beta as gda
 import yaml
 
 from .base import BaseDBConfigGenerator
@@ -49,21 +48,20 @@ class PostgresConfigGenerator(BaseDBConfigGenerator):
             db_config, sort_keys=False, default_flow_style=False
         ).strip()
 
-    def build_datasource_reference(
-        self, context_set_id: str
-    ) -> gda.DatasourceReferences:
-        datasource_ref = gda.DatasourceReferences()
-
-        datasource_ref.cloud_sql_reference = gda.CloudSqlReference(
-            database_reference=gda.CloudSqlDatabaseReference(
-                engine=gda.CloudSqlDatabaseReference.Engine.POSTGRESQL,
-                project_id=self.project,
-                region=self.region,
-                instance_id=self.instance,
-                database_id=self.database,
-            ),
-            agent_context_reference=gda.AgentContextReference(
-                context_set_id=context_set_id
-            ),
-        )
-        return datasource_ref
+    def build_datasource_reference(self, context_set_id: str) -> dict[str, Any]:
+        ref: dict[str, Any] = {
+            "cloud_sql_reference": {
+                "database_reference": {
+                    "engine": "POSTGRESQL",
+                    "project_id": self.project,
+                    "region": self.region,
+                    "instance_id": self.instance,
+                    "database_id": self.database,
+                }
+            }
+        }
+        if context_set_id:
+            ref["cloud_sql_reference"]["agent_context_reference"] = {
+                "context_set_id": context_set_id
+            }
+        return ref

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1}/src/google/cloud/db_context_enrichment/evaluate/db_generators/spanner.py

@@ -1,6 +1,5 @@
 from typing import Any
 
-import google.cloud.geminidataanalytics_v1beta as gda
 import yaml
 
 from .base import BaseDBConfigGenerator
@@ -43,20 +42,21 @@ class SpannerConfigGenerator(BaseDBConfigGenerator):
             db_config, sort_keys=False, default_flow_style=False
         ).strip()
 
-    def build_datasource_reference(
-        self, context_set_id: str
-    ) -> gda.DatasourceReferences:
-        datasource_ref = gda.DatasourceReferences()
-
-        datasource_ref.spanner_reference = gda.SpannerReference(
-            database_reference=gda.SpannerDatabaseReference(
-                engine=gda.SpannerDatabaseReference.Engine.GOOGLE_SQL,
-                project_id=self.project,
-                instance_id=self.instance,
-                database_id=self.database,
-            ),
-            agent_context_reference=gda.AgentContextReference(
-                context_set_id=context_set_id
-            ),
-        )
-        return datasource_ref
+    def build_datasource_reference(self, context_set_id: str) -> dict[str, Any]:
+        database_ref: dict[str, Any] = {
+            "engine": "GOOGLE_SQL",
+            "project_id": self.project,
+            "instance_id": self.instance,
+            "database_id": self.database,
+        }
+        if graph_ids := self.params.get("graph_ids"):
+            if not isinstance(graph_ids, list) or not all(
+                isinstance(g, str) for g in graph_ids
+            ):
+                raise ValueError("graph_ids must be a list of strings")
+            database_ref["graph_ids"] = graph_ids
+
+        spanner_ref: dict[str, Any] = {"database_reference": database_ref}
+        if context_set_id:
+            spanner_ref["agent_context_reference"] = {"context_set_id": context_set_id}
+        return {"spanner_reference": spanner_ref}

{google_cloud_db_context_engineering-0.7.0 → google_cloud_db_context_engineering-0.7.1/src/google_cloud_db_context_engineering.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: google-cloud-db-context-engineering
-Version: 0.7.0
+Version: 0.7.1
 Summary: A FastMCP server for generating natural language to SQL templates from database schemas.
 Requires-Python: >=3.12
 Description-Content-Type: text/markdown
@@ -95,6 +95,18 @@ You can provide business artifacts directly to the agent from local filesystems
 
 ---
 
+## Installing via a compatible Agent Plugins client
+
+This repository ships a valid [Agent Plugins](https://github.com/agentplugins/agent-plugins-spec) (v1) plugin under [`plugin/`](plugin/), bundling both agent skills and its MCP servers. Any [Agent Plugins–compatible client](https://agent-plugins.org/compatible-clients) (VS Code, Cursor, GitHub Copilot, Kiro, and others) can install it directly using its own built-in plugin command — no extra tooling required — by pointing at this repository:
+
+```
+https://github.com/GoogleCloudPlatform/db-context-enrichment
+```
+
+See your agent's documentation for its exact install command.
+
+---
+
 ## How to Use
 
 Launch your agent harness (Gemini CLI, Claude Code, or Antigravity) in your workspace directory and interact in natural language: