teradataml 20.0.0.4__py3-none-any.whl → 20.0.0.6__py3-none-any.whl

teradataml/automl/model_training.py

@@ -775,7 +775,7 @@ class _ModelTraining:
 
         # Defining training data
         data_types = ['lasso', 'rfe', 'pca']
-        trainng_datas = tuple(DataFrame(self.table_name_mapping[f'{data_type}_train']) for data_type in data_types)
+        trainng_datas = tuple(DataFrame(self.data_mapping[f'{data_type}_train']) for data_type in data_types)
 
         if self.task_type == "Classification":
             response_values = trainng_datas[0].get(self.target_column).drop_duplicate().get_values().flatten().tolist()

teradataml/clients/auth_client.py

@@ -1,6 +1,6 @@
 """
 Unpublished work.
-Copyright (c) 2024 by Teradata Corporation. All rights reserved.
+Copyright (c) 2025 by Teradata Corporation. All rights reserved.
 TERADATA CORPORATION CONFIDENTIAL AND TRADE SECRET
 Primary Owner: gouri.patwardhan@teradata.com
 Secondary Owner: Pradeep.Garre@teradata.com
@@ -33,6 +33,8 @@ class _AuthWorkflow:
                     5. "username" which is the DB user.
                     5. "expiration_time" which is the expiration time for the token and has a default value of
                        31536000 seconds.
+                    6. "valid_from" which states epoch seconds representing time from which JWT token will be valid
+                        and same is used as iat claim in payload.
                 Types: dict
 
         RETURNS:
@@ -52,7 +54,7 @@ class _AuthWorkflow:
             Generate expiry epoch time.
 
         RETURNS:
-            float
+            tuple
         """
         current_epoch_time = int(time.time())
         expiry_epoch_time = current_epoch_time + self.state.get('expiration_time')
@@ -67,7 +69,7 @@ class _AuthWorkflow:
             dict
         """
         # Extract the pem file name without extension.
-        kid = pathlib.Path(self.state.get('pem_file')).stem
+        kid = pathlib.Path(self.state.get('pem_file')).stem if not self.state.get('kid') else self.state['kid']
         header = {
             "alg": "RS256",
             "kid": kid,
@@ -83,12 +85,11 @@ class _AuthWorkflow:
         RETURNS:
             A dictionary with the JWT payload.
         """
-        iat, exp= self._get_epoch_time()
+        _, exp = self._get_epoch_time()
         payload = {
             "aud": [
                 "td:service:authentication"
             ],
-            "iat": iat,
             "exp": exp,
             "iss": "teradataml",
             "multi-use": True,
@@ -96,6 +97,9 @@ class _AuthWorkflow:
             "pat": self.state['pat_token'],
             "sub": self.state['username']
         }
+        # Add iat if required.
+        if self.state['valid_from']:
+            payload.update({"iat": self.state['valid_from']})
         return payload
 
     def _sign_jwt(self, payload, header):
