diff --git a/packages/orm/src/client/options.ts b/packages/orm/src/client/options.ts
index 2ec330d0c..64f50b613 100644
--- a/packages/orm/src/client/options.ts
+++ b/packages/orm/src/client/options.ts
@@ -50,6 +50,8 @@ export type ClientOptions<Schema extends SchemaDef> = {
 
     /**
      * Custom function definitions.
+     *
+     * @private
      */
     functions?: Record<string, ZModelFunction<Schema>>;
 
diff --git a/packages/orm/src/client/plugin.ts b/packages/orm/src/client/plugin.ts
index edf5c86ad..5664abb81 100644
--- a/packages/orm/src/client/plugin.ts
+++ b/packages/orm/src/client/plugin.ts
@@ -24,8 +24,10 @@ export interface RuntimePlugin<Schema extends SchemaDef = SchemaDef> {
      */
     description?: string;
 
+    // TODO: revisit
     /**
      * Custom function implementations.
+     * @private
      */
     functions?: Record<string, ZModelFunction<Schema>>;
 
diff --git a/packages/sdk/package.json b/packages/sdk/package.json
index 7aaada5b9..eebe2bf93 100644
--- a/packages/sdk/package.json
+++ b/packages/sdk/package.json
@@ -25,6 +25,16 @@
                 "types": "./dist/index.d.cts",
                 "default": "./dist/index.cjs"
             }
+        },
+        "./ast": {
+            "import": {
+                "types": "./dist/ast.d.ts",
+                "default": "./dist/ast.js"
+            },
+            "require": {
+                "types": "./dist/ast.d.cts",
+                "default": "./dist/ast.cjs"
+            }
         }
     },
     "dependencies": {
diff --git a/packages/sdk/src/ast.ts b/packages/sdk/src/ast.ts
new file mode 100644
index 000000000..3407e5b41
--- /dev/null
+++ b/packages/sdk/src/ast.ts
@@ -0,0 +1 @@
+export * from '@zenstackhq/language/ast';
diff --git a/packages/sdk/tsup.config.ts b/packages/sdk/tsup.config.ts
index 5a74a9dd1..131257ec3 100644
--- a/packages/sdk/tsup.config.ts
+++ b/packages/sdk/tsup.config.ts
@@ -3,6 +3,7 @@ import { defineConfig } from 'tsup';
 export default defineConfig({
     entry: {
         index: 'src/index.ts',
+        ast: 'src/ast.ts',
     },
     outDir: 'dist',
     splitting: false,