@arizeai/phoenix-client 5.9.0 → 6.0.0

package/src/schemas/llm/phoenixPrompt/responseFormatSchema.ts

@@ -13,7 +13,7 @@ export const phoenixResponseFormatSchema =
       json_schema: z.object({
         name: z.string(),
         description: z.string().optional(),
-        schema: z.record(z.unknown()).optional(),
+        schema: z.record(z.string(), z.unknown()).optional(),
         strict: z.boolean().optional(),
       }),
     })

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

@@ -12,7 +12,7 @@ export const phoenixToolDefinitionSchema = schemaMatches<PromptTool>()(
     function: z.object({
       name: z.string(),
       description: z.string().optional(),
-      parameters: z.record(z.unknown()).optional(),
+      parameters: z.record(z.string(), z.unknown()).optional(),
       strict: z.boolean().optional(),
     }),
   })

package/src/schemas/llm/schemas.ts

@@ -61,11 +61,11 @@ export type LlmProviderToolCalls = z.infer<typeof llmProviderToolCallsSchema>;
 export const toolCallHeuristicSchema = z.object({
   id: z.string().optional(),
   name: z.string().optional(),
-  arguments: z.record(z.unknown()).optional(),
+  arguments: z.record(z.string(), z.unknown()).optional(),
   function: z
     .object({
       name: z.string().optional(),
-      arguments: z.record(z.unknown()).optional(),
+      arguments: z.record(z.string(), z.unknown()).optional(),
     })
     .optional(),
 });

package/src/schemas/llm/types.ts

@@ -1,4 +1,4 @@
-import type { ZodType, ZodTypeAny, ZodTypeDef } from "zod";
+import type { ZodType } from "zod";
 
 import type { PromptProviderSDKs, PromptToolChoice } from "../../types/prompts";
 import type { AnthropicMessagePart } from "./anthropic/messagePartSchemas";
@@ -33,35 +33,35 @@ export type LLMMessagePart =
   | VercelAIChatPart;
 
 export type SDKConverters<
-  MessageSchema extends ZodTypeAny = never,
-  MessagePartSchema extends ZodTypeAny = never,
-  ToolChoiceSchema extends ZodTypeAny = never,
-  ToolCallSchema extends ZodTypeAny = never,
-  ToolDefinitionSchema extends ZodTypeAny = never,
-  ResponseFormatSchema extends ZodTypeAny = never,
+  MessageSchema extends ZodType = never,
+  MessagePartSchema extends ZodType = never,
+  ToolChoiceSchema extends ZodType = never,
+  ToolCallSchema extends ZodType = never,
+  ToolDefinitionSchema extends ZodType = never,
+  ResponseFormatSchema extends ZodType = never,
 > = {
   messages: {
-    toOpenAI: ZodType<OpenAIMessage | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenAIMessage | null, unknown>;
     fromOpenAI: MessageSchema;
   };
   messageParts: {
-    toOpenAI: ZodType<OpenAIChatPart | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenAIChatPart | null, unknown>;
     fromOpenAI: MessagePartSchema;
   };
   toolChoices: {
-    toOpenAI: ZodType<OpenaiToolChoice | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenaiToolChoice | null, unknown>;
     fromOpenAI: ToolChoiceSchema;
   };
   toolCalls: {
-    toOpenAI: ZodType<OpenAIToolCall | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenAIToolCall | null, unknown>;
     fromOpenAI: ToolCallSchema;
   };
   toolDefinitions: {
-    toOpenAI: ZodType<OpenAIToolDefinition | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenAIToolDefinition | null, unknown>;
     fromOpenAI: ToolDefinitionSchema;
   };
   responseFormat?: {
-    toOpenAI: ZodType<OpenAIResponseFormat | null, ZodTypeDef, unknown>;
+    toOpenAI: ZodType<OpenAIResponseFormat | null, unknown>;
     fromOpenAI: ResponseFormatSchema;
   };
 };

package/src/schemas/llm/utils.ts

@@ -1,4 +1,4 @@
-import type { ZodTypeAny } from "zod";
+import type { ZodType } from "zod";
 
 import { isObject } from "../../utils/isObject";
 import { anthropicMessagePartSchema } from "./anthropic/messagePartSchemas";
@@ -30,12 +30,12 @@ import { vercelAIMessageSchema } from "./vercel/messageSchemas";
 import { vercelAIToolDefinitionSchema } from "./vercel/toolSchemas";
 
 export const makeSDKConverters = <
-  MessageSchema extends ZodTypeAny,
-  MessagePartSchema extends ZodTypeAny,
-  ToolChoiceSchema extends ZodTypeAny,
-  ToolCallSchema extends ZodTypeAny,
-  ToolDefinitionSchema extends ZodTypeAny,
-  ResponseFormatSchema extends ZodTypeAny,
+  MessageSchema extends ZodType,
+  MessagePartSchema extends ZodType,
+  ToolChoiceSchema extends ZodType,
+  ToolCallSchema extends ZodType,
+  ToolDefinitionSchema extends ZodType,
+  ResponseFormatSchema extends ZodType,
 >({
   messages,
   messageParts,

package/src/utils/schemaMatches.ts

@@ -15,7 +15,7 @@ export type AssertEqual<T, U> =
  */
 export const schemaMatches =
   <T>() =>
-  <S extends z.ZodType<T, z.ZodTypeDef, unknown>>(
+  <S extends z.ZodType<T, unknown>>(
     schema: AssertEqual<S["_output"], T> extends true
       ? S
       : S & {