File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -43,7 +43,7 @@ def test_main_with_internet_search_tool(self):
4343
4444 @pytest .mark .real
4545 def test_main_with_built_in_tool (self ):
46- praisonai = PraisonAI (agent_file = "tests/built-in-tools -agents.yaml" )
46+ praisonai = PraisonAI (agent_file = "tests/inbuilt-tool -agents.yaml" )
4747 result = praisonai .run ()
4848 print (f"Result: { result } " )
4949 self .assertIsNotNone (result )
@@ -77,7 +77,10 @@ def test_praisonai_init_command(self):
7777 command = "praisonai --framework autogen --init \" create a 2-agent team to write a simple python game\" "
7878 result = self .run_command (command )
7979 print (f"Result: { result } " )
80- self .assertIn ('created successfully' , result )
80+ # Check for success indicator in the output
81+ success_phrases = ['created successfully' , 'agents.yaml created' , 'File created' ]
82+ found_success = any (phrase in result for phrase in success_phrases )
83+ self .assertTrue (found_success , f"No success message found. Expected one of { success_phrases } . Last 1000 chars: { result [- 1000 :]} " )
8184
8285class TestExamples (unittest .TestCase ):
8386 def test_advanced_example (self ):
You can’t perform that action at this time.
0 commit comments