Add missing license headers, auto call warm up

Author: sjrl

integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/auth.py

@@ -1,6 +1,7 @@
 # SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
 #
 # SPDX-License-Identifier: Apache-2.0
+
 from pathlib import Path
 from typing import Any, Literal, Optional
 from urllib.parse import quote

integrations/snowflake/src/haystack_integrations/components/retrievers/snowflake/snowflake_table_retriever.py

@@ -2,7 +2,7 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-from typing import Any, Literal, Optional
+from typing import Any, Literal, Optional, Union
 from urllib.parse import quote_plus
 
 import polars as pl
@@ -400,7 +400,7 @@ def _execute_query_with_connector(self, query: str) -> Optional[pl.DataFrame]:
             return None
 
     @staticmethod
-    def _empty_response() -> dict[str, Any]:
+    def _empty_response() -> dict[str, Union[DataFrame, str]]:
         """Returns a standardized empty response.
 
         :returns:
@@ -411,7 +411,7 @@ def _empty_response() -> dict[str, Any]:
         return {"dataframe": DataFrame(), "table": ""}
 
     @component.output_types(dataframe=DataFrame, table=str)
-    def run(self, query: str, return_markdown: Optional[bool] = None) -> dict[str, Any]:
+    def run(self, query: str, return_markdown: Optional[bool] = None) -> dict[str, Union[DataFrame, str]]:
         """
         Executes a SQL query against a Snowflake database using ADBC and Polars.
 
@@ -423,8 +423,7 @@ def run(self, query: str, return_markdown: Optional[bool] = None) -> dict[str, A
             - `"table"`: A Markdown-formatted string representation of the DataFrame.
         """
         if not self._warmed_up:
-            msg = "SnowflakeTableRetriever not warmed up. Please call `warm_up()` before running queries."
-            raise RuntimeError(msg)
+            self.warm_up()
 
         # Validate SQL query
         if not query:

integrations/snowflake/tests/test_auth.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 from pathlib import Path
 from unittest.mock import Mock
 

integrations/snowflake/tests/test_snowflake_table_retriever.py

@@ -1,3 +1,7 @@
+# SPDX-FileCopyrightText: 2025-present deepset GmbH <info@deepset.ai>
+#
+# SPDX-License-Identifier: Apache-2.0
+
 import os
 from pathlib import Path
 from typing import Any, Optional