omnata-plugin-devkit 0.13.7a199__tar.gz → 0.13.8a200__tar.gz

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omnata-plugin-devkit
-Version: 0.13.7a199
+Version: 0.13.8a200
 Summary: 
 License-File: LICENSE
 Author: James Weakley
@@ -13,7 +13,7 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: behave (>=1.2,<2.0)
 Requires-Dist: behave_pandas (>=0.5.0,<0.6.0)
-Requires-Dist: omnata-plugin-runtime (>=0.12.5)
+Requires-Dist: omnata-plugin-runtime (==0.12.10a374)
 Requires-Dist: omnata_cli (>=0.2.3,<0.3.0)
 Requires-Dist: pyparsing (>=3.0)
 Requires-Dist: python-slugify (>=8,<9)

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "omnata-plugin-devkit"
-version = "0.13.7a199"
+version = "0.13.8a200"
 description = ""
 authors = ["James Weakley <james.weakley@omnata.com>"]
 readme = "README.md"
@@ -10,10 +10,7 @@ packages = [{ include = "omnata_plugin_devkit", from = "src" }]
 # support 3.10 - 3.13
 python = ">=3.10, <=3.13"
 omnata_cli = "^0.2.3"
-omnata-plugin-runtime = ">=0.12.5"
-#pandas = "^2,<=2.3.3"
-#numpy = "<=2.3.5"
-#jinja2 = "^3,<=3.1.6"
+omnata-plugin-runtime = "==0.12.10a374"
 python-slugify = "^8"
 vcrpy = "^4"
 tabulate = "^0.9"

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/API_LIMITS.sql.jinja

