@arizeai/phoenix-client 6.9.1 → 6.9.2

package/src/schemas/llm/utils.ts

@@ -15,7 +15,7 @@ import { phoenixContentPartSchema } from "./phoenixPrompt/messagePartSchemas";
 import { phoenixMessageSchema } from "./phoenixPrompt/messageSchemas";
 import { phoenixToolCallSchema } from "./phoenixPrompt/toolCallSchemas";
 import { phoenixToolChoiceSchema } from "./phoenixPrompt/toolChoiceSchemas";
-import { phoenixToolDefinitionSchema } from "./phoenixPrompt/toolSchemas";
+import { phoenixToolFunctionDefinitionSchema } from "./phoenixPrompt/toolSchemas";
 import { llmProviderToolDefinitionSchema } from "./schemas";
 import type {
   LLMMessagePart,
@@ -206,7 +206,7 @@ export const detectToolDefinitionProvider = (
     };
   }
   const { success: phoenixSuccess, data: phoenixData } =
-    phoenixToolDefinitionSchema.safeParse(toolDefinition);
+    phoenixToolFunctionDefinitionSchema.safeParse(toolDefinition);
   if (phoenixSuccess) {
     return { provider: "PHOENIX", validatedToolDefinition: phoenixData };
   }

package/src/types/prompts.ts

@@ -161,11 +161,24 @@ export type PromptChatMessagePart =
   | components["schemas"]["ToolCallContentPart"]
   | components["schemas"]["ToolResultContentPart"];
 
+export type PromptToolFunction = components["schemas"]["PromptToolFunction"];
+export type PromptToolRaw = components["schemas"]["PromptToolRaw"];
+
 /**
  * The Phoenix prompt tool type from the API.
  */
 export type PromptTool = components["schemas"]["PromptTools"]["tools"][number];
 
+export function isPromptToolRaw(tool: PromptTool): tool is PromptToolRaw {
+  return tool.type === "raw";
+}
+
+export function isPromptToolFunction(
+  tool: PromptTool
+): tool is PromptToolFunction {
+  return tool.type === "function";
+}
+
 /**
  * The Phoenix prompt tool choice type from the API.
  */