flock-core 0.4.506__py3-none-any.whl → 0.4.508__py3-none-any.whl

flock/evaluators/declarative/declarative_evaluator.py

@@ -26,6 +26,8 @@ class DeclarativeEvaluatorConfig(FlockEvaluatorConfig):
     use_cache: bool = True
     temperature: float = 0.0
     max_tokens: int = 4096
+    max_retries: int = 3
+    max_tool_calls: int = 10
     stream: bool = Field(
         default=False,
         description="Enable streaming output from the underlying DSPy program.",
@@ -51,110 +53,132 @@ class DeclarativeEvaluator(
     _cost: float = PrivateAttr(default=0.0)
     _lm_history: list = PrivateAttr(default_factory=list)
 
+    # def __init__(self, name: str, config: DeclarativeEvaluatorConfig) -> None:
+    #     super().__init__(name=name, config=config)
+    # self._configure_language_model(
+    #     model=config.model,
+    #     use_cache=config.use_cache,
+    #     temperature=config.temperature,
+    #     max_tokens=config.max_tokens,
+    # )
+
     async def evaluate(
-        self, agent: FlockAgent, inputs: dict[str, Any], tools: list[Any], mcp_tools: list[Any] | None = None
+        self,
+        agent: FlockAgent,
+        inputs: dict[str, Any],
+        tools: list[Any],
+        mcp_tools: list[Any] | None = None,
     ) -> dict[str, Any]:
         """Evaluate using DSPy, with optional asynchronous streaming."""
         # --- Setup Signature and LM ---
-        try:
-            from rich.console import Console
-
-            console = Console()
-            _dspy_signature = self.create_dspy_signature_class(
-                agent.name,
-                agent.description,
-                f"{agent.input} -> {agent.output}",
-            )
-            # --- Get output field names ---
-            # dspy.Signature holds fields in .output_fields attribute
-            output_field_names = list(_dspy_signature.output_fields.keys())
-            if not output_field_names:
-                logger.warning(
-                    f"DSPy signature for agent '{agent.name}' has no defined output fields. Streaming might not produce text."
-                )
-            # -----------------------------
 
-            self._configure_language_model(
+        with dspy.context(
+            lm=dspy.LM(
                 model=self.config.model or agent.model,
-                use_cache=self.config.use_cache,
+                cache=self.config.use_cache,
                 temperature=self.config.temperature,
                 max_tokens=self.config.max_tokens,
+                num_retries=self.config.max_retries,
             )
-            agent_task = self._select_task(
-                _dspy_signature,
-                override_evaluator_type=self.config.override_evaluator_type,
-                tools=tools,
-                mcp_tools=mcp_tools,
-                kwargs=self.config.kwargs,
-            )
-        except Exception as setup_error:
-            logger.error(
-                f"Error setting up DSPy task for agent '{agent.name}': {setup_error}",
-                exc_info=True,
-            )
-            raise RuntimeError(
-                f"DSPy task setup failed: {setup_error}"
-            ) from setup_error
-
-        # --- Conditional Evaluation (Stream vs No Stream) ---
-        if self.config.stream:
-            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."
-                )
+        ):
+            try:
+                from rich.console import Console
 
-            streaming_task = dspy.streamify(agent_task, is_async_program=True)
-            stream_generator: Generator = streaming_task(**inputs)
-            delta_content = ""
-
-            console.print("\n")
-            async for chunk in stream_generator:
-                if (
-                    hasattr(chunk, "choices")
-                    and chunk.choices
-                    and hasattr(chunk.choices[0], "delta")
-                    and chunk.choices[0].delta
-                    and hasattr(chunk.choices[0].delta, "content")
-                ):
-                    delta_content = chunk.choices[0].delta.content
-
-                if delta_content:
-                    console.print(delta_content, end="")
-
-                result_dict, cost, lm_history = self._process_result(
-                    chunk, inputs
+                console = Console()
+                _dspy_signature = self.create_dspy_signature_class(
+                    agent.name,
+                    agent.description,
+                    f"{agent.input} -> {agent.output}",
                 )
-                self._cost = cost
-                self._lm_history = lm_history
-
-            console.print("\n")
-            return self.filter_thought_process(
-                result_dict, self.config.include_thought_process
-            )
-
-        else:  # Non-streaming path
-            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)
-                result_dict, cost, lm_history = self._process_result(
-                    result_obj, inputs
+                # --- Get output field names ---
+                # dspy.Signature holds fields in .output_fields attribute
+                output_field_names = list(_dspy_signature.output_fields.keys())
+                if not output_field_names:
+                    logger.warning(
+                        f"DSPy signature for agent '{agent.name}' has no defined output fields. Streaming might not produce text."
+                    )
+                # -----------------------------
+
+                agent_task = self._select_task(
+                    _dspy_signature,
+                    override_evaluator_type=self.config.override_evaluator_type,
+                    tools=tools,
+                    max_tool_calls=self.config.max_tool_calls,
+                    mcp_tools=mcp_tools,
+                    kwargs=self.config.kwargs,
+                )
+            except Exception as setup_error:
+                logger.error(
+                    f"Error setting up DSPy task for agent '{agent.name}': {setup_error}",
+                    exc_info=True,
                 )
-                self._cost = cost
-                self._lm_history = lm_history
+                raise RuntimeError(
+                    f"DSPy task setup failed: {setup_error}"
+                ) from setup_error
+
+            # --- Conditional Evaluation (Stream vs No Stream) ---
+            if self.config.stream:
+                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."
+                    )
+
+                streaming_task = dspy.streamify(
+                    agent_task, is_async_program=True
+                )
+                stream_generator: Generator = streaming_task(**inputs)
+                delta_content = ""
+
+                console.print("\n")
+                async for chunk in stream_generator:
+                    if (
+                        hasattr(chunk, "choices")
+                        and chunk.choices
+                        and hasattr(chunk.choices[0], "delta")
+                        and chunk.choices[0].delta
+                        and hasattr(chunk.choices[0].delta, "content")
+                    ):
+                        delta_content = chunk.choices[0].delta.content
+
+                    if delta_content:
+                        console.print(delta_content, end="")
+
+                    result_dict, cost, lm_history = self._process_result(
+                        chunk, inputs
+                    )
+                    self._cost = cost
+                    self._lm_history = lm_history
+
+                console.print("\n")
                 return self.filter_thought_process(
                     result_dict, self.config.include_thought_process
                 )
-            except Exception as e:
-                logger.error(
-                    f"Error during non-streaming evaluation for agent '{agent.name}': {e}",
-                    exc_info=True,
+
+            else:  # Non-streaming path
+                logger.info(
+                    f"Evaluating agent '{agent.name}' without streaming."
                 )
-                raise RuntimeError(f"Evaluation failed: {e}") from e
+                try:
+                    # Ensure the call is awaited if the underlying task is async
+                    result_obj = await agent_task.acall(**inputs)
+                    result_dict, cost, lm_history = self._process_result(
+                        result_obj, inputs
+                    )
+                    self._cost = cost
+                    self._lm_history = lm_history
+                    return self.filter_thought_process(
+                        result_dict, self.config.include_thought_process
+                    )
+                except Exception as e:
+                    logger.error(
+                        f"Error during non-streaming evaluation for agent '{agent.name}': {e}",
+                        exc_info=True,
+                    )
+                    raise RuntimeError(f"Evaluation failed: {e}") from e
 
     def filter_thought_process(
         self, result_dict: dict[str, Any], include_thought_process: bool

{flock_core-0.4.506.dist-info → flock_core-0.4.508.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flock-core
-Version: 0.4.506
+Version: 0.4.508
 Summary: Declarative LLM Orchestration at Scale
 Author-email: Andre Ratzenberger <andre.ratzenberger@whiteduck.de>
 License-File: LICENSE
@@ -9,20 +9,15 @@ Classifier: Operating System :: OS Independent
 Classifier: Programming Language :: Python :: 3
 Requires-Python: >=3.10
 Requires-Dist: aiosqlite>=0.21.0
-Requires-Dist: chromadb>=0.6.3
 Requires-Dist: cloudpickle>=3.1.1
-Requires-Dist: datasets>=3.2.0
 Requires-Dist: devtools>=0.12.2
 Requires-Dist: dspy==2.6.23
 Requires-Dist: fastapi>=0.115.8
 Requires-Dist: httpx>=0.28.1
-Requires-Dist: inspect-ai>=0.3.88
 Requires-Dist: litellm==1.69.3
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: markdown2>=2.5.3
-Requires-Dist: matplotlib>=3.10.0
 Requires-Dist: mcp>=1.7.1
-Requires-Dist: mem0ai[graph]>=0.1.101
 Requires-Dist: msgpack>=1.1.0
 Requires-Dist: openai==1.75.0
 Requires-Dist: opentelemetry-api>=1.30.0
@@ -31,6 +26,7 @@ Requires-Dist: opentelemetry-exporter-jaeger>=1.21.0
 Requires-Dist: opentelemetry-exporter-otlp>=1.30.0
 Requires-Dist: opentelemetry-instrumentation-logging>=0.51b0
 Requires-Dist: opentelemetry-sdk>=1.30.0
+Requires-Dist: opik>=1.7.26
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pillow>=10.4.0
 Requires-Dist: prometheus-client>=0.21.1
@@ -40,12 +36,9 @@ 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: rouge-score>=0.1.2
-Requires-Dist: sentence-transformers>=3.4.1
 Requires-Dist: temporalio>=1.9.0
 Requires-Dist: thefuzz>=0.22.1
 Requires-Dist: tiktoken>=0.8.0
@@ -53,7 +46,23 @@ Requires-Dist: toml>=0.10.2
 Requires-Dist: tqdm>=4.67.1
 Requires-Dist: uvicorn>=0.34.0
 Requires-Dist: wd-di>=0.2.14
-Requires-Dist: zep-python>=2.0.2
+Requires-Dist: websockets>=15.0.1
+Provides-Extra: all
+Requires-Dist: azure-identity>=1.23.0; extra == 'all'
+Requires-Dist: azure-search-documents>=11.5.2; extra == 'all'
+Requires-Dist: azure-storage-blob>=12.25.1; extra == 'all'
+Requires-Dist: chromadb>=0.6.3; extra == 'all'
+Requires-Dist: datasets>=3.2.0; extra == 'all'
+Requires-Dist: docling>=2.18.0; extra == 'all'
+Requires-Dist: duckduckgo-search>=7.3.2; extra == 'all'
+Requires-Dist: markdownify>=0.14.1; extra == 'all'
+Requires-Dist: matplotlib>=3.10.0; extra == 'all'
+Requires-Dist: mem0ai[graph]>=0.1.101; extra == 'all'
+Requires-Dist: nltk>=3.9.1; extra == 'all'
+Requires-Dist: rouge-score>=0.1.2; extra == 'all'
+Requires-Dist: sentence-transformers>=3.4.1; extra == 'all'
+Requires-Dist: tavily-python>=0.5.0; extra == 'all'
+Requires-Dist: zep-python>=2.0.2; extra == 'all'
 Provides-Extra: all-tools
 Requires-Dist: azure-identity>=1.23.0; extra == 'all-tools'
 Requires-Dist: azure-search-documents>=11.5.2; extra == 'all-tools'
@@ -72,8 +81,17 @@ Requires-Dist: docling>=2.18.0; extra == 'basic-tools'
 Requires-Dist: duckduckgo-search>=7.3.2; extra == 'basic-tools'
 Requires-Dist: markdownify>=0.14.1; extra == 'basic-tools'
 Requires-Dist: tavily-python>=0.5.0; extra == 'basic-tools'
+Provides-Extra: evaluation
+Requires-Dist: datasets>=3.2.0; extra == 'evaluation'
+Requires-Dist: rouge-score>=0.1.2; extra == 'evaluation'
+Requires-Dist: sentence-transformers>=3.4.1; extra == 'evaluation'
 Provides-Extra: llm-tools
 Requires-Dist: nltk>=3.9.1; extra == 'llm-tools'
+Provides-Extra: memory
+Requires-Dist: chromadb>=0.6.3; extra == 'memory'
+Requires-Dist: matplotlib>=3.10.0; extra == 'memory'
+Requires-Dist: mem0ai[graph]>=0.1.101; extra == 'memory'
+Requires-Dist: zep-python>=2.0.2; extra == 'memory'
 Description-Content-Type: text/markdown
 
 <p align="center">

{flock_core-0.4.506.dist-info → flock_core-0.4.508.dist-info}/RECORD

@@ -26,10 +26,10 @@ flock/cli/view_results.py,sha256=dOzK0O1FHSIDERnx48y-2Xke9BkOHS7pcOhs64AyIg0,781
 flock/cli/yaml_editor.py,sha256=K3N0bh61G1TSDAZDnurqW9e_-hO6CtSQKXQqlDhCjVo,12527
 flock/cli/assets/release_notes.md,sha256=bqnk50jxM3w5uY44Dc7MkdT8XmRREFxrVBAG9XCOSSU,4896
 flock/core/__init__.py,sha256=juwyNr3QqKXUS5-E3hlMYRhgqHgQBqgtP12OF3tUCAI,1249
-flock/core/flock.py,sha256=M7nk_GncnqPj8pHI4N-3R4hlPRR6BKsLGvoj6sJE89M,37108
-flock/core/flock_agent.py,sha256=0AzRr5xLM9cIpOUz6LGMjFeqDMun9wmcj2nc-HFl3Z4,48559
-flock/core/flock_evaluator.py,sha256=onbU6r146QVfdrHITF8AMH2U79enAls4QE2gJfpAYVw,1721
-flock/core/flock_factory.py,sha256=AXQzecI0fhbJxdDDP7VckqheYX77OJ6ohG0ZU6SxgE4,13873
+flock/core/flock.py,sha256=iR4i0_z0w2ns_iHbP7FqN--7wlsPIWch1H-BVecPs_I,38205
+flock/core/flock_agent.py,sha256=EHI0p3axr8X364wfHMB2js16MQT7H1Awe6SzJQR7bhw,48573
+flock/core/flock_evaluator.py,sha256=TPy6u6XX3cqkY1r9NW1w2lTwCMNW7pxhFYKLefnEbXg,1820
+flock/core/flock_factory.py,sha256=36eOtha-XG67_kK_VadG7UQ8cMgAcd7y1k5yFkT-pR0,14017
 flock/core/flock_module.py,sha256=ObILimpVaPnaaqYvcBYJJ20lQzfrjgTdADplaNRjHU0,7448
 flock/core/flock_registry.py,sha256=KzdFfc3QC-Dk42G24hdf6Prp3HvGj9ymXR3TTBe-T-A,27161
 flock/core/flock_router.py,sha256=1OAXDsdaIIFApEfo6SRfFEDoTuGt3Si7n2MXiySEfis,2644
@@ -42,13 +42,10 @@ flock/core/api/models.py,sha256=seqKuzhbN37nCNO7KrcJjI2mWuwiOKCLFcJcTPvTtag,3422
 flock/core/api/run_store.py,sha256=bFodJvVyWogzoezVy0cOoWWU3MdEBXf_6_5sBqCRWps,9227
 flock/core/api/runner.py,sha256=3izg6cVk1RoR1hDIDwMAO1gi3lnLcp8DPv7AnJBYx6A,1443
 flock/core/api/service.py,sha256=HRHs4xt-bGeSm5hdN92H1vWQtLzqZalhZxIh6iwww8Y,11381
-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=zdQuB1YWPJmQVv_2_sm1HK7FSnusa3Jl-83PcTuaLUk,7791
 flock/core/context/context_manager.py,sha256=FANSWa6DEhdhtZ7t_9Gza0v80UdpoDOhHbfVOccmjkA,1181
 flock/core/context/context_vars.py,sha256=ASPA29hpENWub4mgRoG62FtTVakCHQZfn6IhJQKe3C8,347
-flock/core/evaluation/utils.py,sha256=ZJkIMC9YT-HA2SPCZ4_bQ98isW1i6nbltVEYbjze-b0,12827
+flock/core/evaluation/utils.py,sha256=S5M0uTFcClphZsR5EylEzrRNK-1434yImiGYL4pR_5o,15380
 flock/core/execution/batch_executor.py,sha256=mHwCI-DHqApCv_EVCN0ZOUd-LCQLjREpxKbAUPC0pcY,15266
 flock/core/execution/evaluation_executor.py,sha256=D9EO0sU-2qWj3vomjmUUi-DOtHNJNFRf30kGDHuzREE,17702
 flock/core/execution/local_executor.py,sha256=rnIQvaJOs6zZORUcR3vvyS6LPREDJTjaygl_Db0M8ao,952
@@ -79,7 +76,7 @@ flock/core/mcp/types/handlers.py,sha256=mhAoSS6vA3K7YP_mF9Sb2Zn2RvFM8fXxtbwC4zgU
 flock/core/mcp/types/types.py,sha256=NFVFv0BXy4P79SGe8rKBO4VH41xkpF6Y-yvU3Fmzx6w,4872
 flock/core/mcp/util/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flock/core/mcp/util/helpers.py,sha256=Xlf4iKW_lZxsVMTRoOnV29JsJfAppfmEJrb6sIcoCH4,636
-flock/core/mixin/dspy_integration.py,sha256=iJLKyMyebQ-gMKGgw9v1FokLRe4J2vUFm1qJYp2x1-U,17685
+flock/core/mixin/dspy_integration.py,sha256=LQN4cfcG6Myi6kX0-2bKCuK1K_k_lu3SNaTdHHk3lEU,17745
 flock/core/mixin/prompt_parser.py,sha256=eOqI-FK3y17gVqpc_y5GF-WmK1Jv8mFlkZxTcgweoxI,5121
 flock/core/serialization/__init__.py,sha256=CML7fPgG6p4c0CDBlJ_uwV1aZZhJKK9uy3IoIHfO87w,431
 flock/core/serialization/callable_registry.py,sha256=sUZECTZWsM3fJ8FDRQ-FgLNW9hF26nY17AD6fJKADMc,1419
@@ -96,7 +93,7 @@ flock/core/util/loader.py,sha256=j3q2qem5bFMP2SmMuYjb-ISxsNGNZd1baQmpvAnRUUk,224
 flock/core/util/spliter.py,sha256=rDLnZX158PWkmW8vi2UfMLAMRXcHQFUIydAABd-lDGw,7154
 flock/evaluators/__init__.py,sha256=Y0cEkx0dujRmy--TDpKoTqFSLzbyFz8BwEOv8kdSUhg,22
 flock/evaluators/declarative/__init__.py,sha256=Y0cEkx0dujRmy--TDpKoTqFSLzbyFz8BwEOv8kdSUhg,22
-flock/evaluators/declarative/declarative_evaluator.py,sha256=KOdZcQcezb-3cBBCH-d75tDTdaf4MJxFmsomC92oQRE,6436
+flock/evaluators/declarative/declarative_evaluator.py,sha256=tulTpUOXhF-wMe5a9ULpsCiS1o2Z-DOXOUTqOSzVYqI,7397
 flock/evaluators/memory/memory_evaluator.py,sha256=ySwz7kcc8suXMJ7gKNSWThW8iOMlE8lUcUzEAHvv8rw,3559
 flock/evaluators/test/test_case_evaluator.py,sha256=3Emcoty0LOLLBIuPGxSpKphuZC9Fu1DTr1vbGg-hd0Q,1233
 flock/evaluators/zep/zep_evaluator.py,sha256=6_5vTdU0yJAH8I8w3-MPXiAZx6iUPhAVCsHjrHzkPLM,2058
@@ -557,8 +554,8 @@ flock/workflow/agent_execution_activity.py,sha256=Gy6FtuVAjf0NiUXmC3syS2eJpNQF4R
 flock/workflow/flock_workflow.py,sha256=iSUF_soFvWar0ffpkzE4irkDZRx0p4HnwmEBi_Ne2sY,9666
 flock/workflow/temporal_config.py,sha256=3_8O7SDEjMsSMXsWJBfnb6XTp0TFaz39uyzSlMTSF_I,3988
 flock/workflow/temporal_setup.py,sha256=YIHnSBntzOchHfMSh8hoLeNXrz3B1UbR14YrR6soM7A,1606
-flock_core-0.4.506.dist-info/METADATA,sha256=kVh7DDErkkrI4zEXvQV2lR1BNmlvGfSiA9dYGuObKXo,21599
-flock_core-0.4.506.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flock_core-0.4.506.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
-flock_core-0.4.506.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
-flock_core-0.4.506.dist-info/RECORD,,
+flock_core-0.4.508.dist-info/METADATA,sha256=_THooyyskEuH1CU8OfPv_CcHecZ1lEtSTFjeoOS5Q24,22584
+flock_core-0.4.508.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flock_core-0.4.508.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
+flock_core-0.4.508.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
+flock_core-0.4.508.dist-info/RECORD,,

flock/core/api/ui/routes.py

@@ -1,271 +0,0 @@
-# src/flock/core/api/ui/routes.py
-"""FastHTML UI routes for the Flock API."""
-
-from typing import TYPE_CHECKING
-
-# --- Conditional FastHTML Imports ---
-try:
-    import httpx
-    from fasthtml.common import *
-
-    # Import Form explicitly with an alias to avoid collisions
-    from fasthtml.common import Form as FHForm
-
-    FASTHTML_AVAILABLE = True
-except ImportError:
-    FASTHTML_AVAILABLE = False
-
-    # Define necessary dummies if not available
-    class Request:
-        pass
-
-    class Titled:
-        pass
-
-    class Div:
-        pass
-
-    class H1:
-        pass
-
-    class P:
-        pass
-
-    class H2:
-        pass
-
-    class Pre:
-        pass
-
-    class Code:
-        pass
-
-    class Label:
-        pass
-
-    class Select:
-        pass
-
-    class Option:
-        pass
-
-    class FHForm:
-        pass  # Dummy alias if not available
-
-    class Button:
-        pass
-
-    class Span:
-        pass
-
-    class Script:
-        pass
-
-    class Style:
-        pass
-
-    class Hidden:
-        pass
-
-    class Textarea:
-        pass
-
-    class Input:
-        pass
-
-    def fast_app():
-        return None, None
-
-    def picolink():
-        return None
-# ------------------------------------
-
-# Use TYPE_CHECKING to avoid circular import errors for type hints
-if TYPE_CHECKING:
-    from flock.core.api.main import FlockAPI
-
-# Import logger and utils needed by UI routes
-from flock.core.logging.logging import get_logger
-
-logger = get_logger("api.ui")
-
-
-def create_ui_app(
-    flock_api_instance: "FlockAPI",
-    api_host: str,
-    api_port: int,
-    server_name: str,
-) -> Any:
-    """Creates and configures the FastHTML application and its routes."""
-    if not FASTHTML_AVAILABLE:
-        raise ImportError("FastHTML is not installed. Cannot create UI.")
-    logger.debug("Creating FastHTML application instance for UI")
-
-    # Use the passed FlockAPI instance to access necessary data/methods
-    flock_instance = flock_api_instance.flock
-    parse_input_spec_func = (
-        flock_api_instance._parse_input_spec
-    )  # Get reference to parser
-
-    fh_app, fh_rt = fast_app(
-        hdrs=(
-            Script(src="https://unpkg.com/htmx.org@1.9.10/dist/htmx.min.js"),
-            picolink,  # Pass directly
-            Style("""
-            body { padding: 20px; max-width: 800px; margin: auto; font-family: sans-serif; }
-            label { display: block; margin-top: 1rem; font-weight: bold;}
-            input, select, textarea { width: 100%; margin-top: 0.25rem; padding: 0.5rem; border: 1px solid #ccc; border-radius: 4px; box-sizing: border-box; }
-            input[type=checkbox] { width: auto; margin-right: 0.5rem; vertical-align: middle; }
-            label[for^=input_] { font-weight: normal; display: inline; margin-top: 0;} /* Style for checkbox labels */
-            button[type=submit] { margin-top: 1.5rem; padding: 0.75rem 1.5rem; background-color: #007bff; color: white; border: none; border-radius: 4px; cursor: pointer; font-size: 1rem;}
-            button[type=submit]:hover { background-color: #0056b3; }
-            #result-area { margin-top: 2rem; background-color: #f8f9fa; padding: 15px; border: 1px solid #dee2e6; border-radius: 5px; white-space: pre-wrap; word-wrap: break-word; font-family: monospace; }
-            .htmx-indicator { display: none; margin-left: 10px; font-style: italic; color: #6c757d; }
-            .htmx-request .htmx-indicator { display: inline; }
-            .htmx-request.htmx-indicator { display: inline; }
-            .error-message { color: #721c24; margin-top: 10px; font-weight: bold; background-color: #f8d7da; border: 1px solid #f5c6cb; padding: 10px; border-radius: 5px;}
-        """),
-        )
-    )
-
-    @fh_rt("/get-agent-inputs")
-    def get_agent_inputs(request: Request):
-        """Endpoint called by HTMX to get agent input fields."""
-        agent_name = request.query_params.get("agent_name")
-        logger.debug(f"UI requesting inputs for agent: {agent_name}")
-        if not agent_name:
-            return Div("Please select an agent.", cls="error-message")
-
-        # Access agents via the passed FlockAPI instance
-        agent_def = flock_instance.agents.get(agent_name)
-        if not agent_def:
-            logger.warning(f"Agent '{agent_name}' not found for UI.")
-            return Div(f"Agent '{agent_name}' not found.", cls="error-message")
-
-        # Use the parsing function from the FlockAPI instance
-        input_fields = parse_input_spec_func(agent_def.input or "")
-        logger.debug(f"Parsed input fields for {agent_name}: {input_fields}")
-
-        inputs_html = []
-        for field in input_fields:
-            field_id = f"input_{field['name']}"
-            label_text = f"{field['name']}"
-            if field["type"] != "bool":
-                label_text += f" ({field['type']})"
-            label = Label(label_text, fr=field_id)
-            input_attrs = dict(
-                id=field_id,
-                name=f"inputs.{field['name']}",
-                type=field["html_type"],
-            )
-            if field.get("step"):
-                input_attrs["step"] = field["step"]
-            if field.get("desc"):
-                input_attrs["placeholder"] = field["desc"]
-            if field.get("rows"):
-                input_attrs["rows"] = field["rows"]
-
-            if field["html_type"] == "textarea":
-                input_el = Textarea(**input_attrs)
-            elif field["html_type"] == "checkbox":
-                input_el = Div(
-                    Input(**input_attrs, value="true"),
-                    Label(f" Enable?", fr=field_id),
-                )
-            else:
-                input_el = Input(**input_attrs)
-
-            inputs_html.append(
-                Div(label, input_el, style="margin-bottom: 1rem;")
-            )
-
-        inputs_html.append(
-            Hidden(
-                id="selected_agent_name", name="agent_name", value=agent_name
-            )
-        )
-        return (
-            Div(*inputs_html)
-            if inputs_html
-            else P("This agent requires no input.")
-        )
-
-    @fh_rt("/")
-    async def ui_root(request: Request):
-        """Serves the main UI page."""
-        logger.info("Serving main UI page /ui/")
-        agents_list = []
-        error_msg = None
-        api_url = f"http://{api_host}:{api_port}/agents"
-        try:
-            async with httpx.AsyncClient() as client:
-                logger.debug(f"UI fetching agents from {api_url}")
-                response = await client.get(api_url)
-                response.raise_for_status()
-                agent_data = response.json()
-                agents_list = agent_data.get("agents", [])
-                logger.debug(f"Fetched {len(agents_list)} agents for UI")
-        except Exception as e:
-            error_msg = f"UI Error: Could not fetch agent list from API at {api_url}. Details: {e}"
-            logger.error(error_msg, exc_info=True)
-
-        options = [
-            Option("-- Select Agent --", value="", selected=True, disabled=True)
-        ] + [
-            Option(
-                f"{agent['name']}: {agent['description']}", value=agent["name"]
-            )
-            for agent in agents_list
-        ]
-
-        # Use FHForm alias here
-        content = Div(
-            H2(f"Agent Runner"),
-            P(
-                "Select an agent, provide the required inputs, and click 'Run Flock'."
-            ),
-            Label("Select Starting Agent:", fr="agent_select"),
-            Select(
-                *options,
-                id="agent_select",
-                name="agent_name",
-                hx_get="/ui/get-agent-inputs",
-                hx_trigger="change",
-                hx_target="#agent-inputs-container",
-                hx_indicator="#loading-indicator",
-            ),
-            FHForm(
-                Div(id="agent-inputs-container", style="margin-top: 1rem;"),
-                Button("Run Flock", type="submit"),
-                Span(
-                    " Processing...",
-                    id="loading-indicator",
-                    cls="htmx-indicator",
-                ),
-                hx_post="/ui/run-agent-form",  # Target the dedicated form endpoint
-                hx_target="#result-area",
-                hx_swap="innerHTML",
-                hx_indicator="#loading-indicator",
-            ),
-            H2("Result"),
-            Div(
-                Pre(
-                    Code(
-                        "Result will appear here...",
-                        id="result-content",
-                        class_="language-json",
-                    )
-                ),
-                id="result-area",
-                style="min-height: 100px;",
-            ),
-        )
-
-        if error_msg:
-            content = Div(
-                H1("Flock UI - Error"), P(error_msg, cls="error-message")
-            )
-
-        return Titled(f"{server_name}", content)
-
-    return fh_app