@@ -128,6 +132,6 @@ class _AuthWorkflow:
         RETURNS:
             str
         """
-        jwt = self._sign_jwt(self._generate_payload(), self._generate_header())
-        self.state['jwt'] = jwt
-        return(jwt)
+        jwt_token = self._sign_jwt(self._generate_payload(), self._generate_header())
+        self.state['jwt'] = jwt_token
+        return jwt_token

teradataml/clients/keycloak_client.py

@@ -0,0 +1,165 @@
+"""
+Unpublished work.
+Copyright (c) 2025 by Teradata Corporation. All rights reserved.
+TERADATA CORPORATION CONFIDENTIAL AND TRADE SECRET
+Primary Owner: shivani.kondewar@teradata.com
+Secondary Owner: aanchal.kavedia@teradata.com
+This file implements _KeycloakManager class used to store data for keycloak token generation.
+"""
+
+import time
+from enum import Enum
+
+from teradataml import TeradataMlException, MessageCodes
+from teradataml.common.constants import HTTPRequest
+from teradataml.context.context import _get_user
+
+
+class GrantType(Enum):
+    PASS = "password"
+    REFRESH = "refresh_token"
+    CLIENT = "client_credentials"
+
+
+class _KeycloakManager():
+
+    def __init__(self, auth_url, client_id):
+        """
+        DESCRIPTION:
+            Constructor to initiate keycloak manager.
+
+        PARAMETERS:
+            auth_url:
+                Required Argument.
+                Specifies the endpoint URL for a keycloak server.
+                Types: str
+
+            client_id:
+                Required Argument.
+                Specifies the client/service for which keycloak server avails a token.
+                Types: str
+
+        RETURNS:
+            Instance of _KeycloakManager.
+
+        RAISES:
+            None
+
+        EXAMPLES:
+            >>> token_generator = _KeycloakManager(auth_url=auth_url,
+            ...                                    username=username,
+            ...                                    client_id=client_id)
+        """
+        self._auth_url = auth_url
+        self._client_id = client_id
+        self._header = {"Content-Type": "application/x-www-form-urlencoded"}
+        self._expires_at = None
+        self._refresh_expires_at = None
+
+    def generate_token(self, username=None, password=None, refresh=False):
+        """
+        DESCRIPTION:
+            Method generates keycloak token.
+            Token can be generated using any one of the following information:
+                * Username and password
+                * Refresh token
+
+        PARAMETERS:
+            username:
+                Optional Argument.
+                Specifies the username for database user for which keycloak token is
+                to be generated. If not specified, then user associated with current
+                connection is used.
+                Types: str
+
+            password:
+                Optional Argument.
+                Specifies the password for database user.
+                Types: str
+
+            refresh:
+                Optional Argument.
+                Specifies the boolean flag to indicate if token needs to be generated
+                using existing refresh token or not.
+                Default value: False
+                Types: Boolean
+
+        RETURNS:
+            Keycloak token
+
+        RAISES:
+            RuntimeError.
+
+        EXAMPLES:
+            # Example 1: Generate the authentication token using username and password.
+            >>> keycloak_obj.generate_token(username="username", password="password")
+
+            # Example 2: Generate the authentication token using default username for
+            #            current session and provided password.
+            >>> keycloak_obj.generate_token(password="password")
+
+            # Example 3: Generate the authentication token using refresh token which is
+            #            already available.
+            >>> keycloak_obj.generate_token(refresh=True)
+            """
+        # Define the payload.
+        if password:
+            payload = {
+                "grant_type": GrantType.PASS.value,
+                "client_id": self._client_id,
+                "username": username if username else _get_user(),
+                "password": password
+            }
+        if refresh:
+            payload = {
+                "grant_type": GrantType.REFRESH.value,
+                "client_id": self._client_id,
+                "refresh_token": self._refresh_token
+            }
+
+        # Make the POST request.
+        # Importing locally to avoid circular import.
+        from teradataml import UtilFuncs
+        response = UtilFuncs._http_request(self._auth_url, HTTPRequest.POST,
+                                           headers=self._header, data=payload)
+
+        # Check the response.
+        if 200 <= response.status_code < 300:
+            response_data = response.json()
+            self._auth_token = response_data['access_token']
+            self._refresh_token = response_data['refresh_token']
+            # 30 is buffer time to assume delay in processing of response_data.
+            self._expires_at = time.time() + response_data['expires_in'] - 30
+            self._refresh_expires_at = time.time() + response_data['refresh_expires_in'] - 30
+        else:
+            raise
+
+        return self._auth_token
+
+    def get_token(self):
+        """
+        DESCRIPTION:
+            Function to get keycloak token.
+            If access token is not expired, existing token is returned.
+            If access token is expired, and refresh token is alive, new access token is generated.
+            If both access token and refresh token are expired, TeradataMlException is raised
+
+        RETURNS:
+            Keycloak token
+
+        RAISES:
+            TeradataMlException.
+
+        """
+        # If existing auth_token is expired, regenerate using refresh token.
+        if self._expires_at and time.time() > self._expires_at:
+            # If refresh token is expired, raise error.
+            if self._refresh_expires_at and time.time() > self._refresh_expires_at:
+                raise TeradataMlException("Refresh token for keycloak is expired."
+                                          " Execute set_auth_token() to set fresh authentication token.",
+                                          MessageCodes.FUNC_EXECUTION_FAILED)
+            else:
+                # Regenerate fresh access token using refresh token.
+                self.generate_token(refresh=True)
+
+        return self._auth_token

teradataml/common/constants.py

@@ -20,7 +20,7 @@ from teradatasqlalchemy.types import (BYTE, VARBYTE, BLOB)
 from teradatasqlalchemy import (CHAR, CLOB)
 from teradatasqlalchemy import (PERIOD_DATE, PERIOD_TIME, PERIOD_TIMESTAMP)
 from teradatasqlalchemy import (INTERVAL_YEAR, INTERVAL_YEAR_TO_MONTH, INTERVAL_MONTH,
-                                INTERVAL_DAY,INTERVAL_DAY_TO_HOUR, INTERVAL_DAY_TO_MINUTE,
+                                INTERVAL_DAY, INTERVAL_DAY_TO_HOUR, INTERVAL_DAY_TO_MINUTE,
                                 INTERVAL_DAY_TO_SECOND, INTERVAL_HOUR,
                                 INTERVAL_HOUR_TO_MINUTE, INTERVAL_HOUR_TO_SECOND,
                                 INTERVAL_MINUTE, INTERVAL_MINUTE_TO_SECOND,
@@ -65,6 +65,8 @@ class SQLConstants(Enum):
     SQL_HELP_DATALAKE = 34
     CONSTRAINT = ["check_constraint", "primary_key_constraint",
                   "foreign_key_constraint", "unique_key_constraint"]
+    SQL_TD_OTF_METADATA = 35
+    SQL_TD_OTF_SNAPSHOT = 36
 
 
 class TeradataConstants(Enum):
@@ -82,7 +84,7 @@ class TeradataConstants(Enum):
     # Order of operators
     # shouldn't be changed. This is the order in which join condition is tested - first, operators
     # with two characters and then the operators with single character.
-    SUPPORTED_ENGINES = {"ENGINE_SQL" : {"name" : "sqle", "file" : "sqlengine_alias_definitions"}}
+    SUPPORTED_ENGINES = {"ENGINE_SQL": {"name": "sqle", "file": "sqlengine_alias_definitions"}}
     SUPPORTED_VANTAGE_VERSIONS = {"vantage1.0": "v1.0", "vantage1.1": "v1.1",
                                   "vantage1.3": "v1.3", "vantage2.0": "v1.1"}
     RANGE_SEPARATORS = [":"]
@@ -90,7 +92,7 @@ class TeradataConstants(Enum):
 
 class AEDConstants(Enum):
     AED_NODE_NOT_EXECUTED = 0
-    AED_NODE_EXECUTED     = 1
+    AED_NODE_EXECUTED = 1
     AED_DB_OBJECT_NAME_BUFFER_SIZE = 128
     AED_NODE_TYPE_BUFFER_SIZE = 32
     AED_ASSIGN_DROP_EXISITING_COLUMNS = "Y"
@@ -148,7 +150,7 @@ class TeradataTableKindConstants(Enum):
     TABLE = "table"
     VIEW = "view"
     TEMP = "temp"
-    ALL  = "all"
+    ALL = "all"
     ML_PATTERN = "ml_%"
     VOLATILE_TABLE_NAME = 'Table Name'
     REGULAR_TABLE_NAME = 'TableName'
@@ -220,7 +222,7 @@ class ModelCatalogingConstants(Enum):
 
 
 class CopyToConstants(Enum):
-    DBAPI_BATCHSIZE                      = 16383
+    DBAPI_BATCHSIZE = 16383
 
 
 class PTITableConstants(Enum):
@@ -377,34 +379,34 @@ class GeospatialConstants(Enum):
         "buffer": lambda x: "ST_Buffer",
         "contains": lambda x: "ST_Contains",
         "crosses": lambda x: "ST_Crosses",
-        "difference": lambda x: "ST_Difference", # M
+        "difference": lambda x: "ST_Difference",  # M
         "disjoint": lambda x: "ST_Disjoint",
-        "distance": lambda x: "ST_Distance", # M
-        "distance_3D": lambda x: "ST_3DDistance", # M
+        "distance": lambda x: "ST_Distance",  # M
+        "distance_3D": lambda x: "ST_3DDistance",  # M
         "geom_equals": lambda x: "ST_Equals",
         "intersection": lambda x: "ST_Intersection",
-        #"intersect": lambda x: "ST_Intersect", # M
+        # "intersect": lambda x: "ST_Intersect", # M
         "make_2D": lambda x: "Make_2D",
         "overlaps": lambda x: "ST_Overlaps",
         "relates": lambda x: "ST_Relate",
         "simplify": lambda x: "SimplifyPreserveTopology",
-        "sym_difference": lambda x: "ST_SymDifference", # M
+        "sym_difference": lambda x: "ST_SymDifference",  # M
         "touches": lambda x: "ST_Touches",
         "transform": lambda x: "ST_Transform",
         "union": lambda x: "ST_Union",
         "within": lambda x: "ST_Within",
-        "wkb_geom_to_sql": lambda x: "ST_WKBToSQL", # M
-        "wkt_geom_to_sql": lambda x: "ST_WKTToSQL", # M
+        "wkb_geom_to_sql": lambda x: "ST_WKBToSQL",  # M
+        "wkt_geom_to_sql": lambda x: "ST_WKTToSQL",  # M
         "set_srid": lambda x: "ST_SRID",
 
         ## *** Geometry Type ST_Point Methods *** ##
         "set_x": lambda x: "ST_X",
         "set_y": lambda x: "ST_Y",
         "set_z": lambda x: "ST_Z",
-        "spherical_buffer": lambda x: "ST_SphericalBufferMBR", # M
-        "spherical_distance": lambda x: "ST_SphericalDistance", # M
-        "spheroidal_buffer": lambda x: "ST_SpheroidalBufferMBR", # M
-        "spheroidal_distance": lambda x: "ST_SpheroidalDistance", # M
+        "spherical_buffer": lambda x: "ST_SphericalBufferMBR",  # M
+        "spherical_distance": lambda x: "ST_SphericalDistance",  # M
+        "spheroidal_buffer": lambda x: "ST_SpheroidalBufferMBR",  # M
+        "spheroidal_distance": lambda x: "ST_SpheroidalDistance",  # M
 
         ## *** Geometry Type ST_LineString Methods *** ##
         "line_interpolate_point": lambda x: "ST_Line_Interpolate_Point",
@@ -509,9 +511,18 @@ class TableOperatorConstants(Enum):
                                                     "delimiter(' ') " \
                                                     "returns('package VARCHAR({2}), " \
                                                     "version VARCHAR({2})'))"
-    
+
     SCRIPT_LIST_FILES_QUERY = "SELECT DISTINCT * FROM SCRIPT (SCRIPT_COMMAND " \
-                       "('ls ./{}') RETURNS ('Files VARCHAR({})'))"
+                              "('ls ./{}') RETURNS ('Files VARCHAR({})'))"
+
+
+    # OpenBlas by default is multi-threaded, needs to be set to single-threaded.
+    OPENBLAS_NUM_THREADS = "OPENBLAS_NUM_THREADS=1"
+
+    # Query to create a DataFrame with a range of numbers.
+    RANGE_QUERY = "WITH RECURSIVE NumberSeries (id) AS (SELECT id AS id from {0} "\
+                  "UNION ALL SELECT id {3} {1} FROM NumberSeries WHERE id {3} {1} {4} {2}) "\
+                  "SELECT id FROM NumberSeries;"
 
 class ValibConstants(Enum):
     # A dictionary that maps teradataml name of the exposed VALIB function name
@@ -1355,7 +1366,7 @@ class SQLFunctionConstants(Enum):
         "regexp_replace": "REGEXP_REPLACE",
         "regexp_similar": "REGEXP_SIMILAR",
         "regexp_substr": "REGEXP_SUBSTR",
-        
+
         # DateTime Functions
         'week_begin': 'td_week_begin',
         'week_start': 'td_week_begin',
@@ -1426,6 +1437,7 @@ class TeradataReservedKeywords(Enum):
                                "TYPE"
                                ]
 
+
 class TeradataAnalyticFunctionTypes(Enum):
     SQLE = "FASTPATH"
     UAF = "UAF"
@@ -1435,15 +1447,15 @@ class TeradataAnalyticFunctionTypes(Enum):
 
 
 class TeradataAnalyticFunctionInfo(Enum):
-
-    FASTPATH = {"func_type": "sqle", "lowest_version": "16.20", "display_function_type_name" :"SQLE"}
+    FASTPATH = {"func_type": "sqle", "lowest_version": "16.20", "display_function_type_name": "SQLE"}
     UAF = {"func_type": "uaf", "lowest_version": "17.20", "display_function_type_name": "UAF",
-           "metadata_class" : "_AnlyFuncMetadataUAF"}
+           "metadata_class": "_AnlyFuncMetadataUAF"}
     TABLE_OPERATOR = {"func_type": "tableoperator", "lowest_version": "17.00 ",
-                      "display_function_type_name" :"TABLE OPERATOR"}
+                      "display_function_type_name": "TABLE OPERATOR"}
     BYOM = {"func_type": "byom", "lowest_version": None, "display_function_type_name": "BYOM"}
     STORED_PROCEDURE = {"func_type": "storedprocedure", "lowest_version": "17.20", "display_function_type_name": "UAF",
-           "metadata_class" : "_AnlyFuncMetadataUAF"}
+                        "metadata_class": "_AnlyFuncMetadataUAF"}
+
 
 class TeradataUAFSpecificArgs(Enum):
     INPUT_MODE = "input_mode"
@@ -1451,9 +1463,11 @@ class TeradataUAFSpecificArgs(Enum):
     OUTPUT_FMT_INDEX = "output_fmt_index"
     OUTPUT_FMT_INDEX_STYLE = "output_fmt_index_style"
 
+
 class Query(Enum):
     VANTAGE_VERSION = "SELECT InfoData FROM DBC.DBCInfoV where InfoKey = 'VERSION'"
 
+
 class DriverEscapeFunctions(Enum):
     # Holds variables for the teradatasql driver escape functions to be used
     NATIVE_SQL = "{fn teradata_nativesql}"
@@ -1494,6 +1508,13 @@ class AsyncStatusColumns(Enum):
     ADDITIONAL_DETAILS = "Additional Details"
 
 
+class AsyncOpStatus(Enum):
+    # Holds valid status for asynchronous operatiosns in UES.
+    FILE_INSTALLED = "File Installed"
+    ERRED = "Errored"
+    FINISHED = "Finished"
+
+
 class CloudProvider(Enum):
     # Holds variable names for Cloud Providers.
     AWS = "AWS"
@@ -1503,7 +1524,8 @@ class CloudProvider(Enum):
     X_MS_VERSION = "2019-12-12"
     X_MS_BLOB_TYPE = "BlockBlob"
 
-class SessionParamsSQL:
+
+class SessionParamsSQL(Enum):
     # Holds the SQL Statements for Session params.
     TIMEZONE = "SET TIME ZONE {}"
     ACCOUNT = "SET SESSION ACCOUNT = '{}' FOR {}"
@@ -1523,7 +1545,8 @@ class SessionParamsSQL:
     QUERY_BAND = "SET QUERY_BAND = {} FOR {}"
     UDFSEARCHPATH = "SET SESSION UDFSEARCHPATH = {} FOR FUNCTION = {}"
 
-class SessionParamsPythonNames:
+
+class SessionParamsPythonNames(Enum):
     # Holds the SQL Statements for Session params.
     TIMEZONE = "Session Time Zone"
     ACCOUNT = "Account Name"
@@ -1532,3 +1555,25 @@ class SessionParamsPythonNames:
     DATABASE = "Current DataBase"
     DATEFORM = 'Current DateForm'
 
+
+class AutoMLConstants(Enum):
+    # List stores feature selection methods
+    FEATURE_SELECTION_MTDS = ["lasso", "rfe", "pca"]
+
+
+class AuthMechs(Enum):
+    """
+    Enum to hold permitted values for authentication mechanism.
+    """
+    OAUTH = "OAuth"
+    JWT = "JWT"
+    PAT = "PAT"
+    BASIC = "BASIC"
+    KEYCLOAK = "KEYCLOAK"
+
+class TDServices(Enum):
+    """
+    Enum to hold permitted values for types for services availed on Teradata vantage.
+    """
+    VECTORSTORE = "vectorstore"
+    MOPS = "MODELOPS" # For future reference

teradataml/common/exceptions.py

@@ -12,6 +12,12 @@ teradataml.common.exceptions
 An Error class for the teradataml python module
 """
 
+
+import warnings
+import json
+import pprint
+
+
 class TeradataMlException(Exception):
     """
     Teradata Ml Exception class.
@@ -71,3 +77,29 @@ class TeradataMlException(Exception):
         """
         super(TeradataMlException, self).__init__(msg)
         self.code = code
+
+class TeradatamlRestException(TeradataMlException):
+    def __init__(self, msg, code, raw_message=None):
+        super().__init__(msg, code)
+
+        self._msg = msg
+        self._raw_message = raw_message
+    
+    @property
+    def json_response(self):
+        """
+        Returns the JSON response from the server.
+        """
+        if self._raw_message is not None and self._raw_message.startswith("Error message: "):
+            failure_msg = f"Could not convert msg to json. Message is \n{self._raw_message}"
+            try:
+                json_resp = self._msg.split("Error message: ")[1]
+                json_resp = json.loads(json_resp)
+                return pprint.pformat(json_resp)
+            except json.JSONDecodeError as _:
+                warnings.warn(failure_msg)
+        elif self._raw_message is None:
+            return None
+        else:
+            warnings.warn(failure_msg)
+    

teradataml/common/messagecodes.py

@@ -125,6 +125,7 @@ class ErrorInfoCodes(Enum):
     SPECIFY_AT_LEAST_ONE_ARG = 'TDML_2037'
     CANNOT_USE_TOGETHER_WITH = 'TDML_2042'
     TABLE_DOES_NOT_EXIST = 'TDML_2046'
