tencent-wedata-feature-engineering-dev 0.1.47__py3-none-any.whl → 0.2.2__py3-none-any.whl

{tencent_wedata_feature_engineering_dev-0.1.47.dist-info → tencent_wedata_feature_engineering_dev-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: tencent-wedata-feature-engineering-dev
-Version: 0.1.47
+Version: 0.2.2
 Summary: Wedata Feature Engineering Library Development
 Home-page: 
 Author: meahqian
@@ -16,4 +16,10 @@ Requires-Dist: feast[redis]==0.49.0
 Requires-Dist: grpcio==1.74.0
 Requires-Dist: tencentcloud-sdk-python
 Requires-Dist: ipython
-
+Dynamic: author
+Dynamic: classifier
+Dynamic: description-content-type
+Dynamic: license
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary

{tencent_wedata_feature_engineering_dev-0.1.47.dist-info → tencent_wedata_feature_engineering_dev-0.2.2.dist-info}/RECORD

@@ -1,18 +1,20 @@
 wedata/__init__.py,sha256=GYxqkkgH0oH4QtNiOCZHuGkc0sSH1LgEqmhSX6sB4So,200
 wedata/feature_store/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/client.py,sha256=rXQfSC14vvxPed9P0PoboDTwhD47aPwmGndY825Sl6k,19484
+wedata/feature_store/client.py,sha256=7eBWrhjsr5mxyCRnG0rI6eTMocFDUDlfOr3MwW9GBa0,19197
 wedata/feature_store/mlflow_model.py,sha256=OCUuccOoO0NXWSzIPoGeL03Ha1Q3aQTJW2RlJrTCmzc,554
 wedata/feature_store/cloud_sdk_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/cloud_sdk_client/client.py,sha256=NtzV0EmLTzMLYqWhnUyOx3bmxoIIgxw8REid-Gy96Pc,4234
-wedata/feature_store/cloud_sdk_client/models.py,sha256=7_QUq0kZcrcclRMsIYFoqBrlzVwaHoVY-yU5SHIrJWM,19789
-wedata/feature_store/cloud_sdk_client/utils.py,sha256=6ESwVhlrftnp0h9ojTzbB-m-0hktLI1PLcyk6zpNgrs,857
+wedata/feature_store/cloud_sdk_client/client.py,sha256=I4l3B_4DoikCQruUXL1aXgBgDFZ9pt-aCcS2UXK5qLQ,6290
+wedata/feature_store/cloud_sdk_client/models.py,sha256=aQ8mqPBbzfaX2AzGTBx3-QjDRqqiw9KMmdSpIdY4f4E,24856
+wedata/feature_store/cloud_sdk_client/utils.py,sha256=vffI9rAGkW12dX7IN-QUmIIG865_WY75GtKJgVF6aho,1098
 wedata/feature_store/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wedata/feature_store/common/log/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+wedata/feature_store/common/log/logger.py,sha256=jwTecUnTrR6LL5ocCzTuuPLqW2swjEf4HZSFHTEY5Fc,1287
 wedata/feature_store/common/protos/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wedata/feature_store/common/protos/feature_store_pb2.py,sha256=oMIUGGeGNP84g_nFqOQwTXjV1GiU2ehSOy7CyFu2__g,4207
 wedata/feature_store/common/store_config/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wedata/feature_store/common/store_config/redis.py,sha256=9R5npM2s1u0o9IakmpbRsFdJC0vNar_uvA62OLWuXBs,1145
 wedata/feature_store/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/constants/constants.py,sha256=5kg-0IfYZaB2MbQtwIWa6riTeq-OwFJkGqN1ncc3aqE,1959
+wedata/feature_store/constants/constants.py,sha256=FXKE8BHnp04SDnE2XNrtXHk7U6dpgo8NTvtbaSCiC64,1958
 wedata/feature_store/constants/engine_types.py,sha256=42mI-kNDDtoA4_I3iqDe4FkF2M2l_Bt4Q1V6WUB-_k0,921
 wedata/feature_store/entities/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wedata/feature_store/entities/column_info.py,sha256=-AR6EKHwgoqIkRHFyguxVEtnYt6fvusWHkEjF4kvS0A,5141
@@ -32,14 +34,14 @@ wedata/feature_store/entities/training_set.py,sha256=ylt1h6Z_xU8hKYvnvd80CeewTGS
 wedata/feature_store/feast_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wedata/feature_store/feast_client/feast_client.py,sha256=mCv-OiKehfgcOJhJV0wXMRs5d7e2zEBYmVmDguk0rxU,20728
 wedata/feature_store/feature_table_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/feature_table_client/feature_table_client.py,sha256=qyr-jS-nIyphFVZXcUXV_HSfAu-8c19f0b8iG5rYsl8,42669
+wedata/feature_store/feature_table_client/feature_table_client.py,sha256=a9jI_SkAtzdVHs763YvC1wClAB9SlF0wVZazjJIDqks,43699
 wedata/feature_store/spark_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wedata/feature_store/spark_client/spark_client.py,sha256=aTenEqfZoJYMrph98qjNHZ-H4dgNKnMaH14st8bCVRQ,11797
 wedata/feature_store/training_set_client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/training_set_client/training_set_client.py,sha256=wwxTMGPm9QZXjGFv9XI2hA5N1pHntxhAJ6Grcfg6C2g,23572
+wedata/feature_store/training_set_client/training_set_client.py,sha256=aFx2BLGoKg8OlawS39DQlyyGXqGrS2h62_NsAClD4tU,23186
 wedata/feature_store/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-wedata/feature_store/utils/common_utils.py,sha256=vkpoXxZTd6S-2MgdyTQZ6P_ckdqNSK50ECuVBG2BwfI,12314
-wedata/feature_store/utils/env_utils.py,sha256=kMCl6gyqDxjN5IZ7-wZMR0R4YY-Evh_7NHxugSYPWQc,2736
+wedata/feature_store/utils/common_utils.py,sha256=FbHL0bQ14LpR5NRQc7NhPLFCoBuiCfaKEAW5_BdUWzU,12313
+wedata/feature_store/utils/env_utils.py,sha256=Zj6569qDZR5fhfpKimOzwasaVT04q2c3BIUy5Uctv2g,3160
 wedata/feature_store/utils/feature_lookup_utils.py,sha256=mNV6RhBdpv1iTZduCA9YwXwkeJCwU5MFQ1MkFeD9IhY,22003
 wedata/feature_store/utils/feature_spec_utils.py,sha256=j8t-zel2_r8Q9m88BmFKkHMdkGNIduWJB-28OZDASRY,11613
 wedata/feature_store/utils/feature_utils.py,sha256=KKq28bVB_lCuhnR9Hk6JegJBOVgcelWlvrRM-F9onkA,2796
@@ -58,7 +60,7 @@ wedata/tempo/ml.py,sha256=WtGa2szn6PditvZsTZoxo7wFDe4k1SRoMZ-jgNGIjvE,4323
 wedata/tempo/resample.py,sha256=h81RVVmCl4ect-YKE-KZZHPDi1rGI3sh-YIb-Btz0ck,9698
 wedata/tempo/tsdf.py,sha256=S4lZfxhSRFiezYoYS6gvGsl1mZA3zp-MWEKFHYZpDg0,70968
 wedata/tempo/utils.py,sha256=I9I6l2DMwUoY213L04Yc1UR_zTWgSkj1BVo4ZwzQd4Y,7977
-tencent_wedata_feature_engineering_dev-0.1.47.dist-info/METADATA,sha256=3_S5oTDhWSK_udwh2Kuvp0XXFBCbU_wOJCysFI8BXeQ,582
-tencent_wedata_feature_engineering_dev-0.1.47.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-tencent_wedata_feature_engineering_dev-0.1.47.dist-info/top_level.txt,sha256=Xa0v1rh__RvfVTVDirW5r5UBKg7ZO_iuTeXfp8MNo2A,7
-tencent_wedata_feature_engineering_dev-0.1.47.dist-info/RECORD,,
+tencent_wedata_feature_engineering_dev-0.2.2.dist-info/METADATA,sha256=rvc2LgvO0vPYJQLogsjfx3gDN43xFlSm4UdZSHk_Mj8,732
+tencent_wedata_feature_engineering_dev-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+tencent_wedata_feature_engineering_dev-0.2.2.dist-info/top_level.txt,sha256=Xa0v1rh__RvfVTVDirW5r5UBKg7ZO_iuTeXfp8MNo2A,7
+tencent_wedata_feature_engineering_dev-0.2.2.dist-info/RECORD,,

{tencent_wedata_feature_engineering_dev-0.1.47.dist-info → tencent_wedata_feature_engineering_dev-0.2.2.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any
 

wedata/feature_store/client.py

@@ -288,7 +288,7 @@ class FeatureStoreClient:
         )
 
     def score_batch(
-            self, model_uri: str, df: DataFrame, result_type: str = "double"
+            self, model_uri: str, df: DataFrame, result_type: str = "double", timestamp_key: str = None
     ) -> DataFrame:
         """
         Evaluate the model on the provided :class:`DataFrame <pyspark.sql.DataFrame>`.
@@ -399,9 +399,10 @@ class FeatureStoreClient:
             df=df,
             result_type=result_type,
             client_name=FEATURE_STORE_CLIENT,
+            timestamp_key=timestamp_key,
         )
 
-    def publish_table(self, table_name: str, data_source_name: str, cloud_secret_id: str, cloud_secret_key: str,
+    def publish_table(self, table_name: str, data_source_name: str,
                       database_name: Optional[str] = None,
                       is_cycle: bool = False, cycle_obj: TaskSchedulerConfiguration = None,
                       is_use_default_online: bool = True, online_config: RedisStoreConfig = None):
@@ -414,8 +415,6 @@ class FeatureStoreClient:
         Args:
             table_name: Name of the offline feature table
             data_source_name: Name of the data source
-            cloud_secret_id: Cloud secret ID for authentication
-            cloud_secret_key: Cloud secret key for authentication
             database_name: Database name (optional)
             is_cycle: Whether to enable periodic publishing (default: False)
             cycle_obj: Periodic task configuration object (required if is_cycle is True)
@@ -428,8 +427,6 @@ class FeatureStoreClient:
         """
         return self._feature_table_client.publish_table(table_name=table_name, database_name=database_name,
                                                         data_source_name=data_source_name,
-                                                        cloud_secret_key=cloud_secret_key,
-                                                        cloud_secret_id=cloud_secret_id,
                                                         is_cycle=is_cycle, cycle_obj=cycle_obj,
                                                         is_use_default_online=is_use_default_online,
                                                         online_config=online_config)

wedata/feature_store/cloud_sdk_client/client.py

@@ -4,8 +4,9 @@ from tencentcloud.wedata.v20210820.wedata_client import WedataClient
 from tencentcloud.wedata.v20250806.wedata_client import WedataClient as WedataClientV2
 from tencentcloud.common import credential
 from tencentcloud.common.exception.tencent_cloud_sdk_exception import TencentCloudSDKException
-from wedata.feature_store.cloud_sdk_client.utils import get_client_profile, set_request_header
+from wedata.feature_store.cloud_sdk_client.utils import get_client_profile, set_request_header, is_mock
 import wedata.feature_store.cloud_sdk_client.models as models
+from wedata.feature_store.common.log.logger import get_logger
 
 
 class FeatureCloudSDK:
@@ -22,17 +23,21 @@ class FeatureCloudSDK:
         Returns:
             创建结果响应
         """
+        logger = get_logger()
+        if is_mock():
+            logger.debug("Mock CreateOnlineFeatureTable API")
+            return models.CreateOnlineFeatureTableResponse()
         try:
             params = request._serialize()
             headers = set_request_header(request.headers)
-            print(f"CreateOnlineFeatureTable params: {params}")
-            print(f"CreateOnlineFeatureTable headers: {headers}")
+            logger.debug(f"CreateOnlineFeatureTable params: {params}")
+            logger.debug(f"CreateOnlineFeatureTable headers: {headers}")
             self._client._apiVersion = "2021-08-20"
             body = self._client.call("CreateOnlineFeatureTable", params, headers=headers)
             response = json.loads(body)
             model = models.CreateOnlineFeatureTableResponse()
             model._deserialize(response["Response"])
-            print("CreateOnlineFeatureTable Response: ", response)
+            logger.debug(f"CreateOnlineFeatureTable Response: {response}")
             return model
         except Exception as e:
             if isinstance(e, TencentCloudSDKException):
@@ -49,17 +54,22 @@ class FeatureCloudSDK:
         Returns:
             查询结果响应
         """
+        logger = get_logger()
+        if is_mock():
+            logger.debug("Mock DescribeNormalSchedulerExecutorGroups API")
+            return models.DescribeNormalSchedulerExecutorGroupsResponse()
+
         try:
             params = request._serialize()
             headers = set_request_header(request.headers)
-            print(f"DescribeNormalSchedulerExecutorGroups params: {params}")
-            print(f"DescribeNormalSchedulerExecutorGroups headers: {headers}")
+            logger.debug(f"DescribeNormalSchedulerExecutorGroups params: {params}")
+            logger.debug(f"DescribeNormalSchedulerExecutorGroups headers: {headers}")
             self._client._apiVersion = "2021-08-20"
             body = self._client.call("DescribeNormalSchedulerExecutorGroups", params, headers=headers)
             response = json.loads(body)
             model = models.DescribeNormalSchedulerExecutorGroupsResponse()
             model._deserialize(response["Response"])
-            print("DescribeNormalSchedulerExecutorGroups Response: ", response)
+            logger.debug(f"DescribeNormalSchedulerExecutorGroups Response: {response}")
             return model
         except Exception as e:
             if isinstance(e, TencentCloudSDKException):
@@ -75,20 +85,54 @@ class FeatureCloudSDK:
         Returns:
             刷新结果响应
         """
+        logger = get_logger()
+        if is_mock():
+            logger.debug("Mock RefreshFeatureTable API")
+            return models.RefreshFeatureTableResponse()
         try:
             params = request._serialize()
             headers = set_request_header(request.headers)
-            print(f"RefreshFeatureTable params: {params}")
-            print(f"RefreshFeatureTable headers: {headers}")
+            logger.debug(f"RefreshFeatureTable params: {params}")
+            logger.debug(f"RefreshFeatureTable headers: {headers}")
             self._client_v2._apiVersion = "2025-08-06"
             body = self._client_v2.call("RefreshFeatureTable", params, headers=headers)
             response = json.loads(body)
             model = models.RefreshFeatureTableResponse()
             model._deserialize(response["Response"])
-            print("RefreshFeatureTable Response: ", response)
+            logger.debug(f"RefreshFeatureTable Response: {response}")
+            return model
+        except Exception as e:
+            if isinstance(e, TencentCloudSDKException):
+                raise
+            else:
+                raise TencentCloudSDKException(type(e).__name__, str(e))
+
+    def DescribeFeatureStoreDatabases(self, request: models.DescribeFeatureStoreDatabasesRequest) -> 'models.DescribeFeatureStoreDatabasesResponse':
+        """
+        查询特征库列表
+        Args:
+            request: 查询请求参数
+        Returns:
+            查询结果响应
+        """
+        logger = get_logger()
+        if is_mock():
+            logger.debug("Mock DescribeFeatureStoreDatabases API")
+            return models.DescribeFeatureStoreDatabasesResponse()
+        try:
+            params = request._serialize()
+            headers = set_request_header(request.headers)
+            logger.debug(f"DescribeFeatureStoreDatabases params: {params}")
+            logger.debug(f"DescribeFeatureStoreDatabases headers: {headers}")
+            self._client_v2._apiVersion = "2021-08-20"
+            body = self._client_v2.call("DescribeFeatureStoreDatabases", params, headers=headers)
+            response = json.loads(body)
+            model = models.DescribeFeatureStoreDatabasesResponse()
+            model._deserialize(response["Response"])
+            logger.debug(f"DescribeFeatureStoreDatabases Response: {response}")
             return model
         except Exception as e:
             if isinstance(e, TencentCloudSDKException):
                 raise
             else:
-                raise TencentCloudSDKException(type(e).__name__, str(e))
+                raise TencentCloudSDKException(type(e).__name__, str(e))

wedata/feature_store/cloud_sdk_client/models.py

@@ -1,5 +1,8 @@
+
 from tencentcloud.common.abstract_model import AbstractModel
+import typing
 import warnings
+from wedata.feature_store.cloud_sdk_client.utils import is_warning
 
 
 class OfflineFeatureConfiguration(AbstractModel):
@@ -85,8 +88,8 @@ class OfflineFeatureConfiguration(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class TaskSchedulerConfiguration(AbstractModel):
@@ -190,8 +193,8 @@ class TaskSchedulerConfiguration(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class OnlineFeatureConfiguration(AbstractModel):
@@ -257,8 +260,8 @@ class OnlineFeatureConfiguration(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class CreateOnlineFeatureTableRequest(AbstractModel):
@@ -337,8 +340,8 @@ class CreateOnlineFeatureTableRequest(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class CreateOnlineFeatureTableRsp(AbstractModel):
@@ -374,8 +377,8 @@ class CreateOnlineFeatureTableRsp(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class CreateOnlineFeatureTableResponse(AbstractModel):
@@ -402,8 +405,8 @@ class CreateOnlineFeatureTableResponse(AbstractModel):
             property_name = name[1:]
             if property_name in member_set:
                 member_set.remove(property_name)
-        if len(member_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(member_set))
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class DescribeNormalSchedulerExecutorGroupsData(AbstractModel):
@@ -480,12 +483,12 @@ class DescribeNormalSchedulerExecutorGroupsData(AbstractModel):
         self._Available = params.get("Available")
         self._PythonSubVersions = params.get("PythonSubVersions")
         self._EnvJson = params.get("EnvJson")
-        memeber_set = set(params.keys())
+        member_set = set(params.keys())
         for name, value in vars(self).items():
-            if name in memeber_set:
-                memeber_set.remove(name)
-        if len(memeber_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(memeber_set))
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class DescribeNormalSchedulerExecutorGroupsResponse(AbstractModel):
@@ -511,12 +514,12 @@ class DescribeNormalSchedulerExecutorGroupsResponse(AbstractModel):
                 obj = DescribeNormalSchedulerExecutorGroupsData()
                 obj._deserialize(item)
                 self._Data.append(obj)
-        memeber_set = set(params.keys())
+        member_set = set(params.keys())
         for name, value in vars(self).items():
-            if name in memeber_set:
-                memeber_set.remove(name)
-        if len(memeber_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(memeber_set))
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class DescribeNormalSchedulerExecutorGroupsRequest(AbstractModel):
@@ -547,12 +550,12 @@ class DescribeNormalSchedulerExecutorGroupsRequest(AbstractModel):
     def _deserialize(self, params):
         self._ProjectId = params.get("ProjectId")
         self._OnlyAvailable = params.get("OnlyAvailable")
-        memeber_set = set(params.keys())
+        member_set = set(params.keys())
         for name, value in vars(self).items():
-            if name in memeber_set:
-                memeber_set.remove(name)
-        if len(memeber_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(memeber_set))
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class RefreshFeatureTableRequest(AbstractModel):
@@ -651,12 +654,12 @@ class RefreshFeatureTableRequest(AbstractModel):
         self._DatasourceType = params.get("DatasourceType")
         self._EngineName = params.get("EngineName")
         self._IsTry = params.get("IsTry")
-        memeber_set = set(params.keys())
+        member_set = set(params.keys())
         for name, value in vars(self).items():
-            if name in memeber_set:
-                memeber_set.remove(name)
-        if len(memeber_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(memeber_set))
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
 
 
 class RefreshFeatureTableResponse(AbstractModel):
@@ -678,9 +681,181 @@ class RefreshFeatureTableResponse(AbstractModel):
 
     def _deserialize(self, params):
         self._Data = params.get("Data")
-        memeber_set = set(params.keys())
+        member_set = set(params.keys())
+        for name, value in vars(self).items():
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
+
+
+class FeatureStoreDatabase(AbstractModel):
+    """
+    特征存储库
+    Property:
+        DatabaseName: 特征库名称
+        DatasourceType：数据源类型: EMR/DLC
+        EngineName: 引擎名称
+        ProjectId: 项目ID
+        IsDefault: 是否默认库
+        IsExistDatabase: 是否存在库
+        DatasourceId: 数据源ID
+        OnlineMode: 在线模式: 0-离线; 1-在线
+        DatasourceName: 数据源名称
+    """
+    def __init__(self):
+        self._DatabaseName = None
+        self._DatasourceType = None
+        self._EngineName = None
+        self._ProjectId = None
+        self._IsDefault = None
+        self._IsExistDatabase = None
+        self._DatasourceId = None
+        self._OnlineMode = None
+        self._DatasourceName = None
+
+    @property
+    def DatabaseName(self):
+        return self._DatabaseName
+
+    @DatabaseName.setter
+    def DatabaseName(self, DatabaseName):
+        self._DatabaseName = DatabaseName
+
+    @property
+    def DatasourceType(self):
+        return self._DatasourceType
+
+    @DatasourceType.setter
+    def DatasourceType(self, DatasourceType):
+        self._DatasourceType = DatasourceType
+
+    @property
+    def EngineName(self):
+        return self._EngineName
+
+    @EngineName.setter
+    def EngineName(self, EngineName):
+        self._EngineName = EngineName
+
+    @property
+    def ProjectId(self):
+        return self._ProjectId
+
+    @ProjectId.setter
+    def ProjectId(self, ProjectId):
+        self._ProjectId = ProjectId
+
+    @property
+    def IsDefault(self):
+        return self._IsDefault
+
+    @IsDefault.setter
+    def IsDefault(self, IsDefault):
+        self._IsDefault = IsDefault
+
+    @property
+    def IsExistDatabase(self):
+        return self._IsExistDatabase
+
+    @IsExistDatabase.setter
+    def IsExistDatabase(self, IsExistDatabase):
+        self._IsExistDatabase = IsExistDatabase
+
+    @property
+    def DatasourceId(self):
+        return self._DatasourceId
+
+    @DatasourceId.setter
+    def DatasourceId(self, DatasourceId):
+        self._DatasourceId = DatasourceId
+
+    @property
+    def OnlineMode(self):
+        return self._OnlineMode
+
+    @OnlineMode.setter
+    def OnlineMode(self, OnlineMode):
+        self._OnlineMode = OnlineMode
+
+    @property
+    def DatasourceName(self):
+        return self._DatasourceName
+
+    @DatasourceName.setter
+    def DatasourceName(self, DatasourceName):
+        self._DatasourceName = DatasourceName
+
+    def _deserialize(self, params):
+        self._DatabaseName = params.get("DatabaseName")
+        self._DatasourceType = params.get("DatasourceType")
+        self._EngineName = params.get("EngineName")
+        self._ProjectId = params.get("ProjectId")
+        self._IsDefault = params.get("IsDefault")
+        self._IsExistDatabase = params.get("IsExistDatabase")
+        self._DatasourceId = params.get("DatasourceId")
+        self._OnlineMode = params.get("OnlineMode")
+        self._DatasourceName = params.get("DatasourceName")
+        member_set = set(params.keys())
+        for name, value in vars(self).items():
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
+
+
+class DescribeFeatureStoreDatabasesResponse(AbstractModel):
+    """
+    描述特征库
+    Property:
+        Data: 结果
+    """
+    def __init__(self):
+        self._Data = None
+
+    @property
+    def Data(self) -> typing.List[FeatureStoreDatabase]:
+        return self._Data
+
+    @Data.setter
+    def Data(self, Data):
+        self._Data = Data
+
+    def _deserialize(self, params):
+        self._Data = []
+        for item in params.get("Data", []):
+            obj = FeatureStoreDatabase()
+            obj._deserialize(item)
+            self._Data.append(obj)
+        member_set = set(params.keys())
+        for name, value in vars(self).items():
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))
+
+
+class DescribeFeatureStoreDatabasesRequest(AbstractModel):
+    """
+    Property:
+       ProjectId: 项目ID
+    """
+    def __init__(self):
+        self._ProjectId = None
+
+    @property
+    def ProjectId(self):
+        return self._ProjectId
+
+    @ProjectId.setter
+    def ProjectId(self, ProjectId):
+        self._ProjectId = ProjectId
+
+    def _deserialize(self, params):
+        self._ProjectId = params.get("ProjectId")
+        member_set = set(params.keys())
         for name, value in vars(self).items():
-            if name in memeber_set:
-                memeber_set.remove(name)
-        if len(memeber_set) > 0:
-            warnings.warn("%s fileds are useless." % ",".join(memeber_set))
+            if name in member_set:
+                member_set.remove(name)
+        if len(member_set) > 0 and is_warning():
+            warnings.warn("%s fields are useless." % ",".join(member_set))

wedata/feature_store/cloud_sdk_client/utils.py

@@ -30,3 +30,17 @@ def set_request_header(headers):
         headers["X-Qcloud-User-Id"] = os.environ.get("TEST_USER_ID")
     return headers
 
+
+def is_mock() -> bool:
+    """
+    是否为模拟环境
+    """
+    return os.getenv("IS_MOCK_API") == "true"
+
+
+def is_warning() -> bool:
+    """
+    是否展示警告环境
+    """
+    return os.getenv("IS_CLOUD_API_WARNING") == "true"
+

wedata/feature_store/common/log/logger.py

@@ -0,0 +1,44 @@
+import logging
+import sys
+
+
+class LoggerSingleton:
+    _instance = None
+    
+    def __new__(cls):
+        if cls._instance is None:
+            cls._instance = super().__new__(cls)
+            cls._instance._initialize_logger()
+        return cls._instance
+    
+    def _initialize_logger(self):
+        self.logger = logging.getLogger("wedata-feature-engineering")
+        self.logger.setLevel(logging.INFO)
+
+        # 清除已有的handler，避免重复添加
+        if self.logger.handlers:
+            self.logger.handlers.clear()
+        
+        # 创建formatter，包含时间、文件名和行号
+        formatter = logging.Formatter(
+            fmt='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+            datefmt='%Y-%m-%d %H:%M:%S'
+        )
+        
+        # 创建handler并输出到stdout
+        handler = logging.StreamHandler(sys.stdout)
+        handler.setFormatter(formatter)
+        
+        self.logger.addHandler(handler)
+        
+        # 防止消息传播到父级logger
+        self.logger.propagate = False
+    
+    def get_logger(self, level=logging.INFO):
+        self.logger.setLevel(level)
+        return self.logger
+
+
+def get_logger(level=logging.INFO):
+    """获取单例logger实例"""
+    return LoggerSingleton().get_logger(level)

wedata/feature_store/constants/constants.py

@@ -43,7 +43,7 @@ ML_MODEL = "MLmodel"
 FEATURE_LOOKUP_CLIENT_PIP_PACKAGE = "tencent-wedata-feature-engineering-dev"
 
 # 特征查找版本号
-FEATURE_LOOKUP_CLIENT_MAJOR_VERSION = "0.1.47"
+FEATURE_LOOKUP_CLIENT_MAJOR_VERSION = "0.2.2"
 
 # 特征存储内部数据目录
 FEATURE_STORE_INTERNAL_DATA_DIR = "_wedata_internal/"

wedata/feature_store/feature_table_client/feature_table_client.py

@@ -15,14 +15,13 @@ from wedata.feature_store.constants.constants import (
     FEATURE_TABLE_VALUE, FEATURE_TABLE_PROJECT, FEATURE_TABLE_TIMESTAMP,
     FEATURE_TABLE_BACKUP_PRIMARY_KEY, FEATURE_DLC_TABLE_PRIMARY_KEY)
 from wedata.feature_store.constants.engine_types import EngineTypes
+from wedata.feature_store.common.log.logger import get_logger
 from wedata.feature_store.common.store_config.redis import RedisStoreConfig
 from wedata.feature_store.entities.feature_table import FeatureTable
 from wedata.feature_store.spark_client.spark_client import SparkClient
 from wedata.feature_store.utils import common_utils, env_utils
 from wedata.feature_store.feast_client.feast_client import FeastClient
-from wedata.feature_store.cloud_sdk_client.models import (
-    TaskSchedulerConfiguration, OnlineFeatureConfiguration, OfflineFeatureConfiguration,
-    CreateOnlineFeatureTableRequest, DescribeNormalSchedulerExecutorGroupsRequest, RefreshFeatureTableRequest)
+from wedata.feature_store.cloud_sdk_client import models
 from wedata.feature_store.cloud_sdk_client.client import FeatureCloudSDK
 
 
@@ -44,6 +43,11 @@ class FeatureTableClient:
             self.__cloud_secret_id, self.__cloud_secret_key = env_utils.get_cloud_secret()
         self.__project = env_utils.get_project_id()
         self.__region = env_utils.get_region()
+        self.__logger = get_logger()
+        default_online_table = self._get_offline_default_database()
+        if default_online_table:
+            env_utils.set_default_database(default_online_table.DatabaseName)
+
 
     @property
     def cloud_secret_id(self) -> str:
@@ -310,7 +314,7 @@ class FeatureTableClient:
             raise ValueError(f"Engine type {engine_type} is not supported")
 
         # 打印sql
-        print(f"create table ddl: {ddl}\n")
+        self.__logger.info(f"create table ddl: {ddl}\n")
 
         # 执行DDL
         try:
@@ -320,8 +324,6 @@ class FeatureTableClient:
         except Exception as e:
             raise ValueError(f"Failed to create table: {str(e)}") from e
 
-        print("async table info to feast")
-
         self._feast_client.create_table(
             table_name=table_name,
             primary_keys=primary_keys,
@@ -332,7 +334,7 @@ class FeatureTableClient:
             description=description
         )
 
-        print(f"create table {name} done")
+        self.__logger.info(f"Table '{name}' created successfully. Starting web synchronization.")
 
         try:
             self._sync_table_info(table_name=name, action_name="create",
@@ -340,7 +342,8 @@ class FeatureTableClient:
                                   data_source_name=data_source_name, engine_name=env_utils.get_engine_name(),
                                   is_try=False)
         except tencentcloud.common.exception.TencentCloudSDKException as e:
-            raise RuntimeError(f"Table '{name}' is can't create. {str(e)}")
+            raise RuntimeError(f"Failed to synchronize web data for table '{name}'. "
+                               f"Please manually operate on the web page. Error: {str(e)}")
 
         # 构建并返回FeatureTable对象
         return FeatureTable(
@@ -533,21 +536,22 @@ class FeatureTableClient:
                 raise RuntimeError(f"Table '{name}' is can't create. {str(e)}")
 
             # 执行修改
-            print("alter table sql", alter_sql)
+            self.__logger.info(f"alter table sql: \n {alter_sql}")
             self._spark.sql(alter_sql)
-            print("Execute sql done, start sync table info to feast")
+            self.__logger.debug("Execute sql done, start sync table info to feast")
             self._feast_client.alter_table(full_table_name=table_name, primary_keys=primary_keys,
                                            timestamp_key=timestamp_key)
-            print(f"Successfully register table '{table_name}'")
+            self.__logger.info(f"Successfully register table '{table_name}'. Starting web synchronization.")
 
             try:
                 self._sync_table_info(table_name=name, action_name="create",
                                   database_name=env_utils.get_database_name(database_name),
                                   data_source_name=data_source_name, engine_name=env_utils.get_engine_name(), is_try=False)
             except tencentcloud.common.exception.TencentCloudSDKException as e:
-                raise RuntimeError(f"sync table info failed. you need to sync table info manually. {str(e)}")
-        except ValueError as e:
-            raise  # 直接抛出已知的ValueError
+                raise RuntimeError(f"Failed to synchronize web data for table '{name}'. "
+                                   f"Please manually operate on the web page. Error: {str(e)}")
+        except (ValueError, RuntimeError):
+            raise
         except Exception as e:
             raise RuntimeError(f"Failed to modify properties for table '{table_name}': {str(e)}") from e
 
@@ -624,13 +628,14 @@ class FeatureTableClient:
         try:
             # 检查表是否存在
             if not self._check_table_exists(table_name):
-                print(f"Table '{name}' does not exist")
+                self.__logger.error(f"Table '{name}' does not exist")
                 return
 
             try:
                 feature_view = self._feast_client.get_feature_view(table_name)
             except Exception as e:
-                print(f"Table '{name}' is not a feature table, skip delete. {str(e)}")
+                pass
+                # self.__logger.warning(f"Table '{name}' is not a feature table, skip delete. {str(e)}")
             else:
                 if feature_view.online:
                     raise ValueError(f"Table '{name}' has a online table, please call drop_online_table first")
@@ -643,14 +648,14 @@ class FeatureTableClient:
 
             # 执行删除
             self._spark.sql(f"DROP TABLE {table_name}")
-            print(f"Table '{name}' dropped")
+            self.__logger.info(f"Table '{name}' dropped")
             try:
                 self._feast_client.remove_offline_table(table_name=table_name)
             except Exception as e:
                 raise
                 # raise ValueError(f"Failed to delete table '{name}' in feast: {str(e)}")
             else:
-                print(f"Table '{name}' removed from feast")
+                self.__logger.info(f"Table '{name}' removed from feast")
 
             try:
                 self._sync_table_info(table_name=name, action_name="delete",
@@ -706,7 +711,7 @@ class FeatureTableClient:
         schema_name_list = [field.name for field in tmp_schema_list]
         schema = StructType(tmp_schema_list)
         for field in schema:
-            print(f"{field.name} => {field.dataType}")
+            self.__logger.debug(f"translate {field.name} to feast Type: {field.dataType}")
 
         feast_client = FeastClient(offline_store=self._spark, online_store_config=online_config)
         # 构建离线表的entity的数据过滤
@@ -721,7 +726,7 @@ class FeatureTableClient:
                     full_table_name=full_table_name,
                     columns_name=columns_name_list,
                     entity_rows=[result_row.asDict()])
-                print("=====>read online dataframe:\n", online_view[schema_name_list])
+                self.__logger.debug(f"=====>read online dataframe:\n{online_view[schema_name_list]}")
                 return self._spark.createDataFrame(online_view[schema_name_list], schema=schema, verifySchema=False)
             else:
                 return self._spark.createDataFrame([])
@@ -730,7 +735,7 @@ class FeatureTableClient:
                 full_table_name=full_table_name,
                 columns_name=columns_name_list,
                 entity_rows=entity_row)
-            print("=====>read online dataframe:\n", online_view[schema_name_list])
+            self.__logger.debug(f"=====>read online dataframe:\n{online_view[schema_name_list]}")
             return self._spark.createDataFrame(online_view[schema_name_list], schema=schema, verifySchema=False)
 
     def get_table(
@@ -827,9 +832,9 @@ class FeatureTableClient:
         except Exception as e:
             raise RuntimeError(f"Failed to modify properties for table '{name}': {str(e)}") from e
 
-    def publish_table(self, table_name: str, data_source_name: str, cloud_secret_id: str, cloud_secret_key: str,
+    def publish_table(self, table_name: str, data_source_name: str,
                       database_name: Optional[str] = None,
-                      is_cycle: bool = False, cycle_obj: TaskSchedulerConfiguration = None,
+                      is_cycle: bool = False, cycle_obj: models.TaskSchedulerConfiguration = None,
                       is_use_default_online: bool = True, online_config: RedisStoreConfig = None):
         """
         将离线特征表发布为在线特征表
@@ -852,46 +857,43 @@ class FeatureTableClient:
         # 检查是否已经发布,查看Redis中是否有值
         try:
         # 获取离线表的列名
-            online_data = self._read_online_table(
-                table_name=table_name,
-                database_name=database_name,
-                online_config=online_config)
+            online_data = self._feast_client.get_feature_view(full_table_name)
         except Exception as e:
             print(f"Failed to get online table view for table '{full_table_name}': {str(e)}")
         else:
-            if online_data:
+            if online_data.online:
                 raise ValueError(f"Table '{full_table_name}' has already been published")
 
         # 配置周期性参数
         if is_cycle:
-            if not isinstance(cycle_obj, TaskSchedulerConfiguration):
+            if not isinstance(cycle_obj, models.TaskSchedulerConfiguration):
                 raise ValueError("cycle_obj must be a TaskSchedulerConfiguration object when is_cycle is True")
 
             cycle_obj.CycleType = "CRONTAB_CYCLE"
         else:
-            if isinstance(cycle_obj, TaskSchedulerConfiguration):
+            if isinstance(cycle_obj, models.TaskSchedulerConfiguration):
                 cycle_obj.CycleType = "ONEOFF_CYCLE"
             else:
-                cycle_obj = TaskSchedulerConfiguration()
+                cycle_obj = models.TaskSchedulerConfiguration()
                 cycle_obj.CycleType = "ONEOFF_CYCLE"
                 # 设置默认当前时间延后1分钟
                 cycle_obj.CrontabExpression = (datetime.datetime.now() + datetime.timedelta(minutes=3)).strftime(
                     "%M %H %d %m %w ? %y")
 
         if is_use_default_online:
-            online_feature_config = OnlineFeatureConfiguration()
+            online_feature_config = models.OnlineFeatureConfiguration()
             online_feature_config.UserDefault = True
         else:
             if not isinstance(online_config, RedisStoreConfig):
                 raise ValueError("online_config must be a RedisStoreConfig object when is_use_default_online is False")
 
-            online_feature_config = OnlineFeatureConfiguration()
+            online_feature_config = models.OnlineFeatureConfiguration()
             online_feature_config.UserDefault = False
             online_feature_config.Host = online_config.host
             online_feature_config.Port = online_config.port
             online_feature_config.DB = online_config.db
 
-        offline_feature_config = OfflineFeatureConfiguration()
+        offline_feature_config = models.OfflineFeatureConfiguration()
         offline_feature_config.DatabaseName = env_utils.get_database_name(database_name)
         offline_feature_config.TableName = table_name
 
@@ -902,7 +904,7 @@ class FeatureTableClient:
         offline_feature_config.DatasourceType = env_utils.get_engine_type()
         offline_feature_config.EngineName = env_utils.get_engine_name()
 
-        api_requests = CreateOnlineFeatureTableRequest()
+        api_requests = models.CreateOnlineFeatureTableRequest()
         api_requests.OfflineFeatureConfiguration = offline_feature_config
         api_requests.OnlineFeatureConfiguration = online_feature_config
         api_requests.TaskSchedulerConfiguration = cycle_obj
@@ -910,11 +912,11 @@ class FeatureTableClient:
         region = env_utils.get_region()
         if not os.environ.get("RESOURCE_GROUP_ID", ""):
             res_group_item = _get_default_resource_group(
-                api_requests.ProjectId, cloud_secret_id, cloud_secret_key, region)
+                api_requests.ProjectId, self.__cloud_secret_id, self.__cloud_secret_key, region)
             api_requests.ResourceGroupId = res_group_item.ExecutorGroupId
         else:
             api_requests.ResourceGroupId = os.environ.get("RESOURCE_GROUP_ID")
