SNOW-1305528 : Adding lineage.trace API

CHANGELOG.md

@@ -4,6 +4,7 @@
 
 ### New Features
 - Support stored procedure register with packages given as Python modules.
+- Added snowflake.snowpark.Session.lineage.trace to explore data lineage of snowfake objects.
 
 ## 1.15.0 (2024-04-24)
 

docs/source/index.rst

@@ -27,6 +27,7 @@ information, see the `Snowpark Developer Guide for Python <https://docs.snowflak
    udtf
    observability
    files
+   lineage
    context
    exceptions
 

docs/source/lineage.rst

@@ -0,0 +1,28 @@
+================
+LINEAGE
+================
+
+.. automodule:: snowflake.snowpark.lineage
+  :noindex:
+
+
+.. currentmodule:: snowflake.snowpark.lineage
+
+
+
+
+
+.. rubric:: Classes
+
+.. autosummary::
+    :toctree: api/
+
+    Lineage
+    LineageDirection
+
+.. rubric:: Methods
+
+.. autosummary::
+    :toctree: api/
+
+    ~Lineage.trace

docs/source/session.rst

@@ -80,11 +80,12 @@ Snowpark Session
     Session.custom_package_usage_config
     Session.file
     Session.query_tag
+    Session.lineage
     Session.read
     Session.sproc
     Session.sql_simplifier_enabled
     Session.telemetry_enabled
     Session.udf
     Session.udtf
     Session.session_id
-    Session.connection
+    Session.connection

src/snowflake/snowpark/_internal/error_message.py

@@ -379,6 +379,13 @@ def SERVER_FAILED_FETCH_PANDAS(message: str) -> SnowparkFetchDataException:
             error_code="1406",
         )
 
+    @staticmethod
+    def SERVER_FAILED_FETCH_LINEAGE(message: str) -> SnowparkFetchDataException:
+        return SnowparkFetchDataException(
+            f"Failed to fetch a lineage information. The error is: {message}",
+            error_code="1406",
+        )
+
     @staticmethod
     def SERVER_UDF_UPLOAD_FILE_STREAM_CLOSED(
         dest_filename: str,