flock-core 0.3.23__py3-none-any.whl → 0.3.31__py3-none-any.whl

{flock_core-0.3.23.dist-info → flock_core-0.3.31.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.3.23
+Version: 0.3.31
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE
@@ -12,10 +12,11 @@ Requires-Dist: azure-search-documents>=11.5.2
 Requires-Dist: chromadb>=0.6.3
 Requires-Dist: cloudpickle>=3.1.1
 Requires-Dist: devtools>=0.12.2
-Requires-Dist: dspy==2.5.42
+Requires-Dist: dspy==2.6.16
 Requires-Dist: duckduckgo-search>=7.3.2
 Requires-Dist: fastapi>=0.115.8
 Requires-Dist: httpx>=0.28.1
+Requires-Dist: litellm==1.63.7
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: matplotlib>=3.10.0
 Requires-Dist: msgpack>=1.1.0
@@ -33,6 +34,8 @@ Requires-Dist: pydantic>=2.10.5
 Requires-Dist: python-box>=7.3.2
 Requires-Dist: python-decouple>=3.8
 Requires-Dist: python-dotenv>=1.0.1
+Requires-Dist: python-fasthtml>=0.12.6
+Requires-Dist: pyyaml>=6.0
 Requires-Dist: questionary>=2.1.0
 Requires-Dist: rich>=13.9.4
 Requires-Dist: sentence-transformers>=3.4.1
@@ -59,7 +62,7 @@ Description-Content-Type: text/markdown
 
 ## Overview
 
-Flock is a framework for orchestrating LLM-powered agents. It leverages a **declarative approach** where you simply specify what each agent needs as input and what it produces as output—without having to write lengthy, brittle prompts. Under the hood, Flock transforms these declarations into robust workflows, using cutting-edge components such as Temporal and DSPy to handle fault tolerance, state management, and error recovery.
+Flock is a framework for orchestrating LLM-powered agents. It leverages a **declarative approach** where you simply specify what each agent needs as input and what it produces as output, without having to write lengthy, brittle prompts. Under the hood, Flock transforms these declarations into robust workflows, using cutting-edge components such as Temporal and DSPy to handle fault tolerance, state management, and error recovery.
 
 
 

{flock_core-0.3.23.dist-info → flock_core-0.3.31.dist-info}/RECORD

@@ -1,22 +1,35 @@
-flock/__init__.py,sha256=49_Z1uLb-fA2InJtNHQleB3NDpfamZGq2S01yttJoVI,2122
+flock/__init__.py,sha256=TCFm-Ax0MINIW3-4KdaiGz36raTENp_AY68eO8_a_rQ,2574
 flock/config.py,sha256=O5QJGlStf4DWSK4ovZsKw01ud4YK3_ij6Ay8sWU8ih0,1522
-flock/cli/constants.py,sha256=UWshD25IGos5Ks4P9paGPalY8BXy9YynzHk6HB4xfOo,669
+flock/cli/constants.py,sha256=s2Mird8E0QXqPH1xG8gxG7rXYY0F4eftGfbBC5ceG2k,568
 flock/cli/create_agent.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
-flock/cli/create_flock.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
+flock/cli/create_flock.py,sha256=Lry15fAwvnh0rHl1pIJ1xKqVj-nYBteNAE1G83LRaAo,6123
+flock/cli/execute_flock.py,sha256=rAP-5nFfyOivi5uWG8mloZwXF9Tj9iD8MYSbllPllQ4,5726
 flock/cli/load_agent.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
 flock/cli/load_examples.py,sha256=DkeLUlrb7rGx3nZ04aADU9HXXu5mZTf_DBwT0xhzIv4,7
-flock/cli/load_flock.py,sha256=3JdECvt5X7uyOG2vZS3-Zk5C5SI_84_QZjcsB3oJmfA,932
+flock/cli/load_flock.py,sha256=5bsdjO5olLtpnFEl_UfisnwR8iSQAkOIWI49tnSwJHc,1537
 flock/cli/load_release_notes.py,sha256=qFcgUrMddAE_TP6x1P-6ZywTUjTknfhTDW5LTxtg1yk,599
+flock/cli/loaded_flock_cli.py,sha256=LXjvTjxt84VG67wBfownICGTbzx0Z2JmCOKAg_L2a6Q,5913
+flock/cli/manage_agents.py,sha256=wkNF0IqNFfePrFueR57SILPW885IPqs3U8Cp-fcPPmo,12710
 flock/cli/settings.py,sha256=Z_TXBzCYlCmSaKrJ_CQCdYy-Cj29gpI4kbC_2KzoKqg,27025
+flock/cli/view_results.py,sha256=dOzK0O1FHSIDERnx48y-2Xke9BkOHS7pcOhs64AyIg0,781
+flock/cli/yaml_editor.py,sha256=VpaSwC-Xcbu_gk4HgUeIL7PXNFu1CdstuJ3mRZOkmIk,8096
 flock/cli/assets/release_notes.md,sha256=bqnk50jxM3w5uY44Dc7MkdT8XmRREFxrVBAG9XCOSSU,4896
-flock/core/__init__.py,sha256=mPlvKc0SxC2qCvSlgYeP_7EyV8ptmdn24NO8mlQoCSo,559
-flock/core/flock.py,sha256=IURlcuNvdsnqKkvgXtX4v_pGWQ8Lfb60X--MT0zvxHo,19881
-flock/core/flock_agent.py,sha256=F1wl54ivLQijSHlv_Q_znw_ElzdDlqpi-KugVUyHTRE,12238
-flock/core/flock_api.py,sha256=2rHnmEdtT5KPZYwGesRT7LqwbrgKClODHT-O56u7pcQ,7140
+flock/core/__init__.py,sha256=6NmSXsdQOoOPWjWqdF8BYiUveb54CjH8Ta0WAjNM0Ps,574
+flock/core/flock.py,sha256=iUHelhSO85qQclHbiJfQzLRepwUFcf3DGBDnrOwgzsM,23722
+flock/core/flock_agent.py,sha256=jkDwB7yVGYyLfg-WuwvR2X4XblXOs5DrB2dB30xBL7Q,24754
 flock/core/flock_evaluator.py,sha256=dOXZeDOGZcAmJ9ahqq_2bdGUU1VOXY4skmwTVpAjiVw,1685
 flock/core/flock_factory.py,sha256=MGTkJCP1WGpV614f87r1vwe0tqAvBCoH9PlqtqDyJDk,2828
 flock/core/flock_module.py,sha256=96aFVYAgwpKN53xGbivQDUpikOYGFCxK5mqhclOcxY0,3003
+flock/core/flock_registry.py,sha256=vQ0N8WSBwOaMLicfmCuFmmOn5ORdF1WhDAJC1BqcHkA,19438
 flock/core/flock_router.py,sha256=A5GaxcGvtiFlRLHBTW7okh5RDm3BdKam2uXvRHRaj7k,2187
+flock/core/api/__init__.py,sha256=OKlhzDWZJfA6ddBwxQUmATY0TSzESsH032u00iVGvdA,228
+flock/core/api/endpoints.py,sha256=m-QAyizCHGh3sd5IXaDEhquSxNgTZLpUVTXLdVjiDVw,9086
+flock/core/api/main.py,sha256=-MJClTcmkyUnaP9fmVu2CcHagFy_9lKk6CAbF05kraY,8928
+flock/core/api/models.py,sha256=mMPCMCD52Txc56_yscwZQG0RRPCJTnIrVKQ8WbI7gts,1111
+flock/core/api/run_store.py,sha256=kmrb0bq2Et5JiSxFWskAAf5a4jTeEFVZShTDlX5jfAk,2773
+flock/core/api/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+flock/core/api/ui/routes.py,sha256=nS-wWO94mshE5ozWfOQZ-HOvtes_1qxDVcqpMZtU5JQ,8885
+flock/core/api/ui/utils.py,sha256=V7PqYHNK519hFJ8jvvwf7bGpbBXCRz_HQG3BDCCqlNA,4802
 flock/core/context/context.py,sha256=8bjRLZ74oacRNBaHmDNXdQKfB-95poF7Pp03n2k0zcQ,6437
 flock/core/context/context_manager.py,sha256=FANSWa6DEhdhtZ7t_9Gza0v80UdpoDOhHbfVOccmjkA,1181
 flock/core/context/context_vars.py,sha256=zYTMi9b6mNSSEHowEQUOTpEDurmAjaUcyBCgfKY6-cU,300
@@ -24,7 +37,7 @@ flock/core/execution/local_executor.py,sha256=rnIQvaJOs6zZORUcR3vvyS6LPREDJTjayg
 flock/core/execution/temporal_executor.py,sha256=OF_uXgQsoUGp6U1ZkcuaidAEKyH7XDtbfrtdF10XQ_4,1675
 flock/core/interpreter/python_interpreter.py,sha256=RaUMZuufsKBNQ4FAeSaOgUuxzs8VYu5TgUUs-xwaxxM,26376
 flock/core/logging/__init__.py,sha256=Q8hp9-1ilPIUIV0jLgJ3_cP7COrea32cVwL7dicPnlM,82
-flock/core/logging/logging.py,sha256=uWtUZnQI_sdtzLfXo61JPENSNNwplvEuGMPIhsYd-gQ,12448
+flock/core/logging/logging.py,sha256=-ViLNLTm5wqbvVsc6xYekOsAY38ZCIn6j1wciC7lgeM,15417
 flock/core/logging/telemetry.py,sha256=3E9Tyj6AUR6A5RlIufcdCdWm5BAA7tbOsCa7lHoUQaU,5404
 flock/core/logging/trace_and_logged.py,sha256=5vNrK1kxuPMoPJ0-QjQg-EDJL1oiEzvU6UNi6X8FiMs,2117
 flock/core/logging/formatters/enum_builder.py,sha256=LgEYXUv84wK5vwHflZ5h8HBGgvLH3sByvUQe8tZiyY0,981
@@ -35,13 +48,14 @@ flock/core/logging/span_middleware/baggage_span_processor.py,sha256=gJfRl8FeB6jd
 flock/core/logging/telemetry_exporter/base_exporter.py,sha256=rQJJzS6q9n2aojoSqwCnl7ZtHrh5LZZ-gkxUuI5WfrQ,1124
 flock/core/logging/telemetry_exporter/file_exporter.py,sha256=nKAjJSZtA7FqHSTuTiFtYYepaxOq7l1rDvs8U8rSBlA,3023
 flock/core/logging/telemetry_exporter/sqlite_exporter.py,sha256=CDsiMb9QcqeXelZ6ZqPSS56ovMPGqOu6whzBZRK__Vg,3498
-flock/core/mixin/dspy_integration.py,sha256=P5G4Y04nl5hFwFbJXCkQ-0TMR1L4skLL2IM_FlUjH_c,8364
+flock/core/mixin/dspy_integration.py,sha256=vlf6rJnR9EsfZi5KyFLEXIbUvhpBhodctn-mirYo7Pk,17382
 flock/core/mixin/prompt_parser.py,sha256=eOqI-FK3y17gVqpc_y5GF-WmK1Jv8mFlkZxTcgweoxI,5121
-flock/core/registry/agent_registry.py,sha256=TUClh9e3eA6YzZC1CMTlsTPvQeqb9jYHewi-zPpcWM8,4987
-flock/core/serialization/__init__.py,sha256=j0uw1g-7SOEhD5Ozgc8WpbZNpDWRmu58UOGg4_QTu8U,320
+flock/core/serialization/__init__.py,sha256=CML7fPgG6p4c0CDBlJ_uwV1aZZhJKK9uy3IoIHfO87w,431
+flock/core/serialization/callable_registry.py,sha256=sUZECTZWsM3fJ8FDRQ-FgLNW9hF26nY17AD6fJKADMc,1419
 flock/core/serialization/json_encoder.py,sha256=gAKj2zU_8wQiNvdkby2hksSA4fbPNwTjup_yz1Le1Vw,1229
 flock/core/serialization/secure_serializer.py,sha256=n5-zRvvXddgJv1FFHsaQ2wuYdL3WUSGPvG_LGaffEJo,6144
-flock/core/serialization/serializable.py,sha256=SymJ0YrjBx48mOBItYSqoRpKuzIc4vKWRS6ScTzre7s,2573
+flock/core/serialization/serializable.py,sha256=ymuJmm8teLbyUFMakkD7W3a_N8H4igZ1ITQnOqoLt0o,11270
+flock/core/serialization/serialization_utils.py,sha256=YxlPoHFPtj5otWUlKEsKphWxnd_d6b8yMZN_5MDlzB4,7577
 flock/core/tools/azure_tools.py,sha256=9Bi6IrB5pzBTBhBSxpCVMgx8HBud8nl4gDp8aN0NT6c,17031
 flock/core/tools/basic_tools.py,sha256=hEG14jNZ2itVvubCHTfsWkuJK6yuNwBtuFj2Js0VHZs,9043
 flock/core/tools/llm_tools.py,sha256=Bdt4Dpur5dGpxd2KFEQyxjfZazvW1HCDKY6ydMj6UgQ,21811
@@ -50,17 +64,17 @@ flock/core/tools/dev_tools/github.py,sha256=a2OTPXS7kWOVA4zrZHynQDcsmEi4Pac5MfSj
 flock/core/util/cli_helper.py,sha256=QSpP10WRNcjXzVFwpTQA8lSBy7707Qlv7uCit1XjUms,49808
 flock/core/util/hydrator.py,sha256=6qNwOwCZB7r6y25BZ--0PGofrAlfMaXbDKFQeP5NLts,11196
 flock/core/util/input_resolver.py,sha256=g9vDPdY4OH-G7qjas5ksGEHueokHGFPMoLOvC-ngeLo,5984
-flock/evaluators/declarative/declarative_evaluator.py,sha256=f8ldgZZp94zC4CoGzBufKvbvtckCGBe9EHTOoAZfZK0,1695
+flock/evaluators/declarative/declarative_evaluator.py,sha256=WZ74LG81JcuApG2KcTk8plh0fFqDhJjtl6ubW1K-fqc,1750
 flock/evaluators/memory/azure_search_evaluator.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flock/evaluators/memory/memory_evaluator.py,sha256=SmerXyNaqm8DTV0yw-WqWkn9DXIf6x-nPG1eyTV6NY8,3452
 flock/evaluators/natural_language/natural_language_evaluator.py,sha256=6nVEeh8_uwv_h-d3FWlA0GbzDzRtdhvxCGKirHtyvOU,2012
 flock/evaluators/zep/zep_evaluator.py,sha256=9NOELl7JAuUcx_FQrxY6b-_vN3MjwDyW7ZppPIGeCFc,1954
 flock/modules/azure-search/azure_search_module.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flock/modules/callback/callback_module.py,sha256=volGGgHtY19qj1wHR6m5a_hmXSbV3Ca3uY6I76YmcfU,2833
-flock/modules/memory/memory_module.py,sha256=dZ30eOFqIlAz0a5IKJMoXgJ-VyPEqApAOX0OQjhGA1I,14733
+flock/modules/memory/memory_module.py,sha256=bSkdFBW-Pp5ldHhXi8v4kfRM7zknfLR2fsOtbTosucI,14916
 flock/modules/memory/memory_parser.py,sha256=FLH7GL8XThvHiCMfX3eQH7Sz-f62fzhAUmO6_gaDI7U,4372
 flock/modules/memory/memory_storage.py,sha256=CNcLDMmvv0x7Z3YMKr6VveS_VCa7rKPw8l2d-XgqokA,27246
-flock/modules/output/output_module.py,sha256=kfHlnhdECC2k7uF_0VZ9_152zHI0qHJ2ZqGwYSsnoro,7621
+flock/modules/output/output_module.py,sha256=wOXMCyldFslj2iCMn2W5VZrSHCauMXwsW7YoFEHEcLY,7846
 flock/modules/performance/metrics_module.py,sha256=UD9OjY4-zAvauMD7YyDYqE1gyIhzpdr3JkBT8j9knxY,16790
 flock/modules/zep/zep_module.py,sha256=x7JG6O6xnwwum0RETIqKYbA3xzdcvX2aUuns0Cl0c2Q,6014
 flock/platform/docker_tools.py,sha256=fpA7-6rJBjPOUBLdQP4ny2QPgJ_042nmqRn5GtKnoYw,1445
@@ -410,12 +424,12 @@ flock/themes/zenburned.toml,sha256=UEmquBbcAO3Zj652XKUwCsNoC2iQSlIh-q5c6DH-7Kc,1
 flock/themes/zenwritten-dark.toml,sha256=To5l6520_3UqAGiEumpzGWsHhXxqu9ThrMildXKgIO0,1669
 flock/themes/zenwritten-light.toml,sha256=G1iEheCPfBNsMTGaVpEVpDzYBHA_T-MV27rolUYolmE,1666
 flock/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-flock/workflow/activities.py,sha256=yah-lHjMW6_Ww1gt7hMXBis1cJRlcbHx0uLsMB9oNZ0,9066
+flock/workflow/activities.py,sha256=eVZDnxGJl_quNO-UTV3YgvTV8LrRaHN3QDAA1ANKzac,9065
 flock/workflow/agent_activities.py,sha256=NhBZscflEf2IMfSRa_pBM_TRP7uVEF_O0ROvWZ33eDc,963
 flock/workflow/temporal_setup.py,sha256=VWBgmBgfTBjwM5ruS_dVpA5AVxx6EZ7oFPGw4j3m0l0,1091
 flock/workflow/workflow.py,sha256=I9MryXW_bqYVTHx-nl2epbTqeRy27CAWHHA7ZZA0nAk,1696
-flock_core-0.3.23.dist-info/METADATA,sha256=qwn0OaG4E4j-5vjAPcqSv6jqpxtL8e5NcuhyHgKHkCY,20649
-flock_core-0.3.23.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flock_core-0.3.23.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
-flock_core-0.3.23.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
-flock_core-0.3.23.dist-info/RECORD,,
+flock_core-0.3.31.dist-info/METADATA,sha256=wWrCmQuj7lyJu1mmgezo7gmkgjwyGh8HC3i-4DyF2xI,20745
+flock_core-0.3.31.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flock_core-0.3.31.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
+flock_core-0.3.31.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
+flock_core-0.3.31.dist-info/RECORD,,

flock/core/flock_api.py

@@ -1,214 +0,0 @@
-"""REST API server for Flock."""
-
-import uuid
-from datetime import datetime
-from typing import Any
-
-import uvicorn
-from fastapi import BackgroundTasks, FastAPI, HTTPException
-from pydantic import BaseModel, Field
-
-from flock.core.flock import Flock
-from flock.core.logging.logging import get_logger
-
-logger = get_logger("api")
-
-
-class FlockAPIRequest(BaseModel):
-    """Request model for running an agent."""
-
-    agent_name: str = Field(..., description="Name of the agent to run")
-    inputs: dict[str, Any] = Field(
-        default_factory=dict, description="Input data for the agent"
-    )
-    async_run: bool = Field(
-        default=False, description="Whether to run asynchronously"
-    )
-
-
-class FlockAPIResponse(BaseModel):
-    """Response model for run requests."""
-
-    run_id: str = Field(..., description="Unique ID for this run")
-    status: str = Field(..., description="Status of the run")
-    result: dict[str, Any] | None = Field(
-        None, description="Run result if completed"
-    )
-    started_at: datetime = Field(..., description="When the run started")
-    completed_at: datetime | None = Field(
-        None, description="When the run completed"
-    )
-    error: str | None = Field(None, description="Error message if failed")
-
-
-class FlockAPI:
-    """REST API server for Flock.
-
-    Provides HTTP endpoints for:
-    - Running agents
-    - Checking run status
-    - Getting run results
-    """
-
-    def __init__(self, flock: Flock):
-        self.flock = flock
-        self.app = FastAPI(title="Flock API")
-        self.runs: dict[str, FlockAPIResponse] = {}
-
-        # Register routes
-        self._setup_routes()
-
-    def _setup_routes(self):
-        """Set up API routes."""
-
-        @self.app.post("/run/flock", response_model=FlockAPIResponse)
-        async def run_flock(
-            request: FlockAPIRequest, background_tasks: BackgroundTasks
-        ):
-            """Run an agent with the provided inputs."""
-            try:
-                # Generate run ID
-                run_id = str(uuid.uuid4())
-
-                # Create initial response
-                response = FlockAPIResponse(
-                    run_id=run_id, status="starting", started_at=datetime.now()
-                )
-                self.runs[run_id] = response
-
-                if request.async_run:
-                    # Start run in background
-                    background_tasks.add_task(
-                        self._run_flock,
-                        run_id,
-                        request.agent_name,
-                        request.inputs,
-                    )
-                    response.status = "running"
-                else:
-                    # Run synchronously
-                    await self._run_flock(
-                        run_id, request.agent_name, request.inputs
-                    )
-
-                return self.runs[run_id]
-
-            except Exception as e:
-                logger.error(f"Error starting run: {e!s}")
-                raise HTTPException(status_code=500, detail=str(e))
-
-        @self.app.post("/run/agent", response_model=FlockAPIResponse)
-        async def run_agent(
-            request: FlockAPIRequest, background_tasks: BackgroundTasks
-        ):
-            """Run an agent with the provided inputs."""
-            try:
-                # Generate run ID
-                run_id = str(uuid.uuid4())
-
-                # Create initial response
-                response = FlockAPIResponse(
-                    run_id=run_id, status="starting", started_at=datetime.now()
-                )
-                self.runs[run_id] = response
-
-                if request.async_run:
-                    # Start run in background
-                    background_tasks.add_task(
-                        self._run_agent,
-                        run_id,
-                        request.agent_name,
-                        request.inputs,
-                    )
-                    response.status = "running"
-                else:
-                    # Run synchronously
-                    await self._run_agent(
-                        run_id, request.agent_name, request.inputs
-                    )
-
-                return self.runs[run_id]
-
-            except Exception as e:
-                logger.error(f"Error starting run: {e!s}")
-                raise HTTPException(status_code=500, detail=str(e))
-
-        @self.app.get("/run/{run_id}", response_model=FlockAPIResponse)
-        async def get_run_status(run_id: str):
-            """Get the status of a run."""
-            if run_id not in self.runs:
-                raise HTTPException(status_code=404, detail="Run not found")
-            return self.runs[run_id]
-
-        @self.app.get("/agents")
-        async def list_agents():
-            """List all available agents."""
-            return {
-                "agents": [
-                    {
-                        "name": agent.name,
-                        "description": agent.description,
-                        "input_schema": agent.input,
-                        "output_schema": agent.output,
-                    }
-                    for agent in self.flock.agents.values()
-                ]
-            }
-
-    async def _run_agent(
-        self, run_id: str, agent_name: str, inputs: dict[str, Any]
-    ):
-        """Execute an agent run."""
-        try:
-            # Get the agent
-            if agent_name not in self.flock.agents:
-                raise ValueError(f"Agent '{agent_name}' not found")
-
-            agent = self.flock.agents[agent_name]
-
-            # Run the agent
-            result = await agent.run_async(inputs)
-
-            # Update run status
-            self.runs[run_id].status = "completed"
-            self.runs[run_id].result = result
-            self.runs[run_id].completed_at = datetime.now()
-
-        except Exception as e:
-            logger.error(f"Error in run {run_id}: {e!s}")
-            self.runs[run_id].status = "failed"
-            self.runs[run_id].error = str(e)
-            self.runs[run_id].completed_at = datetime.now()
-
-    async def _run_flock(
-        self, run_id: str, agent_name: str, inputs: dict[str, Any]
-    ):
-        """Execute an agent run."""
-        try:
-            # Get the agent
-            if agent_name not in self.flock.agents:
-                raise ValueError(f"Agent '{agent_name}' not found")
-
-            result = await self.flock.run_async(
-                start_agent=agent_name, input=inputs
-            )
-
-            # Update run status
-            self.runs[run_id].status = "completed"
-            self.runs[run_id].result = result
-            self.runs[run_id].completed_at = datetime.now()
-
-        except Exception as e:
-            logger.error(f"Error in run {run_id}: {e!s}")
-            self.runs[run_id].status = "failed"
-            self.runs[run_id].error = str(e)
-            self.runs[run_id].completed_at = datetime.now()
-
-    def start(self, host: str = "0.0.0.0", port: int = 8344):
-        """Start the API server."""
-        uvicorn.run(self.app, host=host, port=port)
-
-    async def stop(self):
-        """Stop the API server."""
-        # Cleanup if needed
-        pass

flock/core/registry/agent_registry.py

@@ -1,120 +0,0 @@
-"""Registry for storing and managing agents and tools with logging and tracing integration."""
-
-from collections.abc import Callable
-
-from opentelemetry import trace
-
-from flock.core.flock_agent import FlockAgent
-from flock.core.flock_module import FlockModule
-from flock.core.logging.logging import get_logger
-
-logger = get_logger("registry")
-tracer = trace.get_tracer(__name__)
-
-
-class Registry:
-    """Registry for storing and managing agents and tools.
-
-    This singleton class maintains a centralized registry of agents and tools,
-    which is particularly important for Temporal workflows where only basic Python
-    types can be passed between activities.
-    """
-
-    _instance = None
-
-    def __new__(cls):
-        with tracer.start_as_current_span("Registry.__new__") as span:
-            if cls._instance is None:
-                cls._instance = super().__new__(cls)
-                cls._instance._initialize()
-                logger.info("Registry instance created")
-                span.set_attribute("instance.created", True)
-            return cls._instance
-
-    def _initialize(self):
-        with tracer.start_as_current_span("Registry._initialize"):
-            self._agents: list[FlockAgent] = []
-            self._tools: list[tuple[str, Callable]] = []
-            self._modules: list[FlockModule] = []
-            logger.info("Registry initialized", agents_count=0, tools_count=0)
-
-    def register_tool(self, tool_name: str, tool: Callable) -> None:
-        with tracer.start_as_current_span("Registry.register_tool") as span:
-            span.set_attribute("tool_name", tool_name)
-            try:
-                self._tools.append((tool_name, tool))
-                logger.info("Tool registered", tool_name=tool_name)
-            except Exception as e:
-                logger.error(
-                    "Error registering tool", tool_name=tool_name, error=str(e)
-                )
-                span.record_exception(e)
-                raise
-
-    def register_agent(self, agent: FlockAgent) -> None:
-        with tracer.start_as_current_span("Registry.register_agent") as span:
-            span.set_attribute("agent_name", agent.name)
-            try:
-                self._agents.append(agent)
-                logger.info("Agent registered", agent=agent.name)
-            except Exception as e:
-                logger.error(
-                    "Error registering agent", agent=agent.name, error=str(e)
-                )
-                span.record_exception(e)
-                raise
-
-    def get_agent(self, name: str) -> FlockAgent | None:
-        with tracer.start_as_current_span("Registry.get_agent") as span:
-            span.set_attribute("search_agent_name", name)
-            try:
-                for agent in self._agents:
-                    if agent.name == name:
-                        logger.info("Agent found", agent=name)
-                        span.set_attribute("found", True)
-                        return agent
-                logger.warning("Agent not found", agent=name)
-                span.set_attribute("found", False)
-                return None
-            except Exception as e:
-                logger.error("Error retrieving agent", agent=name, error=str(e))
-                span.record_exception(e)
-                raise
-
-    def get_tool(self, name: str) -> Callable | None:
-        with tracer.start_as_current_span("Registry.get_tool") as span:
-            span.set_attribute("search_tool_name", name)
-            try:
-                for tool_name, tool in self._tools:
-                    if tool_name == name:
-                        logger.info("Tool found", tool=name)
-                        span.set_attribute("found", True)
-                        return tool
-                logger.warning("Tool not found", tool=name)
-                span.set_attribute("found", False)
-                return None
-            except Exception as e:
-                logger.error("Error retrieving tool", tool=name, error=str(e))
-                span.record_exception(e)
-                raise
-
-    def get_tools(self, names: list[str] | None) -> list[Callable]:
-        with tracer.start_as_current_span("Registry.get_tools") as span:
-            span.set_attribute("search_tool_names", str(names))
-            try:
-                if not names:
-                    logger.info("No tool names provided")
-                    return []
-                tools = [self.get_tool(name) for name in names]
-                found_tools = [tool for tool in tools if tool is not None]
-                logger.info(
-                    "Tools retrieved", requested=names, found=len(found_tools)
-                )
-                span.set_attribute("found_tools_count", len(found_tools))
-                return found_tools
-            except Exception as e:
-                logger.error(
-                    "Error retrieving tools", names=str(names), error=str(e)
-                )
-                span.record_exception(e)
-                raise