@@ -6,7 +6,8 @@ create or replace procedure PLUGIN.API_LIMITS(
                                     OTHER_SECRETS_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CONFIGURATION_FORM.sql.jinja

@@ -2,7 +2,8 @@ create or replace procedure PLUGIN.CONFIGURATION_FORM(
                                                 CONNECTION_FORM_REQUEST OBJECT)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CONNECTION_FORM.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.CONNECTION_FORM(CONNECTIVITY_OPTION VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CONNECTION_TEST.sql.jinja

@@ -7,7 +7,8 @@ create or replace procedure PLUGIN.CONNECTION_TEST(
                                     OTHER_SECRETS_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CONSTRUCT_FORM_OPTION.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.CONSTRUCT_FORM_OPTION(FUNCTION_NAME VARCHAR,STORED_VALUES ARRAY)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CREATE_BILLING_EVENTS.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.CREATE_BILLING_EVENTS(BILLING_EVENT_REQUEST OBJECT)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/CREATE_GENERIC_SECRET_OBJECT.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.CREATE_GENERIC_SECRET_OBJECT(SECRET_NAME VARCHAR, SECRET_CONTENTS VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 HANDLER = 'run'
 COMMENT = $$

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/GET_MISSING_APP_PRIVILEGES.sql.jinja

@@ -1,7 +1,7 @@
 create or replace procedure PLUGIN.GET_MISSING_APP_PRIVILEGES()
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
 PACKAGES = ('snowflake-snowpark-python','snowflake-native-apps-permission')
 HANDLER = 'run'
 COMMENT = $$

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/INBOUND_LIST_STREAMS.sql.jinja

@@ -8,7 +8,8 @@ create or replace procedure PLUGIN.INBOUND_LIST_STREAMS(
                                                 SELECTED_STREAMS ARRAY)
 returns table(RESULT object)
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/LIST_STAGES.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.LIST_STAGES()
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/NETWORK_ADDRESSES.sql.jinja

@@ -5,7 +5,8 @@ create or replace procedure PLUGIN.NETWORK_ADDRESSES(
                                     )
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/NGROK_POST_TUNNEL_FIELDS.sql.jinja

@@ -6,7 +6,8 @@ create or replace procedure PLUGIN.NGROK_POST_TUNNEL_FIELDS(
                                                 FUNCTION_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/OUTBOUND_RECORD_VALIDATOR.sql.jinja

@@ -1,7 +1,8 @@
 create or replace function PLUGIN.OUTBOUND_RECORD_VALIDATOR(SYNC_PARAMETERS OBJECT,FIELD_MAPPINGS OBJECT,TRANSFORM_RECORD OBJECT,SOURCE_TYPES OBJECT)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/RETRIEVE_SECRETS.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.RETRIEVE_SECRETS(OAUTH_SECRET_NAME VARCHAR,OTHER_SECRETS_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/RETRIEVE_SECRETS_UDF.sql.jinja

@@ -1,7 +1,7 @@
 create or replace function PLUGIN.RETRIEVE_SECRETS_UDF(OAUTH_SECRET_NAME VARCHAR,OTHER_SECRETS_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
 HANDLER = 'run'
 COMMENT = $$
 Allows this plugin to retrieve its own secrets during CONFIGURE_API calls.

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/SET_CONNECTION_OBJECTS.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.SET_CONNECTION_OBJECTS(PARAMETERS OBJECT)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/SYNC.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.SYNC(SYNC_REQUEST OBJECT)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/TUNNEL_TEST.sql.jinja

@@ -2,7 +2,8 @@ create or replace procedure PLUGIN.TUNNEL_TEST(TUNNEL_TYPE VARCHAR,
                                                 OTHER_SECRETS_NAME VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/jinja_templates/UPDATE_GENERIC_SECRET_OBJECT.sql.jinja

@@ -1,7 +1,8 @@
 create or replace procedure PLUGIN.UPDATE_GENERIC_SECRET_OBJECT(SECRET_NAME VARCHAR,SECRET_CONTENTS VARCHAR,MERGE_WITH_OTHER_SECRET VARCHAR)
 returns object
 language python
-RUNTIME_VERSION = '3.10'
+RUNTIME_VERSION = '3.12'
+ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository
 PACKAGES = ({{packages}})
 IMPORTS = ('/app.zip')
 HANDLER = 'run'

{omnata_plugin_devkit-0.13.7a199 → omnata_plugin_devkit-0.13.8a200}/src/omnata_plugin_devkit/plugin_uploader.py

@@ -14,7 +14,6 @@ import distutils.core
 import inspect
 import json
 import os
-import re
 import shutil
 import sys
 import typing
@@ -22,7 +21,7 @@ import zipfile
 import copy
 import uuid
 from pathlib import Path
-from typing import Any, Dict, List, Literal, Optional
+from typing import Any, Dict, List, Literal, Optional, cast
 
 from pathlib import Path
 
@@ -34,8 +33,6 @@ from omnata_plugin_runtime.omnata_plugin import (
     UDFDefinition, UDTFDefinition
 )
 from omnata_plugin_devkit.snowcli.cli.plugins.snowpark import package_utils
-from omnata_plugin_devkit.snowcli.cli.plugins.snowpark import snowpark_shared
-from omnata_plugin_devkit.snowcli.cli.plugins.snowpark.models import PypiOption
 from snowflake.snowpark import Session
 import importlib.metadata
 
@@ -67,7 +64,7 @@ class PluginUploader:
         manifest: PluginManifest,
         anaconda_packages: List[str],
         bundled_packages: List[str],
-        icon_source: str,
+        icon_source: Optional[str],
         plugin_class_name: str,
         has_custom_validator: bool,
         plugin_runtime_version: str,
@@ -124,7 +121,7 @@ class PluginUploader:
         version_comment: str = "",
         log_level: str = "INFO",
         udf_directory_override: Optional[str] = None,
-    ) -> str:
+    ) -> Optional[str]:
         """
         Creates a stage and uploads the latest plugin artifacts, Omnata runtime etc.
         If the schema name is not provided, the plugin_id from the manifest will be used.
@@ -180,9 +177,10 @@ class PluginUploader:
         else:
             tier_file = Path(os.path.join(plugin_directory, "tier.txt"))
             if tier_file.exists():
-                plugin_tier = tier_file.read_text(encoding="utf-8").strip()
-                if plugin_tier not in ["standard","premium","byo","free"]:
-                    raise ValueError(f"Invalid tier {plugin_tier} in {tier_file}")
+                plugin_tier_raw = tier_file.read_text(encoding="utf-8").strip()
+                if plugin_tier_raw not in ["standard","premium","byo","free"]:
+                    raise ValueError(f"Invalid tier {plugin_tier_raw} in {tier_file}")
+                plugin_tier = cast(Literal["byo","free","standard","premium"], plugin_tier_raw)
         print(f"Using plugin tier: {plugin_tier}")
         license_url_file = Path(os.path.join(plugin_directory, "license_url.txt"))
         license_url: Optional[str] = (
@@ -214,7 +212,7 @@ class PluginUploader:
         # do checks first
 
         icon_file_name = "icon.svg"
-        icon_source = None
+        icon_source:Optional[str] = None
         if os.path.exists(os.path.join(plugin_directory, icon_file_name)):
             print("Using icon from icon.svg")
             with open(
@@ -246,60 +244,34 @@ class PluginUploader:
         if os.path.exists("build.zip"):
             os.remove("build.zip")
 
-        print("Creating Snowpark package for Snowflake")
-        # we allow native libraries to be included in the package, but then check ourselves
-        snowpark_shared.snowpark_package(
-            source=Path('.'),
-            artifact_file=Path('build.zip'),
-            pypi_download=PypiOption.YES,
-            check_anaconda_for_pypi_deps=True,
-            package_native_libraries=PypiOption.YES,
-        )
-        with zipfile.ZipFile('build.zip', 'r') as zip_read:
-            with zipfile.ZipFile('app.zip', 'w') as zip_write:
-                # Copy each file, skipping those in the specified folder
-                # some packages are named inconsistently which makes them hard to remove from the artifact
-                for item in zip_read.infolist():
-                    if not item.filename.startswith('.packages') and \
-                        not item.filename.startswith('udf_direct_imports') and \
-                        not item.filename.startswith('pydantic_core'):
-                        if item.filename.endswith('.so'):
-                            raise ValueError(f"Native library found in package: {item.filename}")
-                        data = zip_read.read(item.filename)
-                        zip_write.writestr(item, data)
+        print("Building app.zip from plugin source")
+        with zipfile.ZipFile('app.zip', 'w', zipfile.ZIP_DEFLATED) as zf:
+            for root, dirs, files in os.walk('.'):
+                dirs[:] = [d for d in dirs if not d.startswith('.') and d != '__pycache__' and d != 'udf_direct_imports']
+                for fname in files:
+                    if fname.startswith('.'):
+                        continue
+                    if fname.endswith('.so'):
+                        raise ValueError(f"Native library found in package: {os.path.join(root, fname)}")
+                    full = os.path.join(root, fname)
+                    zf.write(full, os.path.relpath(full, '.'))
         self.session.sql(
             f"put file://app.zip @{database_name}.{schema_name}.{stage_name} AUTO_COMPRESS=FALSE OVERWRITE=TRUE"
         ).collect()
 
-        if not os.path.exists(".packages"):
-            raise ValueError(".packages directory not found after packaging")
-        subfolders = [
-            f.path
-            for f in os.scandir(".packages")
-            if f.is_dir() and os.path.join(".packages","omnata_plugin_runtime-") in f.path
-        ]
-        if len(subfolders) != 1:
+        requirements_list = package_utils.parse_requirements("requirements.txt")
+        all_packages: List[str] = [r.line for r in requirements_list]
+
+        runtime_reqs = [r for r in requirements_list if r.name == "omnata-plugin-runtime"]
+        if len(runtime_reqs) != 1 or len(runtime_reqs[0].specs) == 0:
             raise ValueError(
-                "Failed to find omnata-plugin-runtime in downloaded package metadata"
+                "requirements.txt must contain a pinned omnata-plugin-runtime entry (e.g. omnata-plugin-runtime==X.Y.Z)"
             )
-        subfolder = subfolders[0]
-        regex_matches = re.search(
-            r"^\.packages.omnata_plugin_runtime-(.*)\.dist-info$", subfolder
-        )
-        if regex_matches is None:
+        op, omnata_plugin_runtime_version = runtime_reqs[0].specs[0]
+        if op != "==":
             raise ValueError(
-                "Failed to parse version of omnata-plugin-runtime package from metadata"
+                f"omnata-plugin-runtime must be pinned with '==' (got '{op}{omnata_plugin_runtime_version}')"
             )
-        omnata_plugin_runtime_version = regex_matches.group(1)
-        anaconda_packages = package_utils.parse_requirements("requirements.snowflake.txt")
-        anaconda_package_names: List[str] = [
-            r.name
-            for r in anaconda_packages
-        ]
-        other_packages = package_utils.parse_requirements("requirements.other.txt")
-        other_package_names: List[str] = [
-            r.name for r in other_packages
-        ]
         os.chdir(original_wd)
         # icon_file_name_full = os.path.abspath(os.path.join(plugin_directory,icon_file_name))
         # print(f'Loading icon file {icon_file_name_full} into stage')
@@ -310,37 +282,10 @@ class PluginUploader:
         plugin_fqn: str = f"{manifest.developer_id}__{manifest.plugin_id}".upper()
 
         install_script = "setup_script.sql"
-        default_packages = ["dictdiffer", "jinja2", "requests", "pydantic","snowflake-telemetry-python"]
-        if len([w for w in anaconda_packages if "snowflake-snowpark-python" in w.name]) == 0:
-            # if not present at all, pin it to a specific version
-            anaconda_package_names = anaconda_package_names + ["snowflake-snowpark-python==1.22.1"]
-            print("snowflake-snowpark-python not present, adding to default packages")
-        else:
-            snowpark_package = [w for w in anaconda_packages if "snowflake-snowpark-python" in w.name][0]
-            if len(snowpark_package.specs) == 0:
-                print("snowflake-snowpark-python found unpinned, replacing with pinned version")
-                anaconda_package_names = [
-                    w if w != "snowflake-snowpark-python" else "snowflake-snowpark-python==1.22.1"
-                    for w in anaconda_package_names
-                ]
-            else:
-                print(f"{snowpark_package.name} found already pinned ({snowpark_package.line}), leaving alone")
-                anaconda_package_names = [
-                    w.name if w.name != "snowflake-snowpark-python" else w.line
-                    for w in anaconda_packages
-                ]
-        # remove snowflake-connector-python if it's found its way into the other packages or the anaconda packages
-        other_package_names = [
-            w for w in other_package_names if "snowflake-connector-python" not in w
-        ]
-        anaconda_package_names = [
-            w for w in anaconda_package_names if "snowflake-connector-python" not in w
-        ]
         has_custom_validator = (
             plugin_class_instance.outbound_record_validator.__module__ != 'omnata_plugin_runtime.omnata_plugin'
         )
         print(f"Has custom validator: {has_custom_validator} ({plugin_class_instance.outbound_record_validator.__module__})")
-        all_packages = list(set(default_packages + anaconda_package_names))
         packages_to_include = [
             f"'{p}'" for p in all_packages
         ]
@@ -544,7 +489,7 @@ $${}$$;
                 print(f"Found {udf.language} UDF: {udf.name}, consumer facing: {udf.expose_to_consumer}")
                 if udf.language == "python":
                     udf.packages = all_packages
-                setup_script.write(str(udf))
+                setup_script.write(_inject_artifact_repository(str(udf)))
                 if udf.expose_to_consumer is True:
                     setup_script.write(
                         f"grant usage on function UDFS.{function_signature(input=udf,include_name=True,include_params='data_types_only',connection_parameter='leave')} to application role CONSUMER_FUNCTION_CALLER;\n"
@@ -570,7 +515,7 @@ $${}$$;
                 print(f"Found {udtf.language} UDTF: {udtf.name}, consumer facing: {udtf.expose_to_consumer}")
                 if udtf.language == "python":
                     udtf.packages = all_packages
-                setup_script.write(str(udtf))
+                setup_script.write(_inject_artifact_repository(str(udtf)))
                 if udtf.expose_to_consumer is True:
                     setup_script.write(
                         f"grant usage on function UDFS.{function_signature(input=udtf,include_name=True,include_params='data_types_only',connection_parameter='leave')} to application role CONSUMER_FUNCTION_CALLER;\n"
@@ -604,8 +549,8 @@ $${}$$;
             setup_script.write(
                 self.plugin_info_udf_definition(
                     manifest=manifest,
-                    anaconda_packages=anaconda_package_names,
-                    bundled_packages=other_package_names,
+                    anaconda_packages=all_packages,
+                    bundled_packages=[],
                     icon_source=icon_source,
                     plugin_class_name=plugin_class_name,
                     has_custom_validator=has_custom_validator,
@@ -738,8 +683,8 @@ GRANT USAGE ON FUNCTION PLUGIN.TOKEN() TO application role OMNATA_MANAGEMENT;
             self.session.sql(
                 self.plugin_info_udf_definition(
                     manifest=manifest,
-                    anaconda_packages=anaconda_package_names,
-                    bundled_packages=other_package_names,
+                    anaconda_packages=all_packages,
+                    bundled_packages=[],
                     icon_source=icon_source,
                     plugin_class_name=plugin_class_name,
                     has_custom_validator=has_custom_validator,
@@ -755,6 +700,14 @@ GRANT USAGE ON FUNCTION PLUGIN.TOKEN() TO application role OMNATA_MANAGEMENT;
         return schema_name
 
 
+def _inject_artifact_repository(udf_sql: str) -> str:
+    return udf_sql.replace(
+        "PACKAGES = (",
+        "ARTIFACT_REPOSITORY = snowflake.snowpark.pypi_shared_repository\nPACKAGES = (",
+        1,
+    )
+
+
 def prepare_version_comment(version_comment:str):
     """
     The manifest file takes a comment, but since it's in YAML you have to do a bunch of silly work to make it valid and multiline.