flock-core 0.5.0b2__tar.gz → 0.5.0b5__tar.gz

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.5.0b2
+Version: 0.5.0b5
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/docs/architecture_report.md

@@ -102,7 +102,7 @@ graph TB
     subgraph "FlockAgent (Minimal Core)"
         Agent[FlockAgent]
         ComponentsList[components: list]
-        NextAgent[next_agent: str | None]
+        NextAgent["next_agent: str | None"]
         Config[config: FlockAgentConfig]
     end
     
@@ -206,7 +206,7 @@ graph LR
     subgraph "Key Improvements"
         SRP[Single Responsibility Principle]
         Lazy[Lazy Loading]
-        Consistency[Consistent Naming (_helper)]
+        Consistency["Consistent Naming (_helper)"]
         Maintainability[Easier Maintenance]
     end
     

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/legacy/flock_agent_original.py

@@ -11,7 +11,7 @@ from datetime import datetime
 from typing import TYPE_CHECKING, Any, TypeVar
 
 from flock.core.config.flock_agent_config import FlockAgentConfig
-from flock.core.mcp.flock_mcp_server import FlockMCPServerBase
+from flock.core.mcp.flock_mcp_server import FlockMCPServer
 from flock.core.serialization.json_encoder import FlockJSONEncoder
 from flock.workflow.temporal_config import TemporalActivityConfig
 
@@ -97,7 +97,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
             description="List of callable tools the agent can use. These must be registered.",
         )
     )
-    servers: list[str | FlockMCPServerBase] | None = Field(
+    servers: list[str | FlockMCPServer] | None = Field(
         default=None,
         description="List of MCP Servers the agent can use to enhance its capabilities. These must be registered.",
     )
@@ -151,7 +151,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
         input: SignatureType = None,
         output: SignatureType = None,
         tools: list[Callable[..., Any]] | None = None,
-        servers: list[str | FlockMCPServerBase] | None = None,
+        servers: list[str | FlockMCPServer] | None = None,
         evaluator: "FlockEvaluator | None" = None,
         handoff_router: "FlockRouter | None" = None,
         # Use dict for modules
@@ -364,9 +364,9 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
 
                     FlockRegistry = get_registry()  # Get the registry
                     for server in self.servers:
-                        registered_server: FlockMCPServerBase | None = None
+                        registered_server: FlockMCPServer | None = None
                         server_tools = []
-                        if isinstance(server, FlockMCPServerBase):
+                        if isinstance(server, FlockMCPServer):
                             # check if registered
                             server_name = server.config.name
                             registered_server = FlockRegistry.get_server(
@@ -825,7 +825,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
             )
             serialized_servers = []
             for server in self.servers:
-                if isinstance(server, FlockMCPServerBase):
+                if isinstance(server, FlockMCPServer):
                     serialized_servers.append(server.config.name)
                 else:
                     # Write it down as a list of server names.
@@ -1088,7 +1088,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
                 if isinstance(server_name, str):
                     # Case 1 (default behavior): A server name is passe.
                     agent.servers.append(server_name)
-                elif isinstance(server_name, FlockMCPServerBase):
+                elif isinstance(server_name, FlockMCPServer):
                     # Case 2 (highly unlikely): If someone somehow manages to pass
                     # an instance of a server during the deserialization step (however that might be achieved)
                     # check the registry, if the server is already registered, if not, register it

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/legacy/flock_agent_refactored_backup.py

@@ -20,7 +20,7 @@ from flock.core.context.context import FlockContext
 from flock.core.flock_evaluator import FlockEvaluator, FlockEvaluatorConfig
 from flock.core.flock_module import FlockModule, FlockModuleConfig
 from flock.core.flock_router import FlockRouter, FlockRouterConfig
-from flock.core.mcp.flock_mcp_server import FlockMCPServerBase
+from flock.core.mcp.flock_mcp_server import FlockMCPServer
 
 # Mixins and Serialization components
 from flock.core.mixin.dspy_integration import DSPyIntegrationMixin
@@ -82,7 +82,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
             description="List of callable tools the agent can use. These must be registered.",
         )
     )
