uncomment get_evaluated_component

src/utils.py

@@ -58,22 +58,22 @@ def get_result_description(sub_catalog_name, criteria_name):
     }
     return messages[criteria_name]
 
-# def get_evaluated_component(sub_catalog_name, criteria_name):
-#     component = None
-#     if sub_catalog_name == 'harmful_content_in_user_prompt':
-#         component = "user"
-#     elif sub_catalog_name == 'harmful_content_in_assistant_response':
-#         component = 'assistant'
-#     elif sub_catalog_name == 'rag_hallucination_risks':
-#         if criteria_name == "context_relevance":
-#             component = "context"
-#         elif criteria_name == "groundedness":
-#             component = "assistant"
-#         elif criteria_name == "answer_relevance":
-#             component = "assistant"
-#     if component is None:
-#         raise Exception('Something went wrong getting the evaluated component')
-#     return component
+def get_evaluated_component(sub_catalog_name, criteria_name):
+    component = None
+    if sub_catalog_name == 'harmful_content_in_user_prompt':
+        component = "user"
+    elif sub_catalog_name == 'harmful_content_in_assistant_response':
+        component = 'assistant'
+    elif sub_catalog_name == 'rag_hallucination_risks':
+        if criteria_name == "context_relevance":
+            component = "context"
+        elif criteria_name == "groundedness":
+            component = "assistant"
+        elif criteria_name == "answer_relevance":
+            component = "assistant"
+    if component is None:
+        raise Exception('Something went wrong getting the evaluated component')
+    return component
 
 def to_title_case(input_string):
     if input_string == 'rag_hallucination_risks':