+    DEPENDENT_METHOD = 'TDML_2113'
 
     # Reserved for Generic Error Messages: 2121 - 2199
     RESERVED_KEYWORD = 'TDML_2121'
@@ -227,6 +228,21 @@ class ErrorInfoCodes(Enum):
     # OpenAF Error codes starting from 2551 - Reserved till 2560.
     SET_REQUIRED_PARAMS = 'TDML_2551'
 
+    # Error codes for OTF. Reserved till 2570.
+    OTF_TABLE_REQUIRED = 'TDML_2561'
+
+    # Rest Excceptions. Reserved for 2570 - 2580
+    REST_HTTP_ERROR = 'TDML_2570'
+    REST_AUTH_MISSING_ARG = 'TDML_2571'
+    REST_NOT_CONFIGURED = 'TDML_2572'
+    REST_DEVICE_CODE_NO_BOTH = 'TDML_2573'
+    REST_DEVICE_CODE_GEN_FAILED = 'TDML_2574'
+    REST_DEVICE_CODE_AUTH_FAILED = 'TDML_2575'
+
+    # Python SDK Error codes starting from 2580 - Reserved till 2590.
+    INFO_NOT_PROVIDED_USE_DEFAULT = 'TDML_W_2580' # Logger warning.
+
+
 class MessageCodes(Enum):
     """
     MessageCodes contains all the messages that are displayed to the user which are informational
@@ -440,4 +456,16 @@ class MessageCodes(Enum):
     CONNECTION_PARAMS = "Required connection parameters are missing. Connection parameters should either be " \
                         "explicitly passed to function or specified using a configuration file, or setting up " \
                         "the environment variables."
+    DEPENDENT_METHOD = "Method(s) {} must be called before calling '{}'."
+    REST_HTTP_ERROR = "Failed to run rest API:\n{}"
+    REST_AUTH_MISSING_ARG = "For '{}' authentication, '{}' is/are not provided in config file or "\
+                            "environment variable or through constructor argument 'auth'."
+    REST_NOT_CONFIGURED = "The argument '{}' is not set. {} endpoint not configured.\n" \
+                          "Try (re)copy the CLI configuration from {} UI -> Session Details -> CLI Config."
+    REST_DEVICE_CODE_NO_BOTH = "Token does not contain access_token or refresh_token. Received token: {}"
+    REST_DEVICE_CODE_GEN_FAILED = "Error generating the device code. Received code: {}."
+    REST_DEVICE_CODE_AUTH_FAILED = "Error authenticating the device code.\n{}."
+
+    INFO_NOT_PROVIDED_USE_DEFAULT = "{} is not provided in path '{}' method '{}' for operationID '{}' using default {}."
+    OTF_TABLE_REQUIRED = "{} is supported only with OTF table."
 

teradataml/common/messages.py

@@ -195,7 +195,16 @@ class Messages():
             [ErrorInfoCodes.PATH_NOT_FOUND, MessageCodes.PATH_NOT_FOUND],
             [ErrorInfoCodes.TARGET_COL_NOT_FOUND_FOR_EVALUATE, MessageCodes.TARGET_COL_NOT_FOUND_FOR_EVALUATE],
             [ErrorInfoCodes.SET_REQUIRED_PARAMS, MessageCodes.SET_REQUIRED_PARAMS],
-            [ErrorInfoCodes.MISSING_ARGS, MessageCodes.CONNECTION_PARAMS]
+            [ErrorInfoCodes.MISSING_ARGS, MessageCodes.CONNECTION_PARAMS],
+            [ErrorInfoCodes.DEPENDENT_METHOD, MessageCodes.DEPENDENT_METHOD],
+            [ErrorInfoCodes.REST_HTTP_ERROR, MessageCodes.REST_HTTP_ERROR],
+            [ErrorInfoCodes.REST_AUTH_MISSING_ARG, MessageCodes.REST_AUTH_MISSING_ARG],
+            [ErrorInfoCodes.REST_NOT_CONFIGURED, MessageCodes.REST_NOT_CONFIGURED],
+            [ErrorInfoCodes.REST_DEVICE_CODE_NO_BOTH, MessageCodes.REST_DEVICE_CODE_NO_BOTH],
+            [ErrorInfoCodes.REST_DEVICE_CODE_GEN_FAILED, MessageCodes.REST_DEVICE_CODE_GEN_FAILED],
+            [ErrorInfoCodes.REST_DEVICE_CODE_AUTH_FAILED, MessageCodes.REST_DEVICE_CODE_AUTH_FAILED],
+            [ErrorInfoCodes.INFO_NOT_PROVIDED_USE_DEFAULT, MessageCodes.INFO_NOT_PROVIDED_USE_DEFAULT],
+            [ErrorInfoCodes.OTF_TABLE_REQUIRED, MessageCodes.OTF_TABLE_REQUIRED]
     ]
 
     @staticmethod
@@ -224,11 +233,11 @@ class Messages():
 
         """
         for msg in Messages.__messages:
