fix: fixing params in test

ucbepic · Oct 9, 2024 · ffe0b18 · ffe0b18
1 parent 9afd468
commit ffe0b18
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/basic/test_cluster.py b/tests/basic/test_cluster.py
@@ -10,7 +10,7 @@ def cluster_config():
         "type": "cluster",
         "embedding_keys": ["concept", "description"],
         "output_key": "categories",
-        "output_schema": {"concept": "string", "description": "string"},
+        "summary_schema": {"concept": "string", "description": "string"},
         "summary_prompt": """
         The following describes two related concepts. What concept
         encompasses both? Try not to be too broad; it might be that one of