Fix autotype history fix (#138)

langchain-ai · Oct 31, 2023 · 6e4967a · 6e4967a
1 parent 9b8e9a4
commit 6e4967a
Showing 1 changed file with 14 additions and 4 deletions.
diff --git a/langserve/playground/src/components/ChatMessagesControlRenderer.tsx b/langserve/playground/src/components/ChatMessagesControlRenderer.tsx
@@ -32,12 +32,22 @@ export const chatMessagesTester = rankWith(
       }
 
       if ("anyOf" in schema.items && schema.items.anyOf != null) {
-        return schema.items.anyOf.every(
-          (schema) =>
+        return schema.items.anyOf.every((schema) => {
+          const isObjectMessage =
             schema.type === "object" &&
             (schema.title?.endsWith("Message") ||
-              schema.title?.endsWith("MessageChunk"))
-        );
+              schema.title?.endsWith("MessageChunk"));
+
+          const isTupleMessage =
+            schema.type === "array" &&
+            schema.minItems === 2 &&
+            schema.maxItems === 2 &&
+            Array.isArray(schema.items) &&
+            schema.items.length === 2 &&
+            schema.items.every((schema) => schema.type === "string");
+
+          return isObjectMessage || isTupleMessage;
+        });
       }
 
       return false;