@arizeai/phoenix-client 1.0.0 → 1.0.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@arizeai/phoenix-client",
-  "version": "1.0.0",
+  "version": "1.0.1",
   "description": "A client for the Phoenix API",
   "main": "dist/src/index.js",
   "module": "dist/esm/index.js",

package/src/prompts/sdks/toAI.ts

@@ -2,6 +2,7 @@ import invariant from "tiny-invariant";
 import {
   safelyConvertMessageToProvider,
   safelyConvertToolChoiceToProvider,
+  safelyConvertToolDefinitionToProvider,
 } from "../../schemas/llm/converters";
 import { formatPromptMessages } from "../../utils/formatPromptMessages";
 import { Variables, toSDKParamsBase } from "./types";
@@ -10,7 +11,6 @@ import {
   type generateText,
   type ToolSet,
   type Tool,
-  jsonSchema,
 } from "ai";
 import { VercelAIToolChoice } from "../../schemas/llm/vercel/toolChoiceSchemas";
 
@@ -69,11 +69,12 @@ export const toAI = <V extends Variables>({
       if (!tool.function.parameters) {
         return acc;
       }
-      acc[tool.function.name] = {
-        type: "function",
-        parameters: jsonSchema(tool.function.parameters),
-        description: tool.function.description,
-      } satisfies Tool;
+      const vercelAIToolDefinition = safelyConvertToolDefinitionToProvider({
+        toolDefinition: tool,
+        targetProvider: "VERCEL_AI",
+      });
+      invariant(vercelAIToolDefinition, "Tool definition is not valid");
+      acc[tool.function.name] = vercelAIToolDefinition satisfies Tool;
       return acc;
     }, {} as ToolSet);
     const hasTools = Object.keys(tools ?? {}).length > 0;

package/src/schemas/llm/constants.ts

@@ -16,6 +16,7 @@ import {
   openAIToolChoiceToAnthropic,
   openAIToolChoiceToVercelAI,
   openAIToolDefinitionToAnthropic,
+  openAIToolDefinitionToVercelAI,
 } from "./openai/converters";
 import {
   phoenixResponseFormatToOpenAI,
@@ -150,7 +151,7 @@ export const SDKProviderConverterMap = {
     },
     toolDefinitions: {
       toOpenAI: openAIToolDefinitionSchema,
-      fromOpenAI: z.unknown(),
+      fromOpenAI: openAIToolDefinitionToVercelAI,
     },
   }),
 } satisfies Record<PromptProviderSDKs, unknown>;

package/src/schemas/llm/converters.ts

@@ -378,7 +378,9 @@ export const toOpenAIToolDefinition = (
         validatedToolDefinition
       );
     case "VERCEL_AI":
-      return null;
+      return SDKProviderConverterMap.VERCEL_AI.toolDefinitions.toOpenAI.parse(
+        validatedToolDefinition
+      );
     case null:
       return null;
     default:
@@ -415,7 +417,9 @@ export const fromOpenAIToolDefinition = <
         toolDefinition
       );
     case "VERCEL_AI":
-      return null;
+      return SDKProviderConverterMap.VERCEL_AI.toolDefinitions.fromOpenAI.parse(
+        toolDefinition
+      );
     default:
       assertUnreachable(targetProvider);
   }

package/src/schemas/llm/openai/converters.ts

@@ -28,6 +28,7 @@ import { isObject } from "../../../utils/isObject";
 import { VercelAIToolChoice } from "../vercel/toolChoiceSchemas";
 import { openAIToolDefinitionSchema } from "./toolSchemas";
 import { AnthropicToolDefinition } from "../anthropic/toolSchemas";
+import { VercelAIToolDefinition } from "../vercel/toolSchemas";
 
 export const openAIChatPartToAnthropic = openaiChatPartSchema.transform(
   (openai) => {
@@ -381,3 +382,19 @@ export const openAIToolDefinitionToAnthropic =
       input_schema: openai.function.parameters,
     })
   );
+
+/**
+ * Parse incoming object as an OpenAI tool definition and immediately convert to Vercel AI format
+ */
+export const openAIToolDefinitionToVercelAI =
+  openAIToolDefinitionSchema.transform(
+    (openai): VercelAIToolDefinition => ({
+      type: "function",
+      description: openai.function.description,
+      parameters: {
+        _type: undefined,
+        jsonSchema: openai.function.parameters,
+        validate: undefined,
+      },
+    })
+  );

package/src/schemas/llm/types.ts

@@ -22,6 +22,7 @@ import { OpenAIResponseFormat } from "./openai/responseFormatSchema";
 import { AnthropicToolDefinition } from "./anthropic/toolSchemas";
 import { PhoenixToolDefinition } from "./phoenixPrompt/toolSchemas";
 import { PhoenixContentPart } from "./phoenixPrompt/messagePartSchemas";
+import { VercelAIToolDefinition } from "./vercel/toolSchemas";
 
 export type PromptSDKFormat = PromptProviderSDKs | null;
 
@@ -118,7 +119,7 @@ export type ToolDefinitionWithProvider =
     }
   | {
       provider: Extract<PromptSDKFormat, "VERCEL_AI">;
-      validatedToolDefinition: null;
+      validatedToolDefinition: VercelAIToolDefinition;
     }
   | { provider: null; validatedToolDefinition: null };
 

package/src/schemas/llm/utils.ts

@@ -26,6 +26,7 @@ import { phoenixToolCallSchema } from "./phoenixPrompt/toolCallSchemas";
 import { phoenixToolChoiceSchema } from "./phoenixPrompt/toolChoiceSchemas";
 import { phoenixToolDefinitionSchema } from "./phoenixPrompt/toolSchemas";
 import { phoenixContentPartSchema } from "./phoenixPrompt/messagePartSchemas";
+import { vercelAIToolDefinitionSchema } from "./vercel/toolSchemas";
 
 export const makeSDKConverters = <
   MessageSchema extends ZodTypeAny,
@@ -208,6 +209,11 @@ export const detectToolDefinitionProvider = (
   if (phoenixSuccess) {
     return { provider: "PHOENIX", validatedToolDefinition: phoenixData };
   }
+  const { success: vercelSuccess, data: vercelData } =
+    vercelAIToolDefinitionSchema.safeParse(toolDefinition);
+  if (vercelSuccess) {
+    return { provider: "VERCEL_AI", validatedToolDefinition: vercelData };
+  }
   return { provider: null, validatedToolDefinition: null };
 };
 

package/src/schemas/llm/vercel/toolSchemas.ts

@@ -0,0 +1,21 @@
+import { z } from "zod";
+
+/**
+ * Vercel AI Tool Definition Schema
+ *
+ * Produces the same result as if you called `import { jsonSchema } from "ai"`
+ * https://github.com/vercel/ai/blob/83976abfa99bf26f8227cf493386b8a6f0e71fdd/packages/ui-utils/src/schema.ts#L34
+ */
+export const vercelAIToolDefinitionSchema = z.object({
+  type: z.literal("function"),
+  description: z.string().optional(),
+  parameters: z.object({
+    _type: z.unknown().optional().default(undefined),
+    validate: z.unknown().optional().default(undefined),
+    jsonSchema: z.record(z.string(), z.unknown()).optional(),
+  }),
+});
+
+export type VercelAIToolDefinition = z.infer<
+  typeof vercelAIToolDefinitionSchema
+>;