@@ -10,75 +10,27 @@ class TestPraisonAIFramework(unittest.TestCase):
1010 @pytest .mark .real
1111 def test_main_with_autogen_framework (self ):
1212 """Test AutoGen framework integration with real API calls"""
13- # Check if we have a real API key
14- api_key = os .environ .get ('OPENAI_API_KEY' )
15- if not api_key or api_key == 'test-key' or api_key .startswith ('sk-test-' ):
16- self .skipTest ("Skipping real test - no valid OPENAI_API_KEY provided" )
17-
1813 praisonai = PraisonAI (agent_file = 'tests/autogen-agents.yaml' )
19- try :
20- result = praisonai .run ()
21- self .assertIn ('### Output ###' , result )
22- except Exception as e :
23- if ('Invalid API Key' in str (e ) or 'AuthenticationError' in str (e ) or
24- 'InstructorRetryException' in str (e ) or '401' in str (e )):
25- self .skipTest (f"Skipping due to API authentication: { e } " )
26- else :
27- raise
14+ result = praisonai .run ()
15+ self .assertIn ('### Output ###' , result )
2816
2917 @pytest .mark .real
3018 def test_main_with_custom_framework (self ):
3119 """Test CrewAI framework integration with real API calls"""
32- # Check if we have a real API key
33- api_key = os .environ .get ('OPENAI_API_KEY' )
34- if not api_key or api_key == 'test-key' or api_key .startswith ('sk-test-' ):
35- self .skipTest ("Skipping real test - no valid OPENAI_API_KEY provided" )
36-
3720 praisonai = PraisonAI (agent_file = 'tests/crewai-agents.yaml' )
38- try :
39- result = praisonai .run ()
40- self .assertIn ('### Task Output ###' , result )
41- except Exception as e :
42- if ('Invalid API Key' in str (e ) or 'AuthenticationError' in str (e ) or
43- 'InstructorRetryException' in str (e ) or '401' in str (e )):
44- self .skipTest (f"Skipping due to API authentication: { e } " )
45- else :
46- raise
21+ result = praisonai .run ()
22+ self .assertIn ('### Task Output ###' , result )
4723
4824 @pytest .mark .real
4925 def test_main_with_internet_search_tool (self ):
5026 """Test internet search tool integration with real API calls"""
51- # Check if we have a real API key
52- api_key = os .environ .get ('OPENAI_API_KEY' )
53- if not api_key or api_key == 'test-key' or api_key .startswith ('sk-test-' ):
54- self .skipTest ("Skipping real test - no valid OPENAI_API_KEY provided" )
55-
5627 praisonai = PraisonAI (agent_file = 'tests/search-tool-agents.yaml' )
57- try :
58- result = praisonai .run ()
59- self .assertIn ('### Task Output ###' , result )
60- except Exception as e :
61- if ('Invalid API Key' in str (e ) or 'AuthenticationError' in str (e ) or
62- 'InstructorRetryException' in str (e ) or '401' in str (e )):
63- self .skipTest (f"Skipping due to API authentication: { e } " )
64- else :
65- raise
28+ result = praisonai .run ()
29+ self .assertIn ('### Task Output ###' , result )
6630
6731 @pytest .mark .real
6832 def test_main_with_built_in_tool (self ):
6933 """Test built-in tool integration with real API calls"""
70- # Check if we have a real API key
71- api_key = os .environ .get ('OPENAI_API_KEY' )
72- if not api_key or api_key == 'test-key' or api_key .startswith ('sk-test-' ):
73- self .skipTest ("Skipping real test - no valid OPENAI_API_KEY provided" )
74-
7534 praisonai = PraisonAI (agent_file = 'tests/inbuilt-tool-agents.yaml' )
76- try :
77- result = praisonai .run ()
78- self .assertIn ('### Task Output ###' , result )
79- except Exception as e :
80- if ('Invalid API Key' in str (e ) or 'AuthenticationError' in str (e ) or
81- 'InstructorRetryException' in str (e ) or '401' in str (e )):
82- self .skipTest (f"Skipping due to API authentication: { e } " )
83- else :
84- raise
35+ result = praisonai .run ()
36+ self .assertIn ('### Task Output ###' , result )
0 commit comments