@arizeai/phoenix-client
Version:
A client for the Phoenix API
67 lines • 1.88 kB
JavaScript
import { openaiChatPartImageSchema, openaiChatPartTextSchema, } from "./messagePartSchemas.js";
import { openAIToolCallSchema } from "./toolCallSchemas.js";
import z from "zod";
/*
*
* OpenAI Message Schemas
*
*/
export const openAIMessageRoleSchema = z.enum([
"system",
"user",
"assistant",
"developer",
"tool",
// "function",
]);
export const openAIMessageSchema = z.discriminatedUnion("role", [
z
.object({
role: z.literal("assistant"),
content: z.union([openaiChatPartTextSchema.array(), z.string()]),
name: z.string().optional(),
tool_call_id: z.string().optional(),
tool_calls: z.array(openAIToolCallSchema).optional(),
})
.passthrough(),
z
.object({
role: z.literal("tool"),
content: z.union([openaiChatPartTextSchema.array(), z.string()]),
tool_call_id: z.string(),
})
.passthrough(),
z
.object({
role: z.literal("function"),
content: z.string().nullable(),
name: z.string(),
})
.passthrough(),
z
.object({
role: z.literal("user"),
content: z.union([
z.array(z.union([openaiChatPartTextSchema, openaiChatPartImageSchema])),
z.string(),
]),
name: z.string().optional(),
})
.passthrough(),
z
.object({
role: z.literal("system"),
content: z.union([openaiChatPartTextSchema.array(), z.string()]),
name: z.string().optional(),
})
.passthrough(),
z
.object({
role: z.literal("developer"),
content: z.union([openaiChatPartTextSchema.array(), z.string()]),
name: z.string().optional(),
})
.passthrough(),
]);
export const openAIMessagesSchema = z.array(openAIMessageSchema);
//# sourceMappingURL=messageSchemas.js.map