88sys .path .insert (0 , os .path .dirname (os .path .abspath (__file__ )))
99
1010from praisonaiagents import Agent , Task , PraisonAIAgents
11- from praisonaiagents .agent import MultiModelAgent
11+ from praisonaiagents .agent import RouterAgent
1212from praisonaiagents .llm import ModelRouter , TaskComplexity
1313
1414def test_model_router ():
@@ -49,12 +49,12 @@ def test_model_router():
4949 print ("\n ✓ ModelRouter tests completed" )
5050
5151
52- def test_multi_model_agent ():
53- """Test the MultiModelAgent functionality"""
54- print ("\n === Testing MultiModelAgent ===\n " )
52+ def test_router_agent ():
53+ """Test the RouterAgent functionality"""
54+ print ("\n === Testing RouterAgent ===\n " )
5555
56- # Create a multi-model agent
57- agent = MultiModelAgent (
56+ # Create a router agent
57+ agent = RouterAgent (
5858 name = "Test Agent" ,
5959 role = "Test Assistant" ,
6060 goal = "Test multi-model functionality" ,
@@ -74,15 +74,15 @@ def test_multi_model_agent():
7474 )
7575 print (f"Selected model for simple task: { selected } " )
7676
77- print ("\n ✓ MultiModelAgent tests completed" )
77+ print ("\n ✓ RouterAgent tests completed" )
7878
7979
8080def test_integration ():
8181 """Test integration with PraisonAIAgents"""
8282 print ("\n === Testing Integration ===\n " )
8383
84- # Create a simple multi-model agent
85- agent = MultiModelAgent (
84+ # Create a simple router agent
85+ agent = RouterAgent (
8686 name = "Integration Test Agent" ,
8787 role = "Tester" ,
8888 goal = "Test the integration" ,
@@ -107,7 +107,7 @@ def test_integration():
107107 verbose = False
108108 )
109109
110- print ("Created PraisonAIAgents with MultiModelAgent " )
110+ print ("Created PraisonAIAgents with RouterAgent " )
111111 print ("✓ Integration test setup completed" )
112112
113113 # Note: Actual execution would require API keys
@@ -119,14 +119,14 @@ def main():
119119
120120 try :
121121 test_model_router ()
122- test_multi_model_agent ()
122+ test_router_agent ()
123123 test_integration ()
124124
125125 print ("\n ✅ All tests completed successfully!" )
126126 print ("\n 📝 Summary:" )
127127 print ("- ModelRouter can analyze task complexity" )
128128 print ("- ModelRouter can select appropriate models" )
129- print ("- MultiModelAgent can be created and configured" )
129+ print ("- RouterAgent can be created and configured" )
130130 print ("- Integration with PraisonAIAgents works" )
131131 print ("\n 🎉 Multi-provider support is ready to use!" )
132132
0 commit comments