refactor: rename facet to sql_snippet in ContextSet and update version to 0.2.0

Author: wangauone

gemini-extension.json

@@ -1,6 +1,6 @@
 {
   "name": "mcp-db-context-enrichment",
-  "version": "0.1.6",
+  "version": "0.2.0",
   "contextFileName": "./mcp/GEMINI.md",
   "mcpServers": {
     "mcp_db_context_enrichment": {
@@ -9,7 +9,7 @@
         "tool",
         "run",
         "--from",
-        "${extensionPath}${/}mcp${/}dist${/}db_context_enrichment-0.1.6-py3-none-any.whl",
+        "${extensionPath}${/}mcp${/}dist${/}db_context_enrichment-0.2.0-py3-none-any.whl",
         "db-context-enrichment"
       ]
     }

mcp/GEMINI.md

@@ -64,11 +64,11 @@ The `ContextSet` object is a JSON structure that can contain both `templates` an
   ],
   "facets": [
     {
-      "facet": "description LIKE '%luxury%' OR description LIKE '%premium%'",
+      "sql_snippet": "description LIKE '%luxury%' OR description LIKE '%premium%'",
       "intent": "luxury product",
       "manifest": "luxury product",
       "parameterized": {
-        "parameterized_facet": "description LIKE '%luxury%' OR description LIKE '%premium%'",
+        "parameterized_sql_snippet": "description LIKE '%luxury%' OR description LIKE '%premium%'",
         "parameterized_intent": "luxury product"
       }
     }

mcp/dist/db_context_enrichment-0.1.6-py3-none-any.whl

@@ -49,11 +49,11 @@ async def generate_facets_from_pairs(
 
         # 4. Assemble the final facet object
         facet = context.Facet(
-            facet=facet_text,
+            sql_snippet=facet_text,
             intent=intent,
             manifest=manifest,
             parameterized=context.ParameterizedFacet(
-                parameterized_facet=parameterized_result["sql"],
+                parameterized_sql_snippet=parameterized_result["sql"],
                 parameterized_intent=parameterized_result["intent"],
             ),
         )

mcp/dist/db_context_enrichment-0.1.6.tar.gz

@@ -1,4 +1,4 @@
-from pydantic import BaseModel, Field
+from pydantic import AliasChoices, BaseModel, Field
 from typing import List, Optional
 
 
@@ -30,8 +30,12 @@ class Template(BaseModel):
 class ParameterizedFacet(BaseModel):
     """Defines the parameterized version of a SQL facet and intent."""
 
-    parameterized_facet: str = Field(
-        ..., description="The SQL facet with placeholders (eg., )."
+    parameterized_sql_snippet: str = Field(
+        ...,
+        description="The SQL facet with placeholders (eg., ).",
+        validation_alias=AliasChoices(
+            "parameterized_sql_snippet", "parameterized_fragment"
+        ),
     )
     parameterized_intent: str = Field(
         ..., description="The natural language intent with placeholders."
@@ -41,7 +45,11 @@ class ParameterizedFacet(BaseModel):
 class Facet(BaseModel):
     """Represents a single, complete facet."""
 
-    facet: str = Field(..., description="The corresponding, complete SQL facet.")
+    sql_snippet: str = Field(
+        ...,
+        description="The corresponding, complete SQL facet.",
+        validation_alias=AliasChoices("sql_snippet", "fragment"),
+    )
     intent: str = Field(..., description="The user's specific intent.")
     manifest: str = Field(
         ..., description="A general description of what the facet does."
@@ -52,6 +60,12 @@ class Facet(BaseModel):
 class ContextSet(BaseModel):
     """A set of templates and facets."""
 
-    templates: Optional[List[Template]] = Field(None, description="A list of complete templates.")
-    facets: Optional[List[Facet]] = Field(None, description="A list of SQL facets.")
+    templates: Optional[List[Template]] = Field(
+        None, description="A list of complete templates."
+    )
+    facets: Optional[List[Facet]] = Field(
+        None,
+        description="A list of SQL facets.",
+        validation_alias=AliasChoices("facets", "fragments"),
+    )
 

mcp/dist/db_context_enrichment-0.2.0-py3-none-any.whl

@@ -1,6 +1,6 @@
 [project]
 name = "db-context-enrichment"
-version = "0.1.6"
+version = "0.2.0"
 description = "A FastMCP server for generating natural language to SQL templates from database schemas."
 readme = "README.md"
 requires-python = ">=3.12"

mcp/dist/db_context_enrichment-0.2.0.tar.gz

@@ -31,10 +31,10 @@ async def test_generate_facets_from_pairs_simple():
             assert result_context_set.facets is not None
             assert len(result_context_set.facets) == 1
             facet = result_context_set.facets[0]
-            assert facet.facet == "city = 'New York'"
+            assert facet.sql_snippet == "city = 'New York'"
             assert facet.intent == "Find users in New York"
             assert facet.manifest == "Find users in a given city"
-            assert facet.parameterized.parameterized_facet == "city = $1"
+            assert facet.parameterized.parameterized_sql_snippet == "city = $1"
             assert facet.parameterized.parameterized_intent == "Find users in $1"
 
             mock_extract_value_phrases.assert_called_once_with(
@@ -74,13 +74,13 @@ async def test_generate_facets_from_pairs_multiple_phrases():
             assert result_context_set.facets is not None
             assert len(result_context_set.facets) == 1
             facet = result_context_set.facets[0]
-            assert facet.facet == "name = 'John Doe' AND city = 'New York'"
+            assert facet.sql_snippet == "name = 'John Doe' AND city = 'New York'"
             assert facet.intent == "Find users named John Doe in New York"
             assert (
                 facet.manifest == "Find users named a given person in a given city"
             )
             assert (
-                facet.parameterized.parameterized_facet
+                facet.parameterized.parameterized_sql_snippet
                 == "name = $1 AND city = $2"
             )
             assert (
@@ -120,10 +120,10 @@ async def test_generate_facets_from_pairs_empty_phrases():
             assert result_context_set.facets is not None
             assert len(result_context_set.facets) == 1
             facet = result_context_set.facets[0]
-            assert facet.facet == "TRUE"
+            assert facet.sql_snippet == "TRUE"
             assert facet.intent == "List all users"
             assert facet.manifest == "List all users"
-            assert facet.parameterized.parameterized_facet == "TRUE"
+            assert facet.parameterized.parameterized_sql_snippet == "TRUE"
             assert facet.parameterized.parameterized_intent == "List all users"
 
             mock_extract_value_phrases.assert_called_once_with(