mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-07-01 13:18:10 +00:00
8919026326
Some checks failed
Add opt-in extension seams so an application can route memory, knowledge, RAG, and flow persistence through a custom backend without subclassing or threading an explicit instance through every construction site -- mirroring the existing crewai_core.lock_store.set_lock_backend seam. - memory: crewai.memory.storage.factory.set_memory_storage_factory - knowledge: crewai.knowledge.storage.factory.set_knowledge_storage_factory - rag: crewai.rag.factory.register_rag_client_factory (provider registry) - flow: crewai.flow.persistence.factory.set_flow_persistence_factory Each construction site consults the registered factory and falls back to the built-in default when none is set; an explicit instance always wins. Widen Knowledge.storage and the knowledge source base classes to BaseKnowledgeStorage (consistent with BaseAgent.knowledge_storage) so any base-interface backend plugs in. Runtime-free tests cover each seam.
69 lines
2.0 KiB
Python
69 lines
2.0 KiB
Python
"""Tests for the pluggable flow persistence factory seam.
|
|
|
|
We verify our own logic: that ``default_flow_persistence`` returns the
|
|
registered factory's result, and that it falls back to the built-in SQLite
|
|
persistence when no factory is registered.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
from pydantic import BaseModel
|
|
|
|
import crewai.flow.persistence.factory as factory
|
|
from crewai.flow.persistence.base import FlowPersistence
|
|
from crewai.flow.persistence.decorators import persist
|
|
from crewai.flow.persistence.sqlite import SQLiteFlowPersistence
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def reset_factory():
|
|
"""Reset the factory around each test without clobbering preexisting state."""
|
|
original = factory._factory
|
|
factory.set_flow_persistence_factory(None)
|
|
yield
|
|
factory.set_flow_persistence_factory(original)
|
|
|
|
|
|
def test_default_uses_registered_factory():
|
|
sentinel = SQLiteFlowPersistence()
|
|
factory.set_flow_persistence_factory(lambda: sentinel)
|
|
|
|
assert factory.default_flow_persistence() is sentinel
|
|
|
|
|
|
def test_default_falls_back_to_sqlite():
|
|
assert isinstance(factory.default_flow_persistence(), SQLiteFlowPersistence)
|
|
|
|
|
|
def test_persist_decorator_honors_falsy_persistence():
|
|
# @persist with an explicit but falsy FlowPersistence must keep it, not
|
|
# replace it with the default via a truthiness check.
|
|
class _FalsyPersistence(FlowPersistence):
|
|
def __bool__(self) -> bool:
|
|
return False
|
|
|
|
def init_db(self) -> None:
|
|
pass
|
|
|
|
def save_state(
|
|
self,
|
|
flow_uuid: str,
|
|
method_name: str,
|
|
state_data: dict[str, Any] | BaseModel,
|
|
) -> None:
|
|
pass
|
|
|
|
def load_state(self, flow_uuid: str) -> dict[str, Any] | None:
|
|
return None
|
|
|
|
falsy = _FalsyPersistence()
|
|
|
|
@persist(persistence=falsy)
|
|
class _DummyFlow:
|
|
pass
|
|
|
|
assert _DummyFlow.__flow_persistence_config__.persistence is falsy
|