flock-core 0.5.0b21__py3-none-any.whl → 0.5.0b22__py3-none-any.whl

flock/webapp/app/api/execution.py

@@ -1,6 +1,8 @@
 # src/flock/webapp/app/api/execution.py
+import asyncio
 import html
 import json
+import uuid
 from pathlib import Path
 from typing import TYPE_CHECKING, Any, Literal
 
@@ -12,7 +14,7 @@ from fastapi import (  # Ensure Form and HTTPException are imported
     Request,
 )
 from fastapi.encoders import jsonable_encoder
-from fastapi.responses import FileResponse, HTMLResponse
+from fastapi.responses import FileResponse, HTMLResponse, StreamingResponse
 from fastapi.templating import Jinja2Templates
 from werkzeug.utils import secure_filename
 
@@ -40,10 +42,6 @@ from flock.webapp.app.dependencies import (
 )
 
 # Service function now takes app_state
-from flock.webapp.app.services.flock_service import (
-    run_current_flock_service,
-    # get_current_flock_instance IS NO LONGER IMPORTED
-)
 from flock.webapp.app.services.sharing_store import SharedLinkStoreInterface
 
 router = APIRouter()
@@ -59,6 +57,183 @@ def markdown_filter(text):
 templates.env.filters["markdown"] = markdown_filter
 
 