-        client = FeatureCloudSDK(secret_id=cloud_secret_id, secret_key=cloud_secret_key, region=region)
+        client = FeatureCloudSDK(secret_id=self.__cloud_secret_id, secret_key=self.__cloud_secret_key, region=region)
         resp = client.CreateOnlineFeatureTable(api_requests)
         if cycle_obj.CycleType == "ONEOFF_CYCLE":
             print(f"publish online task create success. it will be execute after 3 min. {resp.Data.OnlineTableId} {resp.Data.OfflineTableId} ")
@@ -955,10 +957,23 @@ class FeatureTableClient:
     def _check_table_exists(self, full_table_name: str) -> bool:
         return common_utils.check_spark_table_exists(self._spark, full_table_name)
 
+    def _get_offline_default_database(self) -> Optional[models.FeatureStoreDatabase]:
+        client = FeatureCloudSDK(secret_id=self.__cloud_secret_id, secret_key=self.__cloud_secret_key,
+                                 region=self.__region)
+        req = models.DescribeFeatureStoreDatabasesRequest()
+        req.ProjectId = self.__project
+        rsp = client.DescribeFeatureStoreDatabases(req)
+        if len(rsp.Data) == 0:
+            return None
+        for item in rsp.Data:
+            if item.OnlineMode == 0 and item.IsDefault == 1:
+                return item
+        return None
+
 
 def _get_default_resource_group(project_id: str, secret_id: str, secret_key: str, region: str):
     client = FeatureCloudSDK(secret_id=secret_id, secret_key=secret_key, region=region)
