flock-core 0.5.22__py3-none-any.whl → 0.5.24__py3-none-any.whl

flock/core/agent.py

@@ -226,11 +226,7 @@ class Agent(metaclass=AutoTracedMeta):
             comp_name = self._component_display_name(component)
             priority = getattr(component, "priority", 0)
             logger.info(
-                "Agent %s: utility added: component=%s, priority=%s, total_utilities=%s",
-                self.name,
-                comp_name,
-                priority,
-                len(self.utilities),
+                f"Agent {self.name}: utility added: component={comp_name}, priority={priority}, total_utilities={len(self.utilities)}"
             )
         self.utilities.sort(key=lambda comp: getattr(comp, "priority", 0))
 

flock/core/orchestrator.py

@@ -492,13 +492,17 @@ class Flock(metaclass=AutoTracedMeta):
 
     # Runtime --------------------------------------------------------------
 
-    async def run_until_idle(self) -> None:
+    async def run_until_idle(self, *, wait_for_input: bool = False) -> None:
         """Wait for all scheduled agent tasks to complete.
 
         This method blocks until the blackboard reaches a stable state where no
         agents are queued for execution. Essential for batch processing and ensuring
         all agent cascades complete before continuing.
 
+        Args:
+            wait_for_input: If True, waits for user input before returning (default: False).
+                Useful for debugging or step-by-step execution.
+
         Note:
             Automatically resets circuit breaker counters and shuts down MCP connections
             when idle. Used with publish() for event-driven workflows.
@@ -514,6 +518,12 @@ class Flock(metaclass=AutoTracedMeta):
             >>> await flock.publish_many([task1, task2, task3])
             >>> await flock.run_until_idle()  # All tasks processed in parallel
 
+            >>> # Step-by-step execution with user prompts
+            >>> await flock.publish(task1)
+            >>> await flock.run_until_idle(wait_for_input=True)  # Pauses for user input
+            >>> await flock.publish(task2)
+            >>> await flock.run_until_idle(wait_for_input=True)  # Pauses again
+
         See Also:
             - publish(): Event-driven artifact publishing
             - publish_many(): Batch publishing for parallel execution
@@ -553,6 +563,12 @@ class Flock(metaclass=AutoTracedMeta):
         # Automatically shutdown MCP connections when idle
         await self.shutdown(include_components=False)
 
+        # Wait for user input if requested
+        if wait_for_input:
+            # Use asyncio.to_thread to avoid blocking the event loop
+            # since input() is a blocking I/O operation
+            await asyncio.to_thread(input, "Press any key to continue....")
+
     async def direct_invoke(
         self, agent: Agent, inputs: Sequence[BaseModel | Mapping[str, Any] | Artifact]
     ) -> list[Artifact]:
@@ -715,6 +731,7 @@ class Flock(metaclass=AutoTracedMeta):
         partition_key: str | None = None,
         tags: set[str] | None = None,
         is_dashboard: bool = False,
+        schedule_immediately: bool = True,
     ) -> Artifact:
         """Publish an artifact to the blackboard (event-driven).
 
@@ -727,16 +744,22 @@ class Flock(metaclass=AutoTracedMeta):
             partition_key=partition_key,
             tags=tags,
             is_dashboard=is_dashboard,
+            schedule_immediately=schedule_immediately,
         )
 
     async def publish_many(
-        self, objects: Iterable[BaseModel | dict | Artifact], **kwargs: Any
+        self,
+        objects: Iterable[BaseModel | dict | Artifact],
+        schedule_immediately: bool = True,
+        **kwargs: Any,
     ) -> list[Artifact]:
         """Publish multiple artifacts at once (event-driven).
 
         Delegates to ArtifactManager for batch publishing.
         """
-        return await self._artifact_manager.publish_many(objects, **kwargs)
+        return await self._artifact_manager.publish_many(
+            objects, schedule_immediately=schedule_immediately, **kwargs
+        )
 
     # -----------------------------------------------------------------------------
     # NEW DIRECT INVOCATION API - Explicit Control

flock/orchestrator/artifact_manager.py

@@ -53,6 +53,7 @@ class ArtifactManager:
         partition_key: str | None = None,
         tags: set[str] | None = None,
         is_dashboard: bool = False,
+        schedule_immediately: bool = True,
     ) -> Artifact:
         """Publish an artifact to the blackboard (event-driven).
 
@@ -125,12 +126,18 @@ class ArtifactManager:
                 "Expected BaseModel, dict, or Artifact."
             )
 
-        # Persist and schedule matching agents
-        await self.persist_and_schedule(artifact)
+        if schedule_immediately:
+            await self.persist_and_schedule(artifact)
+        else:
+            await self.persist(artifact)
+
         return artifact
 
     async def publish_many(
-        self, objects: Iterable[BaseModel | dict | Artifact], **kwargs: Any
+        self,
+        objects: Iterable[BaseModel | dict | Artifact],
+        schedule_immediately: bool = True,
+        **kwargs: Any,
     ) -> list[Artifact]:
         """Publish multiple artifacts at once (event-driven).
 
