created utils for efs

Author: remi-td

src/teradata_mcp_server/server.py

@@ -460,52 +460,7 @@ def get_glossary_term(term_name: str)  -> dict:
 # Feature tools leveraging the tdfs4ds package.
 # Run only if the EFS tools are defined in the config
 if any(re.match(pattern, 'fs_*') for pattern in config.get('tool',[])):
-    class FeatureStoreConfig(BaseModel):
-        """
-        Configuration class for the feature store. This model defines the metadata and catalog sources 
-        used to organize and access features, processes, and datasets across data domains.
-        """
-
-        data_domain: Optional[str] = Field(
-            default=None,
-            description="The data domain associated with the feature store, grouping features within the same namespace."
-        )
-
-        entity: Optional[str] = Field(
-            default=None,
-            description="The list of entities, comma separated and in alphabetical order, upper case."
-        )
-
-        db_name: Optional[str] = Field(
-            default=None,
-            description="Name of the database where the feature store is hosted."
-        )
-
-        feature_catalog: Optional[str] = Field(
-            default=None,
-            description=(
-                "Name of the feature catalog table. "
-                "This table contains detailed metadata about features and entities."
-            )
-        )
-
-        process_catalog: Optional[str] = Field(
-            default=None,
-            description=(
-                "Name of the process catalog table. "
-                "Used to retrieve information about feature generation processes, features, and associated entities."
-            )
-        )
-
-        dataset_catalog: Optional[str] = Field(
-            default=None,
-            description=(
-                "Name of the dataset catalog table. "
-                "Used to list and manage available datasets within the feature store."
-            )
-        )
-
-    fs_config = FeatureStoreConfig() 
+    fs_config = td.FeatureStoreConfig() 
 
     @mcp.tool(description="Reconnect to the Teradata database if the connection is lost.")
     async def reconnect_to_database() -> ResponseType:

src/teradata_mcp_server/tools/fs/__init__.py

@@ -1,2 +1,3 @@
 from .fs_resources import *
-from .fs_tools import *
+from .fs_tools import *
+from .fs_utils import *

src/teradata_mcp_server/tools/fs/fs_utils.py

@@ -0,0 +1,47 @@
+from typing import Any, List, Optional
+from pydantic import Field, BaseModel
+
+class FeatureStoreConfig(BaseModel):
+    """
+    Configuration class for the feature store. This model defines the metadata and catalog sources 
+    used to organize and access features, processes, and datasets across data domains.
+    """
+
+    data_domain: Optional[str] = Field(
+        default=None,
+        description="The data domain associated with the feature store, grouping features within the same namespace."
+    )
+
+    entity: Optional[str] = Field(
+        default=None,
+        description="The list of entities, comma separated and in alphabetical order, upper case."
+    )
+
+    db_name: Optional[str] = Field(
+        default=None,
+        description="Name of the database where the feature store is hosted."
+    )
+
+    feature_catalog: Optional[str] = Field(
+        default=None,
+        description=(
+            "Name of the feature catalog table. "
+            "This table contains detailed metadata about features and entities."
+        )
+    )
+
+    process_catalog: Optional[str] = Field(
+        default=None,
+        description=(
+            "Name of the process catalog table. "
+            "Used to retrieve information about feature generation processes, features, and associated entities."
+        )
+    )
+
+    dataset_catalog: Optional[str] = Field(
+        default=None,
+        description=(
+            "Name of the dataset catalog table. "
+            "Used to list and manage available datasets within the feature store."
+        )
+    )