From 3a27d38913ad95e3493da0640228a78480ef8add Mon Sep 17 00:00:00 2001
From: LennartSchmidtKern <lennart.schmidt@kern.ai>
Date: Tue, 22 Jul 2025 17:31:10 +0200
Subject: [PATCH] add text List

---
 run_ac.py | 19 +++++++++++++++----
 1 file changed, 15 insertions(+), 4 deletions(-)

diff --git a/run_ac.py b/run_ac.py
index 1e5cfaf..f01f622 100755
--- a/run_ac.py
+++ b/run_ac.py
@@ -24,6 +24,8 @@ def get_check_data_type_function(data_type: str) -> Tuple[List[Type], Callable]:
         return [list], __check_data_type_embedding_list
     elif data_type == "LLM_RESPONSE":
         return [str], __check_data_type_text
+    elif data_type == "TEXT_LIST":
+        return [list], __check_data_type_text_list
     else:
         raise ValueError(f"Unknown data type: {data_type}")
 
@@ -73,6 +75,15 @@ def __check_data_type_embedding_list(attr_value: Any) -> bool:
     return True
 
 
+def __check_data_type_text_list(attr_value: Any) -> bool:
+    if not isinstance(attr_value, list):
+        return False
+    for e in attr_value:
+        if not isinstance(e, str):
+            return False
+    return True
+
+
 def __print_progress_a2vybg(progress: float) -> None:
     print(f"progress: {progress}", flush=True)
 
@@ -98,7 +109,7 @@ def load_data_dict_a2vybg(record: Dict[str, Any]) -> Dict[str, Any]:
 
 
 def parse_data_to_record_dict_a2vybg(
-    record_chunk: List[Dict[str, Any]]
+    record_chunk: List[Dict[str, Any]],
 ) -> List[Dict[str, Any]]:
     result = []
     for r in record_chunk:
@@ -134,7 +145,7 @@ def save_ac_value_a2vybg(record_id: str, attr_value: Any) -> None:
 
 
 def process_attribute_calculation_a2vybg(
-    record_dict_list: List[Dict[str, Any]]
+    record_dict_list: List[Dict[str, Any]],
 ) -> None:
     for record_dict in record_dict_list:
         attr_value: Any = attribute_calculators.ac(record_dict["data"])
@@ -147,7 +158,7 @@ def check_abort_status_a2vybg() -> bool:
 
 
 async def process_llm_record_batch_a2vybg(
-    record_dict_batch: List[Dict[str, Any]]
+    record_dict_batch: List[Dict[str, Any]],
 ) -> None:
     global should_abort_a2vybg
 
@@ -178,7 +189,7 @@ def make_batches(
 
 
 async def process_async_llm_calls_a2vybg(
-    record_dict_list: List[Dict[str, Any]]
+    record_dict_list: List[Dict[str, Any]],
 ) -> None:
     batch_size = max(amount_a2vybg // int(attribute_calculators.NUM_WORKERS_A2VYBG), 1)
     tasks = [