@@ -150,7 +157,9 @@ class ArtifactManager:
         """
         artifacts = []
         for obj in objects:
-            artifact = await self.publish(obj, **kwargs)
+            artifact = await self.publish(
+                obj, schedule_immediately=schedule_immediately, **kwargs
+            )
             artifacts.append(artifact)
         return artifacts
 
@@ -164,5 +173,14 @@ class ArtifactManager:
         self._orchestrator.metrics["artifacts_published"] += 1
         await self._scheduler.schedule_artifact(artifact)
 
+    async def persist(self, artifact: Artifact) -> None:
+        """Persist artifact to store without scheduling.
+
+        Args:
+            artifact: Artifact to publish
+        """
+        await self._store.publish(artifact)
+        self._orchestrator.metrics["artifacts_published"] += 1
+
 
 __all__ = ["ArtifactManager"]

flock/semantic/embedding_service.py

@@ -4,13 +4,13 @@ This module provides a singleton service for generating and caching embeddings
 using sentence-transformers.
 """
 
-import logging
+from flock.logging.logging import get_logger
 from collections import OrderedDict
 
 import numpy as np
 
 
-logger = logging.getLogger(__name__)
+logger = get_logger(__name__)
 
 
 class LRUCache:
@@ -213,6 +213,10 @@ class EmbeddingService:
 
         similarity = dot_product / (norm1 * norm2)
 
+        logger.debug(
+            f"Computed similarity: {similarity:.4f} between texts '{text1}' and '{text2}'",
+        )
+
         # Clamp to [0, 1] and handle floating point errors
         return float(max(0.0, min(1.0, similarity)))
 