-    servers: list[str | FlockMCPServerBase] | None = Field(
+    servers: list[str | FlockMCPServer] | None = Field(
         default=None,
         description="List of MCP Servers the agent can use to enhance its capabilities. These must be registered.",
     )
@@ -136,7 +136,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
         input: SignatureType = None,
         output: SignatureType = None,
         tools: list[Callable[..., Any]] | None = None,
-        servers: list[str | FlockMCPServerBase] | None = None,
+        servers: list[str | FlockMCPServer] | None = None,
         evaluator: FlockEvaluator | None = None,
         handoff_router: FlockRouter | None = None,
         # Use dict for modules

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/legacy/flock_agent_unified.py

@@ -12,15 +12,15 @@ from flock.core.agent.flock_agent_execution import FlockAgentExecution
 from flock.core.agent.flock_agent_integration import FlockAgentIntegration
 from flock.core.agent.flock_agent_serialization import FlockAgentSerialization
 from flock.core.component.agent_component_base import AgentComponent
-from flock.core.component.evaluation_component_base import (
-    EvaluationComponentBase,
+from flock.core.component.evaluation_component import (
+    EvaluationComponent,
 )
-from flock.core.component.routing_component_base import RoutingModuleBase
+from flock.core.component.routing_component import RoutingModuleBase
 from flock.core.config.flock_agent_config import FlockAgentConfig
 from flock.core.context.context import FlockContext
 from flock.core.flock_router import HandOffRequest
 from flock.core.logging.logging import get_logger
-from flock.core.mcp.flock_mcp_server import FlockMCPServerBase
+from flock.core.mcp.flock_mcp_server import FlockMCPServer
 
 # Mixins and Serialization components
 from flock.core.mixin.dspy_integration import DSPyIntegrationMixin
@@ -80,7 +80,7 @@ class FlockAgentUnified(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
         default=None,
         description="List of callable tools the agent can use. These must be registered.",
     )
-    servers: list[str | FlockMCPServerBase] | None = Field(
+    servers: list[str | FlockMCPServer] | None = Field(
         default=None,
         description="List of MCP Servers the agent can use to enhance its capabilities.",
     )
@@ -132,7 +132,7 @@ class FlockAgentUnified(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
         input: SignatureType = None,
         output: SignatureType = None,
         tools: list[Callable[..., Any]] | None = None,
-        servers: list[str | FlockMCPServerBase] | None = None,
+        servers: list[str | FlockMCPServer] | None = None,
         components: list[AgentComponent] | None = None,
         write_to_file: bool = False,
         wait_for_input: bool = False,
@@ -163,10 +163,10 @@ class FlockAgentUnified(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
     # These provide familiar access patterns while using the unified model
 
     @property
-    def evaluator(self) -> EvaluationComponentBase | None:
+    def evaluator(self) -> EvaluationComponent | None:
         """Get the primary evaluation component for this agent."""
         return next(
-            (c for c in self.components if isinstance(c, EvaluationComponentBase)),
+            (c for c in self.components if isinstance(c, EvaluationComponent)),
             None
         )
 

flock_core-0.5.0b2/src/flock/core/flock_registry.py.backup → flock_core-0.5.0b5/legacy/flock_registry.py

@@ -26,11 +26,11 @@ from typing import (  # Add TYPE_CHECKING
 from pydantic import BaseModel
 
 if TYPE_CHECKING:
+    from flock.core.component.agent_component_base import AgentComponent
     from flock.core.flock_agent import (
         FlockAgent,  # Import only for type checking
     )
-    from flock.core.mcp.flock_mcp_server import FlockMCPServerBase
-    from flock.core.component.agent_component_base import AgentComponent
+    from flock.core.mcp.flock_mcp_server import FlockMCPServer
 
     COMPONENT_BASE_TYPES = (AgentComponent,)
 
@@ -62,7 +62,7 @@ class FlockRegistry:
     _instance = None
 
     _agents: dict[str, FlockAgent]
-    _servers: dict[str, FlockMCPServerBase]
+    _servers: dict[str, FlockMCPServer]
     _callables: dict[str, Callable]
     _types: dict[str, type]
     _components: dict[str, type]  # For Module, Evaluator, Router classes
@@ -159,7 +159,7 @@ class FlockRegistry:
             return None
 
     # --- Server Registration ---
-    def register_server(self, server: FlockMCPServerBase) -> None:
+    def register_server(self, server: FlockMCPServer) -> None:
         """Registers a flock mcp server by its name."""
         if not hasattr(server.config, "name") or not server.config.name:
             logger.error(
@@ -176,7 +176,7 @@ class FlockRegistry:
         self._servers[server.config.name] = server
         logger.debug(f"Registered server: {server.config.name}")
 
-    def get_server(self, name: str) -> FlockMCPServerBase | None:
+    def get_server(self, name: str) -> FlockMCPServer | None:
         """Retrieves a registered FlockMCPServer instance by name."""
         server = self._servers.get(name)
         if not server:

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/legacy/modules/performance/metrics_module.py

@@ -15,7 +15,7 @@ from flock.core.context.context import FlockContext
 from flock.core.flock_agent import FlockAgent
 from flock.core.flock_module import FlockModule, FlockModuleConfig
 from flock.core.flock_registry import flock_component
-from flock.core.mcp.flock_mcp_server import FlockMCPServerBase
+from flock.core.mcp.flock_mcp_server import FlockMCPServer
 
 
 class MetricPoint(BaseModel):
@@ -525,7 +525,7 @@ class MetricsModule(FlockModule):
 
     # --- MCP Server Lifecycle Hooks ---
     async def on_server_error(
-        self, server: FlockMCPServerBase, error: Exception
+        self, server: FlockMCPServer, error: Exception
     ) -> None:
         """Record server error metrics."""
         self._record_metric(
@@ -537,7 +537,7 @@ class MetricsModule(FlockModule):
             },
         )
 
-    async def on_pre_server_init(self, server: FlockMCPServerBase):
+    async def on_pre_server_init(self, server: FlockMCPServer):
         """Initialize metrics collection for server."""
         self._server_start_time = time.time()
 
@@ -549,7 +549,7 @@ class MetricsModule(FlockModule):
                 {"server": server.config.name, "phase": "pre_init"},
             )
 
-    async def on_post_server_init(self, server: FlockMCPServerBase):
+    async def on_post_server_init(self, server: FlockMCPServer):
         """Collect metrics after server starts."""
         if self.config.collect_memory:
             checkpoint_memory = psutil.Process().memory_info().rss
@@ -559,7 +559,7 @@ class MetricsModule(FlockModule):
                 {"server": server.config.name, "phase": "post_init"},
             )
 
-    async def on_pre_server_terminate(self, server: FlockMCPServerBase):
+    async def on_pre_server_terminate(self, server: FlockMCPServer):
         """Collect metrics before server terminates."""
         if self.config.collect_memory:
             checkpoint_memory = psutil.Process().memory_info().rss
@@ -569,7 +569,7 @@ class MetricsModule(FlockModule):
                 {"server": server.config.name, "phase": "pre_terminate"},
             )
 
-    async def on_post_server_terminate(self, server: FlockMCPServerBase):
+    async def on_post_server_terminate(self, server: FlockMCPServer):
         """Collect metrics after server terminates.
 
         Clean up and final metric recording.
@@ -600,7 +600,7 @@ class MetricsModule(FlockModule):
                 json.dump(summary, f, indent=2)
 
     async def on_pre_mcp_call(
-        self, server: FlockMCPServerBase, arguments: Any | None = None
+        self, server: FlockMCPServer, arguments: Any | None = None
     ):
         """Record pre-call metrics."""
         if self.config.collect_cpu:
@@ -630,7 +630,7 @@ class MetricsModule(FlockModule):
             )
 
     async def on_post_mcp_call(
-        self, server: FlockMCPServerBase, result: Any | None = None
+        self, server: FlockMCPServer, result: Any | None = None
     ):
         """Record post-call metrics."""
         if self.config.collect_timing and self._server_start_time:
@@ -661,7 +661,7 @@ class MetricsModule(FlockModule):
             )
 
     async def on_connect(
-        self, server: FlockMCPServerBase, additional_params: dict[str, Any]
+        self, server: FlockMCPServer, additional_params: dict[str, Any]
     ) -> dict[str, Any]:
         """Collect metrics during connect."""
         # We should track the refresh rate for clients

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "flock-core"
-version = "0.5.0b2"
+version = "0.5.0b5"
 description = "Declarative LLM Orchestration at Scale"
 readme = "README.md"
 authors = [

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/requirements.txt

@@ -1,34 +1,32 @@
 # This file was autogenerated by uv via the following command:
 #    uv pip compile --no-deps -o requirements.txt pyproject.toml
-azure-search-documents==11.5.2
+aiosqlite==0.21.0
     # via flock-core (pyproject.toml)
-chromadb==0.6.3
+azure-data-tables==12.7.0
     # via flock-core (pyproject.toml)
 cloudpickle==3.1.1
     # via flock-core (pyproject.toml)
-datasets==3.5.0
+croniter==6.0.0
     # via flock-core (pyproject.toml)
 devtools==0.12.2
     # via flock-core (pyproject.toml)
-dspy==2.6.16
-    # via flock-core (pyproject.toml)
-duckduckgo-search==7.3.2
+dspy==2.6.23
     # via flock-core (pyproject.toml)
 fastapi==0.115.12
     # via flock-core (pyproject.toml)
 httpx==0.28.1
     # via flock-core (pyproject.toml)
-litellm==1.63.7
+litellm==1.69.3
     # via flock-core (pyproject.toml)
 loguru==0.7.3
     # via flock-core (pyproject.toml)
-matplotlib==3.10.1
+markdown2==2.5.3
     # via flock-core (pyproject.toml)
-msgpack==1.1.0
+mcp==1.9.2
     # via flock-core (pyproject.toml)
-nltk==3.9.1
+msgpack==1.1.0
     # via flock-core (pyproject.toml)
-notion-client==2.3.0
+openai==1.75.0
     # via flock-core (pyproject.toml)
 opentelemetry-api==1.30.0
     # via flock-core (pyproject.toml)
@@ -42,6 +40,8 @@ opentelemetry-instrumentation-logging==0.51b0
     # via flock-core (pyproject.toml)
 opentelemetry-sdk==1.30.0
     # via flock-core (pyproject.toml)
+opik==1.7.28
+    # via flock-core (pyproject.toml)
 pandas==2.2.3
     # via flock-core (pyproject.toml)
 pillow==11.1.0
@@ -50,7 +50,7 @@ prometheus-client==0.21.1
     # via flock-core (pyproject.toml)
 psutil==7.0.0
     # via flock-core (pyproject.toml)
-pydantic==2.10.6
+pydantic==2.10.5
     # via flock-core (pyproject.toml)
 pydantic-settings==2.8.1
     # via flock-core (pyproject.toml)
@@ -60,18 +60,16 @@ python-decouple==3.8
     # via flock-core (pyproject.toml)
 python-dotenv==1.1.0
     # via flock-core (pyproject.toml)
-python-fasthtml==0.12.12
-    # via flock-core (pyproject.toml)
 pyyaml==6.0.2
     # via flock-core (pyproject.toml)
 questionary==2.1.0
     # via flock-core (pyproject.toml)
 rich==13.9.4
     # via flock-core (pyproject.toml)
-sentence-transformers==4.0.1
-    # via flock-core (pyproject.toml)
 temporalio==1.9.0
     # via flock-core (pyproject.toml)
+thefuzz==0.22.1
+    # via flock-core (pyproject.toml)
 tiktoken==0.9.0
     # via flock-core (pyproject.toml)
 toml==0.10.2
@@ -80,5 +78,7 @@ tqdm==4.67.1
     # via flock-core (pyproject.toml)
 uvicorn==0.34.0
     # via flock-core (pyproject.toml)
-zep-python==2.0.2
+wd-di==0.2.14
+    # via flock-core (pyproject.toml)
+websockets==15.0.1
     # via flock-core (pyproject.toml)

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/src/flock/cli/execute_flock.py

@@ -142,7 +142,7 @@ def execute_flock(flock: Flock):
 
         # Run the Flock
         result = flock.run(
-            start_agent=start_agent_name,
+            agent=start_agent_name,
             input=input_data,
         )
 

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/src/flock/cli/manage_agents.py

@@ -105,7 +105,7 @@ def _list_agents(flock: Flock):
         model = agent.model or flock.model or "Default"
 
         # Format description (truncate if too long)
-        description = agent.resolved_description
+        description = agent.description
         if description and len(description) > 30:
             description = description[:27] + "..."
         elif not description:
@@ -165,7 +165,7 @@ def _view_agent_details(agent: FlockAgent):
 
     basic_info.add_row("Name", agent.name)
     basic_info.add_row("Model", str(agent.model or "Default"))
-    basic_info.add_row("Description", agent.resolved_description if agent.resolved_description else "N/A")
+    basic_info.add_row("Description", agent.description if agent.description else "N/A")
     basic_info.add_row("Input", str(agent.input))
     basic_info.add_row("Output", str(agent.output))
     basic_info.add_row("Write to File", str(agent.config.write_to_file))
@@ -339,7 +339,7 @@ def _edit_agent(flock: Flock):
     console.print(f"\n[bold underline]Details for Agent: {agent.name}[/]")
     basic_info = Table(show_header=False, box=Box.ROUNDED, padding=(0, 2))
     basic_info.add_row("Name", agent.name)
-    description = agent.resolved_description
+    description = agent.description
     basic_info.add_row("Description", description if description else "N/A")
     basic_info.add_row("Model", agent.model or "Flock Default")
     basic_info.add_row("Input Signature", str(agent.input))

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/src/flock/components/evaluation/declarative_evaluation_component.py

@@ -12,12 +12,12 @@ with workflow.unsafe.imports_passed_through():
 from pydantic import Field, PrivateAttr
 
 from flock.core.component.agent_component_base import AgentComponentConfig
-from flock.core.component.evaluation_component_base import EvaluationComponentBase
+from flock.core.component.evaluation_component import EvaluationComponent
 from flock.core.context.context import FlockContext
-from flock.core.registry import flock_component
 from flock.core.logging.logging import get_logger
 from flock.core.mixin.dspy_integration import DSPyIntegrationMixin
 from flock.core.mixin.prompt_parser import PromptParserMixin
+from flock.core.registry import flock_component
 
 logger = get_logger("components.evaluation.declarative")
 
@@ -45,7 +45,7 @@ class DeclarativeEvaluationConfig(AgentComponentConfig):
 
 @flock_component(config_class=DeclarativeEvaluationConfig)
 class DeclarativeEvaluationComponent(
-    EvaluationComponentBase, DSPyIntegrationMixin, PromptParserMixin
+    EvaluationComponent, DSPyIntegrationMixin, PromptParserMixin
 ):
     """Evaluation component that uses DSPy for generation.
     
@@ -58,7 +58,7 @@ class DeclarativeEvaluationComponent(
         default_factory=DeclarativeEvaluationConfig,
         description="Evaluation configuration",
     )
-    
+
     _cost: float = PrivateAttr(default=0.0)
     _lm_history: list = PrivateAttr(default_factory=list)
 
@@ -75,7 +75,7 @@ class DeclarativeEvaluationComponent(
     ) -> dict[str, Any]:
         """Core evaluation logic using DSPy - migrated from DeclarativeEvaluator."""
         logger.debug(f"Starting declarative evaluation for component '{self.name}'")
-        
+
         # Setup DSPy context with LM (directly from original implementation)
         with dspy.context(
             lm=dspy.LM(
@@ -89,14 +89,14 @@ class DeclarativeEvaluationComponent(
             try:
                 from rich.console import Console
                 console = Console()
-                
+
                 # Create DSPy signature from agent definition
                 _dspy_signature = self.create_dspy_signature_class(
                     agent.name,
                     agent.description,
                     f"{agent.input} -> {agent.output}",
                 )
-                
+
                 # Get output field names for streaming
                 output_field_names = list(_dspy_signature.output_fields.keys())
                 if not output_field_names:
@@ -113,7 +113,7 @@ class DeclarativeEvaluationComponent(
                     mcp_tools=mcp_tools or [],
                     kwargs=self.config.kwargs,
                 )
-                
+
             except Exception as setup_error:
                 logger.error(
                     f"Error setting up DSPy task for agent '{agent.name}': {setup_error}",
@@ -132,7 +132,7 @@ class DeclarativeEvaluationComponent(
     async def _execute_streaming(self, agent_task, inputs: dict[str, Any], agent: Any, console) -> dict[str, Any]:
         """Execute DSPy program in streaming mode (from original implementation)."""
         logger.info(f"Evaluating agent '{agent.name}' with async streaming.")
-        
+
         if not callable(agent_task):
             logger.error("agent_task is not callable, cannot stream.")
             raise TypeError("DSPy task could not be created or is not callable.")
@@ -167,7 +167,7 @@ class DeclarativeEvaluationComponent(
     async def _execute_standard(self, agent_task, inputs: dict[str, Any], agent: Any) -> dict[str, Any]:
         """Execute DSPy program in standard mode (from original implementation)."""
         logger.info(f"Evaluating agent '{agent.name}' without streaming.")
-        
+
         try:
             # Ensure the call is awaited if the underlying task is async
             result_obj = await agent_task.acall(**inputs)

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/src/flock/components/routing/conditional_routing_component.py

@@ -8,11 +8,12 @@ from typing import TYPE_CHECKING, Any, Literal
 from pydantic import Field, model_validator
 
 from flock.core.component.agent_component_base import AgentComponentConfig
-from flock.core.component.routing_component_base import RoutingComponentBase
+from flock.core.component.routing_component import RoutingComponent
 from flock.core.context.context import FlockContext
-from flock.core.registry import flock_component, get_registry
+
 # HandOffRequest removed - using agent.next_agent directly
 from flock.core.logging.logging import get_logger
+from flock.core.registry import flock_component, get_registry
 
 if TYPE_CHECKING:
     from flock.core.flock_agent import FlockAgent
@@ -148,7 +149,7 @@ class ConditionalRoutingConfig(AgentComponentConfig):
 
 
 @flock_component(config_class=ConditionalRoutingConfig)
-class ConditionalRoutingComponent(RoutingComponentBase):
+class ConditionalRoutingComponent(RoutingComponent):
     """Routes workflow based on evaluating a condition against a value in the FlockContext.
     
     Supports various built-in checks (string, number, list, type, bool, existence)
@@ -436,7 +437,7 @@ class ConditionalRoutingComponent(RoutingComponentBase):
 
             next_agent = cfg.success_agent or None  # Stop chain if None
             logger.debug(f"Success route target: '{next_agent}'")
-            
+
             agent.next_agent = next_agent  # Set directly on agent
 
         else:
@@ -481,7 +482,7 @@ class ConditionalRoutingComponent(RoutingComponentBase):
                     logger.debug(
                         f"Failure route target (after retries): '{next_agent}'"
                     )
-                    
+
                     agent.next_agent = next_agent
             else:
                 # --- No Retry Logic ---
@@ -489,5 +490,5 @@ class ConditionalRoutingComponent(RoutingComponentBase):
                     cfg.failure_agent or None
                 )  # Use failure agent or stop
                 logger.debug(f"Failure route target (no retry): '{next_agent}'")
-                
+
                 agent.next_agent = next_agent

{flock_core-0.5.0b2 → flock_core-0.5.0b5}/src/flock/components/routing/default_routing_component.py

@@ -7,10 +7,10 @@ from typing import TYPE_CHECKING, Any
 from pydantic import Field
 
 from flock.core.component.agent_component_base import AgentComponentConfig
-from flock.core.component.routing_component_base import RoutingComponentBase
+from flock.core.component.routing_component import RoutingComponent
 from flock.core.context.context import FlockContext
-from flock.core.registry import flock_component
 from flock.core.logging.logging import get_logger
+from flock.core.registry import flock_component
 
 if TYPE_CHECKING:
     from flock.core.flock_agent import FlockAgent
@@ -27,7 +27,7 @@ class DefaultRoutingConfig(AgentComponentConfig):
 
 
 @flock_component(config_class=DefaultRoutingConfig)
-class DefaultRoutingComponent(RoutingComponentBase):
+class DefaultRoutingComponent(RoutingComponent):
     """Default routing component implementation.
 
     This router simply uses the configured hand_off property to determine the next agent.