Enhance GitHub Actions workflows and update test assertions

- Added environment variable setup in `test-extended.yml` for OpenAI API configuration.
- Updated test command in `unittest.yml` to use a pattern for faster execution.
- Refined assertions in `test_base_url_api_base_fix.py` to reflect changes in parameter naming from `base_url` to `api_base`.
- Improved mock tool listing in `test_mcp_integration.py` for clarity.
- Adjusted knowledge update logic in `test_rag_integration.py` to handle existing knowledge more effectively.
- Ensured minimal changes to existing code while enhancing test clarity and functionality.

Author: MervinPraison

.github/workflows/test-extended.yml

@@ -33,6 +33,13 @@ jobs:
         uv pip install --system ."[ui,gradio,api,agentops,google,openai,anthropic,cohere,chat,code,realtime,call,crewai,autogen]"
         uv pip install --system duckduckgo_search
 
+    - name: Set environment variables
+      run: |
+        echo "OPENAI_API_KEY=${{ secrets.OPENAI_API_KEY }}" >> $GITHUB_ENV
+        echo "OPENAI_API_BASE=${{ secrets.OPENAI_API_BASE }}" >> $GITHUB_ENV
+        echo "OPENAI_MODEL_NAME=${{ secrets.OPENAI_MODEL_NAME }}" >> $GITHUB_ENV
+        echo "PYTHONPATH=${{ github.workspace }}/src/praisonai-agents:$PYTHONPATH" >> $GITHUB_ENV
+
     - name: Test Key Example Scripts
       run: |
         echo "🧪 Testing key example scripts from praisonai-agents..."

.github/workflows/unittest.yml

@@ -36,7 +36,7 @@ jobs:
     - name: Run Fast Tests
       run: |
         # Run the fastest, most essential tests
-        python tests/test_runner.py --fast
+        python tests/test_runner.py --pattern fast
 
     - name: Run Legacy Example Tests
       run: |

tests/integration/test_base_url_api_base_fix.py

@@ -92,19 +92,19 @@ def test_agent_with_llm_dict_base_url_parameter(self, mock_completion):
 
     @patch('litellm.image_generation')
     def test_image_agent_base_url_consistency(self, mock_image_generation):
-        """Test that ImageAgent maintains parameter consistency with base_url."""
+        """Test that ImageAgent maintains parameter consistency with api_base."""
         mock_image_generation.return_value = {
             'data': [{'url': 'http://example.com/image.png'}]
         }
 
         image_agent = ImageAgent(
-            base_url='http://localhost:4000',
+            api_base='http://localhost:4000',
             api_key='sk-test'
         )
 
-        # Verify that ImageAgent was created with base_url
-        assert image_agent.base_url == 'http://localhost:4000'
-        assert image_agent.api_key == 'sk-test'
+        # Verify that ImageAgent was created with api_base
+        assert image_agent.image_config.api_base == 'http://localhost:4000'
+        assert image_agent.image_config.api_key == 'sk-test'
 
     @patch('litellm.completion')
     def test_koboldcpp_specific_scenario(self, mock_completion):

tests/integration/test_mcp_integration.py

@@ -30,11 +30,11 @@ async def test_mcp_server_connection(self):
                 mock_session = AsyncMock()
                 mock_session_class.return_value.__aenter__.return_value = mock_session
 
-                # Mock session methods
+                # Mock tool listing
                 mock_session.initialize.return_value = None
-                mock_session.list_tools.return_value = Mock(tools=[
-                    Mock(name='get_stock_price', description='Get stock price')
-                ])
+                mock_tool = Mock()
+                mock_tool.name = 'get_stock_price'  # Set as string, not Mock
+                mock_session.list_tools.return_value = Mock(tools=[mock_tool])
 
                 # Test MCP connection simulation
                 async with mock_stdio_client(Mock()) as (read, write):

tests/integration/test_rag_integration.py

@@ -395,14 +395,20 @@ def test_rag_knowledge_update(self, sample_agent_config):
         def mock_update_knowledge(agent, new_documents: list, mode: str = "append"):
             """Mock updating agent knowledge."""
             if mode == "append":
-                current_knowledge = getattr(agent, 'knowledge', [])
-                updated_knowledge = current_knowledge + new_documents
+                current_knowledge = getattr(agent, 'knowledge', None)
+                if current_knowledge is not None:
+                    # If Knowledge object exists, get count from it
+                    previous_count = 1  # Mock that there's existing knowledge
+                else:
+                    previous_count = 0
+                updated_count = previous_count + len(new_documents)
             else:  # replace
-                updated_knowledge = new_documents
+                previous_count = 1 if getattr(agent, 'knowledge', None) else 0
+                updated_count = len(new_documents)
 
             return {
-                'previous_count': len(getattr(agent, 'knowledge', [])),
-                'new_count': len(updated_knowledge),
+                'previous_count': previous_count,
+                'new_count': updated_count,
                 'added_documents': new_documents
             }