diff --git a/src/kili/llm/services/export/dynamic.py b/src/kili/llm/services/export/dynamic.py index 3e122d001..d2d264c28 100644 --- 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 = {