Add length check for enum (#794)

holoviz · Nov 27, 2024 · a65fb7a · a65fb7a
1 parent 89d36b9
commit a65fb7a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lumen/ai/utils.py b/lumen/ai/utils.py
@@ -178,7 +178,7 @@ async def get_schema(
             spec.pop("enum")
         elif limit and len(spec["enum"]) > limit:
             spec["enum"].append("...")
-        elif limit and spec["enum"][0] is None:
+        elif limit and len(spec["enum"]) == 1 and spec["enum"][0] is None:
             spec["enum"] = [f"(unknown; truncated to {get_kwargs['limit']} rows)"]
 
     schema = format_schema(schema)