+class ExecutionStreamManager:
+    """In-memory tracker for live streaming sessions."""
+
+    def __init__(self) -> None:
+        self._sessions: dict[str, asyncio.Queue] = {}
+        self._lock = asyncio.Lock()
+
+    async def create_session(self) -> tuple[str, asyncio.Queue]:
+        run_id = uuid.uuid4().hex
+        queue: asyncio.Queue = asyncio.Queue()
+        async with self._lock:
+            self._sessions[run_id] = queue
+        return run_id, queue
+
+    async def get_queue(self, run_id: str) -> asyncio.Queue | None:
+        async with self._lock:
+            return self._sessions.get(run_id)
+
+    async def remove_session(self, run_id: str) -> None:
+        async with self._lock:
+            self._sessions.pop(run_id, None)
+
+
+execution_stream_manager = ExecutionStreamManager()
+stream_logger = get_flock_logger("webapp.execution.stream")
+
+
+async def _execute_agent_with_stream(
+    run_id: str,
+    queue: asyncio.Queue,
+    start_agent_name: str,
+    inputs: dict[str, Any],
+    app_state: Any,
+    template_context: dict[str, Any],
+) -> None:
+    """Run the requested agent while forwarding streaming chunks to the UI."""
+
+    completed = False
+
+    def emit(payload: dict[str, Any]) -> None:
+        try:
+            queue.put_nowait(payload)
+        except asyncio.QueueFull:
+            stream_logger.warning(
+                "Dropping streaming payload for run %s due to full queue", run_id
+            )
+
+    async def terminate_with_error(message: str) -> None:
+        emit({"type": "error", "message": message})
+        await finalize_stream()
+
+    async def finalize_stream() -> None:
+        nonlocal completed
+        if not completed:
+            emit({"type": "complete"})
+            completed = True
+        await queue.put(None)
+
+    current_flock: "Flock | None" = getattr(app_state, "flock_instance", None)
+    run_store: RunStore | None = getattr(app_state, "run_store", None)
+
+    if not current_flock:
+        stream_logger.error("Stream run aborted: no flock loaded in app state.")
+        await terminate_with_error("No Flock loaded in the application.")
+        return
+
+    agent = current_flock.agents.get(start_agent_name)
+    if not agent:
+        stream_logger.error(
+            "Stream run aborted: agent '%s' not found in flock '%s'.",
+            start_agent_name,
+            current_flock.name,
+        )
+        await terminate_with_error(
+            f"Agent '{html.escape(str(start_agent_name))}' not found."
+        )
+        return
+
+    evaluator = getattr(agent, "evaluator", None)
+    previous_callbacks: list[Any] | None = None
+    original_stream_setting: bool | None = None
+
+    if evaluator is not None:
+        previous_callbacks = list(evaluator.config.stream_callbacks or [])
+        original_stream_setting = getattr(evaluator.config, "stream", False)
+
+        def stream_callback(message: Any) -> None:
+            chunk = getattr(message, "chunk", None)
+            signature_field = getattr(message, "signature_field_name", None)
+            if chunk is None:
+                return
+            emit(
+                {
+                    "type": "token",
+                    "chunk": str(chunk),
+                    "field": signature_field,
+                }
+            )
+
+        evaluator.config.stream_callbacks = [
+            *previous_callbacks,
+            stream_callback,
+        ]
+        if not original_stream_setting:
+            evaluator.config.stream = True
+    else:
+        emit(
+            {
+                "type": "status",
+                "message": "Streaming not available for this agent; results will appear when the run completes.",
+            }
+        )
+
+    try:
+        emit(
+            {
+                "type": "status",
+                "message": f"Running agent '{start_agent_name}'...",
+            }
+        )
+        result_data = await current_flock.run_async(
+            agent=start_agent_name, input=inputs, box_result=False
+        )
+
+        if run_store and hasattr(run_store, "add_run_details"):
+            run_identifier = (
+                result_data.get("run_id", run_id)
+                if isinstance(result_data, dict)
+                else run_id
+            )
+            run_store.add_run_details(
+                run_id=run_identifier,
+                agent_name=start_agent_name,
+                inputs=inputs,
+                outputs=result_data,
+            )
+
+        encoded_result = jsonable_encoder(result_data)
+        raw_json = json.dumps(
+            encoded_result, indent=2, ensure_ascii=False
+        ).replace("\\n", "\n")
+
+        template = templates.get_template("partials/_results_display.html")
+        final_html = template.render(
+            {
+                **template_context,
+                "result": result_data,
+                "result_raw_json": raw_json,
+            }
+        )
+
+        emit(
+            {
+                "type": "final",
+                "html": final_html,
+                "result": encoded_result,
+                "raw_json": raw_json,
+            }
+        )
+    except Exception as exc:  # pragma: no cover - defensive logging
+        stream_logger.error(
+            "Streamed execution for agent '%s' failed: %s",
+            start_agent_name,
+            exc,
+            exc_info=True,
+        )
+        await terminate_with_error(f"An error occurred: {html.escape(str(exc))}")
+        return
+    finally:
+        if evaluator is not None:
+            if previous_callbacks is not None:
+                evaluator.config.stream_callbacks = previous_callbacks
+            if original_stream_setting is not None:
+                evaluator.config.stream = original_stream_setting
+    await finalize_stream()
+
+
 @router.get("/htmx/execution-form-content", response_class=HTMLResponse)
 async def htmx_get_execution_form_content(
     request: Request,
@@ -203,35 +378,68 @@ async def htmx_run_flock(
                 f"<p class='error'>Error processing inputs for {html.escape(str(start_agent_name))}: {html.escape(str(e_parse))}</p>"
             )
 
-    result_data = await run_current_flock_service(
-        start_agent_name, inputs, request.app.state
-    )
+    run_id, queue = await execution_stream_manager.create_session()
+    stream_url = str(request.url_for("htmx_stream_run", run_id=run_id))
+    root_path = request.scope.get("root_path", "")
 
-    raw_json_for_template = json.dumps(
-        jsonable_encoder(
-            result_data
-        ),  # ← converts every nested BaseModel, datetime, etc.
-        indent=2,
-        ensure_ascii=False,
+    template_context = {
+        "request": request,
+        "feedback_endpoint": f"{root_path}/ui/api/flock/htmx/feedback",
+        "share_id": None,
+        "flock_name": current_flock_from_state.name,
+        "agent_name": start_agent_name,
+        "flock_definition": current_flock_from_state.to_yaml(),
+    }
+
+    asyncio.create_task(
+        _execute_agent_with_stream(
+            run_id=run_id,
+            queue=queue,
+            start_agent_name=start_agent_name,
+            inputs=inputs,
+            app_state=request.app.state,
+            template_context=template_context,
+        )
     )
-    # Unescape newlines for proper display in HTML <pre> tag
-    result_data_raw_json_str = raw_json_for_template.replace("\\n", "\n")
-    root_path = request.scope.get("root_path", "")
+
     return templates.TemplateResponse(
-        "partials/_results_display.html",
+        "partials/_streaming_results_container.html",
         {
             "request": request,
-            "result": result_data,
-            "result_raw_json": result_data_raw_json_str,
-            "feedback_endpoint": f"{root_path}/ui/api/flock/htmx/feedback",
-            "share_id": None,
-            "flock_name": current_flock_from_state.name,
+            "run_id": run_id,
+            "stream_url": stream_url,
             "agent_name": start_agent_name,
-            "flock_definition": current_flock_from_state.to_yaml(),
+            "flock_name": current_flock_from_state.name,
         },
     )
 
 
+@router.get("/htmx/run-stream/{run_id}")
+async def htmx_stream_run(run_id: str):
+    """Server-Sent Events endpoint streaming live agent output."""
+
+    queue = await execution_stream_manager.get_queue(run_id)
+    if queue is None:
+        return HTMLResponse(
+            "<p class='error'>Streaming session not found or already closed.</p>",
+            status_code=404,
+        )
+
+    async def event_generator():
+        try:
+            while True:
+                payload = await queue.get()
+                if payload is None:
+                    yield "event: close\ndata: {}\n\n"
+                    break
+                data = json.dumps(payload, ensure_ascii=False)
+                yield f"data: {data}\n\n"
+        finally:
+            await execution_stream_manager.remove_session(run_id)
+
+    return StreamingResponse(event_generator(), media_type="text/event-stream")
+
+
 # --- NEW ENDPOINT FOR SHARED RUNS ---
 @router.post("/htmx/run-shared", response_class=HTMLResponse)
 async def htmx_run_shared_flock(

flock/webapp/templates/partials/_streaming_results_container.html

@@ -0,0 +1,195 @@
+<div id="streaming-results-wrapper"
+     data-run-id="{{ run_id }}"
+     data-stream-url="{{ stream_url }}"
+     data-target-id="results-display"
+     data-agent-name="{{ agent_name }}">
+    <header style="margin-bottom: 0.75rem;">
+        <h5 style="margin: 0;">Streaming {{ agent_name }}</h5>
+        <p style="margin: 0; color: var(--pico-muted-color);">Live output appears below while the agent runs.</p>
+    </header>
+
+    <p class="error" data-role="error" hidden></p>
+    <div class="stream-output" data-role="output" style="min-height: 8rem; white-space: normal; word-break: break-word; font-family: var(--pico-code-font-family, monospace);">Connecting to agent…</div>
+    <div data-role="progress" style="margin-top: 0.5rem;" role="status">
+        <progress indeterminate></progress> Streaming response…
+    </div>
+</div>
+
+<script>
+(function () {
+    const script = document.currentScript;
+    if (!script) {
+        return;
+    }
+    const wrapper = script.previousElementSibling;
+    if (!(wrapper instanceof HTMLElement)) {
+        return;
+    }
+    if (wrapper.dataset.streamInit === '1') {
+        return;
+    }
+    wrapper.dataset.streamInit = '1';
+
+    const streamUrl = wrapper.dataset.streamUrl;
+    const runId = wrapper.dataset.runId;
+    const targetId = wrapper.dataset.targetId || 'results-display';
+
+    const outputEl = wrapper.querySelector('[data-role="output"]');
+    const errorEl = wrapper.querySelector('[data-role="error"]');
+    const progressEl = wrapper.querySelector('[data-role="progress"]');
+
+    if (!streamUrl || !runId || !(outputEl instanceof HTMLElement)) {
+        if (errorEl instanceof HTMLElement) {
+            errorEl.textContent = 'Streaming setup failed due to missing metadata.';
+            errorEl.hidden = false;
+        }
+        if (progressEl instanceof HTMLElement) {
+            progressEl.hidden = true;
+        }
+        return;
+    }
+
+    let plainText = outputEl instanceof HTMLElement ? outputEl.textContent || '' : '';
+    let source;
+    const fieldValues = new Map();
+    let latestStatus = '';
+
+    function escapeHtml(value) {
+        return value.replace(/[&<>"']/g, (char) => ({
+            '&': '&amp;',
+            '<': '&lt;',
+            '>': '&gt;',
+            '"': '&quot;',
+            "'": '&#39;',
+        })[char]);
+    }
+
+    function renderTableCell(value) {
+        return escapeHtml(value).replace(/\n/g, '<br>');
+    }
+
+    function renderStream() {
+        if (!(outputEl instanceof HTMLElement)) {
+            return;
+        }
+        if (fieldValues.size > 0) {
+            outputEl.style.whiteSpace = 'normal';
+            let rows = '';
+            fieldValues.forEach((value, field) => {
+                rows += `<tr><td class="stream-field" style="white-space: nowrap; padding-right: 1rem; vertical-align: top;">${escapeHtml(field)}</td><td>${renderTableCell(value)}</td></tr>`;
+            });
+            if (latestStatus) {
+                rows += `<tr class="stream-status"><td class="stream-field" style="white-space: nowrap; padding-right: 1rem; vertical-align: top;">status</td><td>${escapeHtml(latestStatus)}</td></tr>`;
+            }
+            outputEl.innerHTML = `<table class="structured-table streaming-table" style="width:100%; border-collapse: collapse; table-layout: auto;"><tbody>${rows}</tbody></table>`;
+        } else {
+            outputEl.style.whiteSpace = 'pre-wrap';
+            outputEl.textContent = plainText;
+        }
+    }
+
+    function showError(message) {
+        if (errorEl instanceof HTMLElement) {
+            errorEl.textContent = message;
+            errorEl.hidden = false;
+        }
+        if (progressEl instanceof HTMLElement) {
+            progressEl.hidden = true;
+        }
+    }
+
+    function closeStream() {
+        if (source) {
+            source.close();
+            source = undefined;
+        }
+        if (progressEl instanceof HTMLElement) {
+            progressEl.hidden = true;
+        }
+        renderStream();
+    }
+
+    function renderFinal(html, rawJson) {
+        const target = document.getElementById(targetId);
+        if (!target) {
+            return;
+        }
+        target.innerHTML = html;
+        if (window.htmx) {
+            window.htmx.process(target);
+        }
+        if (window.Prism) {
+            window.Prism.highlightAllUnder(target);
+        }
+        if (outputEl instanceof HTMLElement) {
+            outputEl.textContent = '';
+        }
+    }
+
+    try {
+        source = new EventSource(streamUrl);
+    } catch (err) {
+        console.error('Failed to start EventSource', err);
+        showError('Failed to connect for streaming.');
+        return;
+    }
+
+    source.onmessage = (event) => {
+        if (!event.data) {
+            return;
+        }
+        let payload;
+        try {
+            payload = JSON.parse(event.data);
+        } catch (err) {
+            console.warn('Unable to parse streaming payload', err);
+            return;
+        }
+
+        switch (payload.type) {
+            case 'token':
+                if (typeof payload.chunk === 'string') {
+                    if (payload.field) {
+                        const existing = fieldValues.get(payload.field) || '';
+                        fieldValues.set(payload.field, existing + payload.chunk);
+                    } else {
+                        plainText = (plainText === 'Connecting to agent…' ? '' : plainText) + payload.chunk;
+                    }
+                    renderStream();
+                }
+                break;
+            case 'status':
+                if (payload.message) {
+                    latestStatus = payload.message;
+                    if (fieldValues.size === 0) {
+                        plainText = (plainText === 'Connecting to agent…' ? '' : plainText);
+                        if (plainText && !plainText.endsWith('\n')) {
+                            plainText += '\n';
+                        }
+                        plainText += payload.message + '\n';
+                    }
+                    renderStream();
+                }
+                break;
+            case 'error':
+                showError(payload.message || 'An unexpected error occurred while streaming.');
+                closeStream();
+                break;
+            case 'final':
+                closeStream();
+                renderFinal(payload.html || '', payload.raw_json || payload.rawJson);
+                break;
+            case 'complete':
+                closeStream();
+                break;
+            default:
+                break;
+        }
+    };
+
+    source.onerror = () => {
+        showError('Connection lost while streaming.');
+        closeStream();
+    };
+})();
+</script>

{flock_core-0.5.0b21.dist-info → flock_core-0.5.0b22.dist-info}/METADATA

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

{flock_core-0.5.0b21.dist-info → flock_core-0.5.0b22.dist-info}/RECORD

@@ -490,7 +490,7 @@ flock/webapp/app/theme_mapper.py,sha256=QzWwLWpED78oYp3FjZ9zxv1KxCyj43m8MZ0fhfzz
 flock/webapp/app/utils.py,sha256=RF8DMKKAj1XPmm4txUdo2OdswI1ATQ7cqUm6G9JFDzA,2942
 flock/webapp/app/api/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 flock/webapp/app/api/agent_management.py,sha256=5xqO94QjjAYvxImyjKV9EGUQOvo4n3eqs7pGwGPSQJ4,10394
-flock/webapp/app/api/execution.py,sha256=OzTjCP5CxAGdD2YrX7vI2qkQejqikX9Jn8_sq2o6yKA,18163
+flock/webapp/app/api/execution.py,sha256=D-gqeF0JRZ_9TZTuf1-7gLIPL-p8F9Qp66CjVgJKSlw,24758
 flock/webapp/app/api/flock_management.py,sha256=1o-6-36kTnUjI3am_BqLpdrcz0aqFXrxE-hQHIFcCsg,4869
 flock/webapp/app/api/registry_viewer.py,sha256=IoInxJiRR0yFlecG_l2_eRc6l35RQQyEDMG9BcBkipY,1020
 flock/webapp/app/services/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -542,6 +542,7 @@ flock/webapp/templates/partials/_settings_view.html,sha256=f2h9jnDv8-JRkDzsbk_1o
 flock/webapp/templates/partials/_share_chat_link_snippet.html,sha256=N83lNAbkZiDfzZYviKwURPGGErSZhRlxnNzUqXsB7lE,793
 flock/webapp/templates/partials/_share_link_snippet.html,sha256=6en9lOdtu8FwVbtmkJzSQpHQ1WFXHnCbe84FDgAEF3U,1533
 flock/webapp/templates/partials/_sidebar.html,sha256=yfhEcF3xKI5j1c3iq46mU8mmPvgyvCHXe6xT7vsE6KM,4984
+flock/webapp/templates/partials/_streaming_results_container.html,sha256=WVp_IafF1_4puyfs3ueIJ16ehZHiMDBADUXRoZJ1_Yo,6684
 flock/webapp/templates/partials/_structured_data_view.html,sha256=TEaXcMGba9ruxEc_MLxygIO1qWcuSTo1FnosFtGSKWI,2101
 flock/webapp/templates/partials/_theme_preview.html,sha256=THeMYTXzgzHJxzWqaTtUhmJyBZT3saLRAa6wzZa4qnk,1347
 flock/workflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -551,8 +552,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.0b21.dist-info/METADATA,sha256=UV0JZGY5epos-_zJXKP6msrOE9M21KBkIlDRVvbXzvk,9997
-flock_core-0.5.0b21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-flock_core-0.5.0b21.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
-flock_core-0.5.0b21.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
-flock_core-0.5.0b21.dist-info/RECORD,,
+flock_core-0.5.0b22.dist-info/METADATA,sha256=OQrcZ9gIQRqbbhfU0RSIbo7auCybTmemyee39C6XWHc,9997
+flock_core-0.5.0b22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+flock_core-0.5.0b22.dist-info/entry_points.txt,sha256=rWaS5KSpkTmWySURGFZk6PhbJ87TmvcFQDi2uzjlagQ,37
+flock_core-0.5.0b22.dist-info/licenses/LICENSE,sha256=iYEqWy0wjULzM9GAERaybP4LBiPeu7Z1NEliLUdJKSc,1072
+flock_core-0.5.0b22.dist-info/RECORD,,