feat(playground): minimal content part refusal support (#662)

Yonom · Aug 8, 2024 · f03d4da · f03d4da
1 parent d14d604
commit f03d4da
Showing 1 changed file with 32 additions and 2 deletions.
diff --git a/packages/react-playground/src/lib/openai/fromOpenAIMessages.ts b/packages/react-playground/src/lib/openai/fromOpenAIMessages.ts
@@ -16,7 +16,13 @@ export const fromOpenAIMessages = (
     const { role, content } = message;
     switch (role) {
       case "system": {
-        lmMessages.push({ role: "system", content });
+        lmMessages.push({
+          role: "system",
+          content:
+            typeof content === "string"
+              ? content
+              : content.map((c) => c.text).join("\n"),
+        });
         break;
       }
 
@@ -57,7 +63,31 @@ export const fromOpenAIMessages = (
         )[] = [];
 
         if (content != null) {
-          lmContent.push({ type: "text", text: content });
+          lmContent.push(
+            ...(typeof content === "string"
+              ? [{ type: "text" as const, text: content }]
+              : content.map((part) => {
+                  const type = part.type;
+                  switch (type) {
+                    case "text": {
+                      return part;
+                    }
+
+                    case "refusal": {
+                      // TODO handle refusals
+                      return {
+                        type: "text" as const,
+                        text: part.refusal,
+                      };
+                    }
+
+                    default: {
+                      const unsupportedPart: "refusal" = type;
+                      throw new Error(`Unsupported part: ${unsupportedPart}`);
+                    }
+                  }
+                })),
+          );
         }
 
         if (message.tool_calls) {