fix: DIA-1489: backwards compatible autorefinement with non-multiskill (

#241)
HumanSignal · Oct 31, 2024 · 2bf8169 · 2bf8169
1 parent 3665a80
commit 2bf8169
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/adala/skills/collection/prompt_improvement.py b/adala/skills/collection/prompt_improvement.py
@@ -70,6 +70,9 @@ def get_json_template(fields):
         else:
             model_json_schema = self.skill_to_improve.response_model.model_json_schema()
 
+        # TODO: can remove this when only LabelStudioSkill is supported
+        label_config = getattr(self.skill_to_improve, 'label_config', '<View>Not available</View>')
+
         input_variables = self.input_variables
         output_variables = list(model_json_schema['properties'].keys())
         input_json_template = get_json_template(input_variables)
@@ -84,7 +87,7 @@ def get_json_template(fields):
 
 ## Current Labeling Config:
 ```xml
-{self.skill_to_improve.label_config}
+{label_config}
 ```
 
 ## Input variables: