fix(LAB-3307): fix model name on LLM static export

kili-technology · Jan 9, 2025 · 26b3f8a · 26b3f8a
1 parent 8779b87
commit 26b3f8a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/kili/llm/services/export/dynamic.py b/src/kili/llm/services/export/dynamic.py
@@ -51,7 +51,8 @@ def export(
             else:
                 model_names = {
                     chat_item.get("modelName")
-                    for chat_item in asset["labels"][0]["chatItems"]
+                    for label in asset["labels"]
+                    for chat_item in label["chatItems"]
                     if chat_item.get("modelName") is not None
                 }
                 obfuscated_models = {