flock-core 0.5.0b17__py3-none-any.whl → 0.5.0b19__py3-none-any.whl

flock/components/utility/metrics_utility_component.py

@@ -15,9 +15,13 @@ from pydantic import BaseModel, Field, field_validator
 from flock.core.component.agent_component_base import AgentComponentConfig
 from flock.core.component.utility_component import UtilityComponent
 from flock.core.context.context import FlockContext
+from flock.core.logging.logging import get_logger
 from flock.core.mcp.flock_mcp_server import FlockMCPServer
 from flock.core.registry import flock_component
 
+logger = get_logger(__name__)
+
+
 if TYPE_CHECKING:
     from flock.core.flock_agent import FlockAgent
 
@@ -181,7 +185,7 @@ class MetricsUtilityComponent(UtilityComponent):
 
             return dict(metrics)
         except Exception as e:
-            print(f"Error loading metrics from files: {e}")
+            logger.error(f"Error loading metrics from files: {e}")
             return {}
 
     def get_metrics(
@@ -419,11 +423,12 @@ class MetricsUtilityComponent(UtilityComponent):
         if self.config.collect_timing and self._start_time:
             latency = time.time() - self._start_time
             self._record_metric("latency", latency, {"agent": agent.name})
+            print(f"Latency: {latency * 1000:.2f}ms")
 
             # Check for alerts
             if self._should_alert("latency", latency):
                 # In practice, you'd want to integrate with a proper alerting system
-                print(f"ALERT: High latency detected: {latency * 1000:.2f}ms")
+                logger.warning(f"ALERT: High latency detected!")
 
         if self.config.collect_token_usage and result:
             # Calculate output tokens and cost
@@ -440,6 +445,8 @@ class MetricsUtilityComponent(UtilityComponent):
                 else:
                     total_output_cost += cost[1]
 
+            print(f"Total output tokens: {total_output_tokens}")
+
             self._record_metric(
                 "tokens",
                 total_output_tokens,

flock/core/flock_agent.py

@@ -137,6 +137,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
         output: DynamicStr | None = None,
         tools: list[Callable[..., Any]] | None = None,
         servers: list[str | FlockMCPServer] | None = None,
+        tool_whitelist: list[str] | None = None,
         components: list[AgentComponent] | None = None,
         config: FlockAgentConfig | None = None,
         next_agent: DynamicStr | None = None,
@@ -153,6 +154,7 @@ class FlockAgent(BaseModel, Serializable, DSPyIntegrationMixin, ABC):
             output=output,
             tools=tools,
             servers=servers,
+            tool_whitelist=tool_whitelist,
             components=components if components is not None else [],
             config=config,
             temporal_activity_config=temporal_activity_config,

{flock_core-0.5.0b17.dist-info → flock_core-0.5.0b19.dist-info}/METADATA

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

{flock_core-0.5.0b17.dist-info → flock_core-0.5.0b19.dist-info}/RECORD

@@ -34,11 +34,11 @@ flock/components/routing/default_routing_component.py,sha256=ZHt2Kjf-GHB5n7evU5N
 flock/components/routing/llm_routing_component.py,sha256=SAaOFjlnhnenM6QEBn3WIpjjNXO-tFpP44TS73zvqzQ,7502
 flock/components/utility/__init__.py,sha256=JRj932upddjzZMWs1avOupEFr_GZNu21ac66Rhw_XgY,532
 flock/components/utility/memory_utility_component.py,sha256=26Io61bbCGjD8UQ4BltMA5RLkMXp8tQoQmddXbQSrzA,20183
-flock/components/utility/metrics_utility_component.py,sha256=u3Bys0dP7FmTeyZOi4XdMhZHCRYc5miXXJ690-qS1Us,24440
+flock/components/utility/metrics_utility_component.py,sha256=Mck_sFCkfXvNpoSgW2N_WOLnjxazzx8jh79tIx5zJhw,24635
 flock/components/utility/output_utility_component.py,sha256=c4K_PL3bGqdyy_v6dnOrmTqV-MkWKAB2w0HS8kzg82k,7613
 flock/core/__init__.py,sha256=OkjsVjRkAB-I6ibeTKVikZ3MxLIcTIzWKphHTbzbr7s,3231
 flock/core/flock.py,sha256=wRycQlGeaq-Vd75mFpPe02qyWTOEyXthT873iBhA3TI,23388
-flock/core/flock_agent.py,sha256=27dDaedYuaSAU85u9Qoaaaj0V4PX24FYrxjTIr63yjA,12776
+flock/core/flock_agent.py,sha256=4Vdhyk-rdsPEuN3xYBsLBBsfpklad6bNj_it9r6XIDc,12868
 flock/core/flock_factory.py,sha256=Z6GJpYXN9_DXuOqvBH9ir0SMoUw78DkWhrhkm90luAQ,20910
 flock/core/flock_scheduler.py,sha256=ng_s7gyijmc-AmYvBn5rtg61CSUZiIkXPRSlA1xO6VQ,8766
 flock/core/flock_server_manager.py,sha256=tM_nOs37vAbEvxmhwy_DL2JPvgFViWroNxrRSu5MfUQ,4523
@@ -551,8 +551,8 @@ flock/workflow/agent_execution_activity.py,sha256=0exwmeWKYXXxdUqDf4YaUVpn0zl06S
 flock/workflow/flock_workflow.py,sha256=sKFsRIL_bDGonXSNhK1zwu6UechghC_PihJJMidI-VI,9139
 flock/workflow/temporal_config.py,sha256=3_8O7SDEjMsSMXsWJBfnb6XTp0TFaz39uyzSlMTSF_I,3988
 flock/workflow/temporal_setup.py,sha256=KR6MlWOrpMtv8NyhaIPAsfl4tjobt81OBByQvg8Kw-Y,1948
-flock_core-0.5.0b17.dist-info/METADATA,sha256=5ccnb7IKfldm855CT8LnKXxZj0pW0sG28oGdeZncTOA,9997
-flock_core-0.5.0b17.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flock_core-0.5.0b17.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
-flock_core-0.5.0b17.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
-flock_core-0.5.0b17.dist-info/RECORD,,
+flock_core-0.5.0b19.dist-info/METADATA,sha256=IIENs1thHIvHmhkyhR3G62jimiMS7_ImCmVLv3AKAlQ,9997
+flock_core-0.5.0b19.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flock_core-0.5.0b19.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
+flock_core-0.5.0b19.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
+flock_core-0.5.0b19.dist-info/RECORD,,