refactor: streamline test functions in test_scheduler.py

.github/workflows/python-package.yml

@@ -40,6 +40,8 @@ jobs:
         python -m pip install ."[ui,gradio,api,agentops,google,openai,anthropic,cohere,chat,code,realtime,call,crewai,autogen]"
         python -m pip install duckduckgo_search
         python -m pip install pytest-asyncio pytest-cov
+        # Install knowledge dependencies from praisonai-agents
+        python -m pip install "praisonaiagents[knowledge]"
         if [ -f requirements.txt ]; then pip install -r requirements.txt; fi
 
     - name: Debug API Key Status

.github/workflows/test-comprehensive.yml

@@ -51,6 +51,8 @@ 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
         uv pip install --system pytest pytest-asyncio pytest-cov pytest-benchmark
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Set environment variables
       run: |

.github/workflows/test-core.yml

@@ -34,6 +34,8 @@ 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
         uv pip install --system pytest pytest-asyncio pytest-cov
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Set environment variables
       run: |

.github/workflows/test-extended.yml

@@ -34,6 +34,8 @@ jobs:
         cd src/praisonai
         uv pip install --system ."[ui,gradio,api,agentops,google,openai,anthropic,cohere,chat,code,realtime,call,crewai,autogen]"
         uv pip install --system duckduckgo_search
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Set environment variables
       run: |
@@ -92,6 +94,8 @@ jobs:
         cd src/praisonai
         uv pip install --system ."[ui,gradio,api,agentops,google,openai,anthropic,cohere,chat,code,realtime,call,crewai,autogen]"
         uv pip install --system pytest pytest-benchmark
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Run Performance Benchmarks
       run: |

.github/workflows/test-real.yml

@@ -58,6 +58,8 @@ jobs:
         cd src/praisonai
         uv pip install --system ."[ui,gradio,api,agentops,google,openai,anthropic,cohere,chat,code,realtime,call,crewai,autogen]"
         uv pip install --system pytest pytest-asyncio pytest-cov
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Set environment variables
       run: |

.github/workflows/unittest.yml

@@ -27,6 +27,8 @@ 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
         uv pip install --system pytest pytest-asyncio pytest-cov
+        # Install knowledge dependencies from praisonai-agents
+        uv pip install --system "praisonaiagents[knowledge]"
 
     - name: Set environment variables
       run: |

src/praisonai-agents/praisonaiagents/memory/__init__.py

@@ -0,0 +1,15 @@
+"""
+Memory module for PraisonAI Agents
+
+This module provides memory management capabilities including:
+- Short-term memory (STM) for ephemeral context
+- Long-term memory (LTM) for persistent knowledge  
+- Entity memory for structured data
+- User memory for preferences/history
+- Quality-based storage decisions
+- Graph memory support via Mem0
+"""
+
+from .memory import Memory
+
+__all__ = ["Memory"]