File size: 1,756 Bytes
cff1e0e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
"""
Evaluators module.

Import this module to automatically register all evaluators.
"""
from evaluators.base import Evaluator
from evaluators.registry import (
    register_evaluator,
    get_evaluator_class,
    create_evaluator,
    list_available_metrics,
    get_registry
)

# Import all evaluator implementations to register them
# Add new evaluators here as they are created
try:
    from evaluators.impl.talk_type_evaluator import TalkTypeEvaluator
except ImportError:
    pass

try:
    from evaluators.impl.empathy_er_evaluator import EmpathyEREvaluator
except ImportError:
    pass

try:
    from evaluators.impl.empathy_ip_evaluator import EmpathyIPEvaluator
except ImportError:
    pass

try:
    from evaluators.impl.empathy_ex_evaluator import EmpathyEXEvaluator
except ImportError:
    pass

try:
    from evaluators.impl.factuality_evaluator import MentalHealthFactualityEvaluator
except ImportError:
    pass

try:
    from evaluators.impl.emotion_evaluator import EmotionEvaluator
except ImportError:
    pass

try:
    from evaluators.impl.toxicity_evaluator import ToxicityEvaluator
except ImportError:
    pass
# Import examples (optional, for testing)
try:
    from evaluators.examples.example_evaluators import (
        ExampleUtteranceEvaluator,
        ExampleConversationEvaluator,
        ExampleSegmentEvaluator,
        ExampleMixedEvaluator
    )
except ImportError:
    pass

from evaluators.registry import (
    get_ui_labels,
    get_metrics_by_category,
    get_metric_metadata
)

__all__ = [
    "Evaluator",
    "register_evaluator",
    "get_evaluator_class",
    "create_evaluator",
    "list_available_metrics",
    "get_ui_labels",
    "get_metrics_by_category",
    "get_metric_metadata",
    "get_registry",
]