-    request = DescribeNormalSchedulerExecutorGroupsRequest()
+    request = models.DescribeNormalSchedulerExecutorGroupsRequest()
     request.ProjectId = project_id
     resp = client.DescribeNormalSchedulerExecutorGroups(request)
     # 默认取第一个健康可用的资源组进行执行
@@ -972,7 +987,7 @@ def _refresh_table(project_id: str, secret_id: str, secret_key: str, region: str
                    action: str, database_name: str, data_source_name: str, data_source_type: str,
                    engine_name: str, is_try: bool):
     client = FeatureCloudSDK(secret_id=secret_id, secret_key=secret_key, region=region)
-    request = RefreshFeatureTableRequest()
+    request = models.RefreshFeatureTableRequest()
     request.ProjectId = project_id
     request.TableName = table_name
     request.DatabaseName = database_name

wedata/feature_store/training_set_client/training_set_client.py

@@ -360,6 +360,7 @@ class TrainingSetClient:
         env_manager: Optional[str] = None,
         local_uri: Optional[str] = None,
         params: Optional[dict[str, Any]] = None,
+        timestamp_key: str = None,
         **kwargs,
     ) -> DataFrame:
         # TODO:ML 待确定是否需要
@@ -388,16 +389,13 @@ class TrainingSetClient:
             "The provided DataFrame for scoring must have unique column names. Found duplicates {}.",
         )
         artifact_path = os.path.join("artifacts", MODEL_DATA_PATH_ROOT)
-        # print(f"artifact_path: {artifact_path}")
         with (TempDir() as tmp_location):
             local_path = (
                 local_uri
                 if local_uri
                 else common_utils.download_model_artifacts(model_uri, tmp_location.path())
             )
-            # print(f"wedata local_path:{local_path}")
             model_data_path = os.path.join(local_path, artifact_path)
-            # print(f"artifact_path: {artifact_path}")
 
         # Augment local workspace metastore tables from 2L to 3L,
             # this will prevent us from erroneously reading data from other catalogs
@@ -425,6 +423,7 @@ class TrainingSetClient:
         feature_input_keys = []
         for fci in feature_spec.feature_column_infos:
             feature_input_keys.extend([k for k in fci.lookup_key])
+
         on_demand_input_names = uc_utils.get_unique_list_order(
             [
                 input_name
@@ -436,18 +435,13 @@ class TrainingSetClient:
         source_data_names = [
             sdci.name for sdci in feature_spec.source_data_column_infos
         ]
-        # print(f"wedata source_data_names:{source_data_names}")
-
-        print("===>source_data_names:", source_data_names)
 
         feature_output_names = [
             fci.output_name for fci in feature_spec.feature_column_infos
         ]
-        print("====>feature_output_names:", feature_output_names)
         on_demand_output_names = [
             odci.output_name for odci in feature_spec.on_demand_column_infos
         ]
-        print("====>on_demand_output_names:", on_demand_output_names)
         all_output_names = set(
             source_data_names + feature_output_names + on_demand_output_names
         )

wedata/feature_store/utils/common_utils.py

@@ -16,13 +16,9 @@ from mlflow.utils import databricks_utils
 
 from wedata.feature_store.constants import constants
 from wedata.feature_store.constants.constants import MODEL_DATA_PATH_ROOT
+from wedata.feature_store.common.log.logger import get_logger
 from pyspark.sql import SparkSession
 
-import logging
-
-# 配置日志（可选，根据实际情况配置）
-logging.basicConfig(level=logging.ERROR)
-
 
 def validate_table_name(name: str):
     """
@@ -57,14 +53,16 @@ def build_full_table_name(table_name: str, database_name: Optional[str] = None)
     """
 
     feature_store_database_name = os.environ.get("WEDATA_DEFAULT_FEATURE_STORE_DATABASE")
+    logger = get_logger()
     if database_name:
         feature_store_database_name = database_name
 
     if not feature_store_database_name:
-        logging.error("The current user has not configured a default feature database. Please contact the manager account to configure it.")
+        logger.error("The current user has not configured a default feature database. "
+                     "Please contact the manager account to configure it.")
         raise RuntimeError("Feature store is not configured! Please contact the main account to configure it.")
 
-    logging.info("feature database:{}".format(feature_store_database_name))
+    logger.debug("feature database:{}".format(feature_store_database_name))
 
     feature_store_database = f"{feature_store_database_name}.{table_name}"
 

wedata/feature_store/utils/env_utils.py

@@ -68,10 +68,22 @@ def get_database_name(database_name: str) -> str:
                            "please check environment configuration")
 
 
+def set_default_database(database_name: str):
+    """
+    设置默认数据库名称
+    """
+    if not isinstance(database_name, str):
+        raise ValueError("database_name must be a string")
+    os.environ["WEDATA_DEFAULT_FEATURE_STORE_DATABASE"] = database_name
+
+
 def get_engine_name() -> str:
     """
     获取引擎名称
     """
+    # 因为DLC有特殊，所以先判断DLC，如果没有再判断EMR
+    if get_engine_type() == "DLC":
+        return _get_variable("KERNEL_ENGINE")
     return _get_variable("KERNEL_ENGINE_NAME")