{flock_core-0.5.22.dist-info → flock_core-0.5.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.5.22
+Version: 0.5.24
 Summary: Flock: A declrative framework for building and orchestrating AI agents.
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License: MIT
@@ -43,7 +43,7 @@ Description-Content-Type: text/markdown
   <a href="LICENSE" target="_blank"><img alt="License" src="https://img.shields.io/github/license/whiteducksoftware/flock?style=for-the-badge"></a>
   <a href="https://whiteduck.de" target="_blank"><img alt="Built by white duck" src="https://img.shields.io/badge/Built%20by-white%20duck%20GmbH-white?style=for-the-badge&labelColor=black"></a>
   <a href="https://codecov.io/gh/whiteducksoftware/flock" target="_blank"><img alt="Test Coverage" src="https://codecov.io/gh/whiteducksoftware/flock/branch/main/graph/badge.svg?token=YOUR_TOKEN_HERE&style=for-the-badge"></a>
-  <img alt="Tests" src="https://img.shields.io/badge/tests-1300+-brightgreen?style=for-the-badge">
+  <img alt="Tests" src="https://img.shields.io/badge/tests-1400+-brightgreen?style=for-the-badge">
 </p>
 
 ---

{flock_core-0.5.22.dist-info → flock_core-0.5.24.dist-info}/RECORD

@@ -23,10 +23,10 @@ flock/components/orchestrator/circuit_breaker.py,sha256=SH4pNFpe7MnkIRuK5TmRFBBO
 flock/components/orchestrator/collection.py,sha256=oTXDqNZgliwVqZO8e49mX3yEWxflzeZdD7UnbcHWypk,5572
 flock/components/orchestrator/deduplication.py,sha256=w4fwg3nLmb2TXPSSN3GEq5XPwviEEmmwu3fceSejBXw,2464
 flock/core/__init__.py,sha256=pEE4dbPtSEIC1WQYK8USWbF0EmZhccTfDO7eh7lzD-s,574
-flock/core/agent.py,sha256=6Slehv1vWawSdT2oLw7LWm75M1H1PnY7sbnmr5z-OBA,38190
+flock/core/agent.py,sha256=iP6AfNUb_446eu-4NAJsqVuGLyf4H46n-aZUXw1Jqk4,38118
 flock/core/artifacts.py,sha256=dHDjn3DgS0Yo62oFoafQcdYNtqFrqLCzkvUytwm5VGU,2532
 flock/core/context_provider.py,sha256=1RG4lYhDNTshAQA1PwZELtUF2UDdC5NZWcBlE4BKDtc,20260
-flock/core/orchestrator.py,sha256=oebGAPie5TyHDtvVKRWfR9E1UrTtSKl-D7tcCrke_w0,42199
+flock/core/orchestrator.py,sha256=Fgl32VKWzW-xzVhfbEnhJCFiZVJhpNiYR9hHcd12TPE,43214
 flock/core/store.py,sha256=4Zyh73RjjiUTtYTYiZ1pajWVXWKIew0dFERTNgQxRAA,31964
 flock/core/subscription.py,sha256=x2kkbaAbqFqR2woCYhc0a7Xi2k4-EYt8-LCNnTtN-jQ,10768
 flock/core/visibility.py,sha256=uwscg32t6Dp9LuA_EVDT5-_1lotzZWWS9Dl1foyJDxo,2926
@@ -219,7 +219,7 @@ flock/models/__init__.py,sha256=724PD_4bRX0rFdBDXzP0Pwr-OUntq0LIH48bMmtfcM0,266
 flock/models/system_artifacts.py,sha256=f4O-gErJRy5aMFVDh6ZGcj_5Gr130WnG28m5AfhnbxU,1083
 flock/orchestrator/__init__.py,sha256=u4DDXcwCSGrpHYbkdA4eP4W9Uj9vNMF6qxxUtygwmEU,1728
 flock/orchestrator/artifact_collector.py,sha256=XUB4dmx2tzq0p2CkYfP4NowfBkryp3DNA9jb7PWEIDk,6419
-flock/orchestrator/artifact_manager.py,sha256=fXqyU1pRDLf7MRxK86CMHr7uDOeeom_PEZ-3pVZxTj0,5775
+flock/orchestrator/artifact_manager.py,sha256=TR5gZqWehNFkeHxfFUIPU_P7-JnWVjK43sy3THVh-9I,6288
 flock/orchestrator/batch_accumulator.py,sha256=WbIQqQz03LRLPe9yJsf_6TgppofE2tN_7f60P38n2S8,8027
 flock/orchestrator/component_runner.py,sha256=pf068DRGw4unh4K09Y8gt-OLM3FLm6CHQQ3d13jzkmM,14266
 flock/orchestrator/context_builder.py,sha256=Wp7_lzu_F7y-be6uiplur0VGauuP58NSmu1foXFfZc0,6244
@@ -235,7 +235,7 @@ flock/patches/__init__.py,sha256=KfRTpO_rz547iKxJDknymar1P3lS_8AhDh-sR1BkcZ0,194
 flock/patches/dspy_streaming_patch.py,sha256=OQeLYWmL-7rqudphMwERnpvXhXamnvYn3xNcySajacY,2879
 flock/semantic/__init__.py,sha256=ECd5EhdJjnlNA3BJiVlArvvJlVVjUPBlL5aH0Au7P5g,1568
 flock/semantic/context_provider.py,sha256=qkJ2_KHBZk2Mjj3cvtdzdIkvZON79WFLkHUTeyGn7X0,5793
-flock/semantic/embedding_service.py,sha256=R4pDy3sq4phDU2wn71F7gxcVM-HegXVtKZRiw_amu90,7018
+flock/semantic/embedding_service.py,sha256=_oRl9z7srZu55gokLxt7IntsgfL9UcEKNLfpqy3hyQ8,7166
 flock/storage/__init__.py,sha256=9-p-RTpxGJOksfDavbpUsTghSIjDGtWCEC3bpVuYZY0,249
 flock/storage/artifact_aggregator.py,sha256=wMpoXjko9UmfMmQ_qUTzSX-nhb0k8BVRlzRLHlkKEug,5176
 flock/storage/in_memory/__init__.py,sha256=XPYxGNnpDjiv_L1tj8AxfORA_xHP16FV8t5UqrhtFLk,138
@@ -594,8 +594,8 @@ flock/utils/utilities.py,sha256=E3EQDo5SLU4XeoL3-PCc4lOatJHFEkMHKbZhnGhdMG4,1248
 flock/utils/validation.py,sha256=Vqzd8Qi73ttJNSHdr8EUEonyfzAYWVfEyW0fhLs1bA4,1847
 flock/utils/visibility.py,sha256=riJfHFWH8R2vk3DU7PYNi2LMaE8cy8pnMvxxiqWq-4I,2349
 flock/utils/visibility_utils.py,sha256=eGA28aL8FnGyGzsbVNvqI1UoPauu1Jedl5LUZ5oWZZ0,3874
-flock_core-0.5.22.dist-info/METADATA,sha256=Z1NgLhDZlzreyvJzT5HGzmr01NSlGtzC4OiLKx71Uhg,31235
-flock_core-0.5.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flock_core-0.5.22.dist-info/entry_points.txt,sha256=UQdPmtHd97gSA_IdLt9MOd-1rrf_WO-qsQeIiHWVrp4,42
-flock_core-0.5.22.dist-info/licenses/LICENSE,sha256=U3IZuTbC0yLj7huwJdldLBipSOHF4cPf6cUOodFiaBE,1072
-flock_core-0.5.22.dist-info/RECORD,,
+flock_core-0.5.24.dist-info/METADATA,sha256=iyV6ZB20p-jjY0xckuwZnPlmf-bfYOQDlYJkwgrkwNY,31235
+flock_core-0.5.24.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flock_core-0.5.24.dist-info/entry_points.txt,sha256=UQdPmtHd97gSA_IdLt9MOd-1rrf_WO-qsQeIiHWVrp4,42
+flock_core-0.5.24.dist-info/licenses/LICENSE,sha256=U3IZuTbC0yLj7huwJdldLBipSOHF4cPf6cUOodFiaBE,1072
+flock_core-0.5.24.dist-info/RECORD,,