changed flag value to be of any type

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/common/feature_flag.py

@@ -76,7 +76,7 @@ def _is_refresh_needed(self) -> bool:
         )
         return time.monotonic() > refresh_threshold
 
-    def is_feature_enabled(self, name: str, default_value: bool) -> bool:
+    def get_flag_value(self, name: str, default_value: Any) -> Any:
         """
         Checks if a feature is enabled.
         - BLOCKS on the first call until flags are fetched.
@@ -95,10 +95,7 @@ def is_feature_enabled(self, name: str, default_value: bool) -> bool:
             assert self._flags is not None
 
             # Now, return the value from the populated cache.
-            flag_value = self._flags.get(name)
-            if flag_value is None:
-                return default_value
-            return flag_value.lower() == "true"
+            return self._flags.get(name, default_value)
 
     def _refresh_flags(self):
         """Performs a synchronous network request to fetch and update flags."""

src/databricks/sql/telemetry/telemetry_client.py

@@ -110,9 +110,10 @@ def is_telemetry_enabled(connection: "Connection") -> bool:
 
         if connection.enable_telemetry:
             context = FeatureFlagsContextFactory.get_instance(connection)
-            return context.is_feature_enabled(
+            flag_value = context.get_flag_value(
                 TelemetryHelper.TELEMETRY_FEATURE_FLAG_NAME, default_value=False
             )
+            return str(flag_value).lower() == "true"
         else:
             return False