-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
29 lines (23 loc) · 1.16 KB
/
main.py
File metadata and controls
29 lines (23 loc) · 1.16 KB
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
from dataset_generation.generation.dataset_generation import generate
from dataset_generation.generation.dataset_transition import transform
from dataset_generation.generation.instruction_generator import generate_evenly_balanced_dataset
from generic_tool_agents.langchain_agent.evaluation.evaluator import evaluate_atl_langchain_agent
from generic_tool_agents.langgraph_agent.evaluation.evaluator import evaluate_atl_langgraph_agent
from specific_tool_agents.enhanced_specific_agent.evaluation.evaluator import evaluate_enhanced_atl_transformation
from specific_tool_agents.langgraph_agent.evaluation.evaluator import evaluate_atl_transformation_langgraph
if __name__ == "__main__":
# Patterns generation
generate_evenly_balanced_dataset()
# First dataset generation
generate()
# Second dataset generation
transform()
# Evaluation process
# ATL engine agent with LangChain
evaluate_atl_langchain_agent()
# ATL engine agent with LangGraph
evaluate_atl_langgraph_agent()
# ATL transformation-aware agent
evaluate_atl_transformation_langgraph()
# ATL transformation-aware agent
evaluate_enhanced_atl_transformation()