@ton-agent-kit/core 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@ton-agent-kit/core",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "Core SDK for TON Agent Kit — plugin system, wallet abstraction, agent class",
   "main": "src/index.ts",
   "types": "src/index.ts",

package/src/agent.ts

@@ -1,4 +1,5 @@
 import { readFileSync } from "fs";
+import { toJSONSchema } from "zod";
 import { TonClient4 } from "@ton/ton";
 import { PluginRegistry } from "./plugin";
 import type {
@@ -208,42 +209,7 @@ export class TonAgentKit {
    */
   toAITools(): Array<{ type: "function"; function: { name: string; description: string; parameters: any } }> {
     return this.getAvailableActions().map((action) => {
-      let parameters: any = { type: "object", properties: {} };
-
-      // Try Zod v4 native toJSONSchema first
-      try {
-        const { toJSONSchema } = require("zod");
-        const schema = toJSONSchema(action.schema);
-        const { $schema, ...rest } = schema;
-        parameters = { type: "object", properties: {}, ...rest };
-      } catch {
-        // Fallback: extract property names from Zod schema shape
-        try {
-          const shape = (action.schema as any)?._def?.shape || (action.schema as any)?.shape;
-          if (shape && typeof shape === "object") {
-            const props: Record<string, any> = {};
-            for (const [key, val] of Object.entries(shape)) {
-              const def = (val as any)?._def || (val as any)?.def || {};
-              const zodType = def.type || def.typeName || "string";
-              let jsonType = "string";
-              if (zodType === "number" || zodType === "ZodNumber" || zodType === "float64") jsonType = "number";
-              if (zodType === "boolean" || zodType === "ZodBoolean") jsonType = "boolean";
-
-              const isOptional = zodType === "optional" || zodType === "ZodOptional" || def.type === "optional";
-              const innerDef = isOptional ? (def.innerType?.def || def.innerType?._def || {}) : def;
-              const innerType = innerDef.type || innerDef.typeName || "string";
-              let finalType = "string";
-              if (innerType === "number" || innerType === "ZodNumber" || innerType === "float64") finalType = "number";
-              if (innerType === "boolean" || innerType === "ZodBoolean") finalType = "boolean";
-
-              const description = (val as any)?._def?.description || (val as any)?.description || "";
-              props[key] = { type: finalType, ...(description ? { description } : {}) };
-            }
-            parameters = { type: "object", properties: props };
-          }
-        } catch {}
-      }
-
+      const { $schema, ...parameters } = toJSONSchema(action.schema);
       return {
         type: "function" as const,
         function: {
@@ -275,9 +241,8 @@ export class TonAgentKit {
       );
     }
 
-    // Dynamic imports so the SDK works without these deps installed
+    // Dynamic import so the SDK works without openai installed
     const { default: OpenAI } = await import("openai");
-    const { toJSONSchema } = await import("zod");
     const client = new OpenAI({ apiKey, baseURL });
 
     // Ensure plugins are initialized