-            if msg[1] == messagecode :
+            if msg[1] == messagecode:
                 message = "{}({}) {}".format(Messages.__standard_message, msg[0].value, msg[1].value)
                 if len(variables) != 0:
                     message = message.format(*variables)
-                if len(kwargs) != 0 :
+                if len(kwargs) != 0:
                     message = "{} {}".format(message, kwargs)
-
+                break
         return message

teradataml/common/sqlbundle.py

@@ -70,8 +70,9 @@ class SQLBundle:
                 [SQLConstants.SQL_EXEC_STORED_PROCEDURE, "call {0}"],
                 [SQLConstants.SQL_SELECT_COLUMNNAMES_WITH_WHERE, "sel {0} from {1} where {2}"],
                 [SQLConstants.SQL_HELP_DATABASE, "HELP DATABASE {0}"],
-                [SQLConstants.SQL_HELP_DATALAKE, "HELP DATALAKE {0}"]
-
+                [SQLConstants.SQL_HELP_DATALAKE, "HELP DATALAKE {0}"],
+                [SQLConstants.SQL_TD_OTF_METADATA, "SELECT * FROM {0}(ON ({1})) D;"],
+                [SQLConstants.SQL_TD_OTF_SNAPSHOT, "SELECT * FROM {0} FOR SNAPSHOT AS OF {1};"]
         ]
         self._add_sql_version()