flowent 0.0.1 → 0.0.4

package/backend/src/flowent/routes/settings.py

@@ -0,0 +1,324 @@
+from __future__ import annotations
+
+import time
+from copy import deepcopy
+
+from fastapi import APIRouter, HTTPException
+from loguru import logger
+from pydantic import BaseModel, ConfigDict
+
+from flowent.access import initialize_live_access_signature, set_access_code
+from flowent.events import event_bus
+from flowent.settings import (
+    TelegramApprovedChat,
+    TelegramSettings,
+    get_settings,
+    save_settings,
+    serialize_provider,
+    serialize_role,
+    serialize_settings,
+    serialize_telegram_settings,
+)
+from flowent.settings_management import (
+    MISSING,
+    apply_resolved_settings_update,
+    resolve_settings_update,
+)
+
+router = APIRouter()
+
+
+@router.get("/api/settings/bootstrap")
+async def get_settings_bootstrap() -> dict[str, object]:
+    from flowent._version import __version__
+
+    settings = get_settings()
+    return {
+        "settings": serialize_settings(settings),
+        "providers": [serialize_provider(provider) for provider in settings.providers],
+        "roles": [serialize_role(role) for role in settings.roles],
+        "version": __version__,
+    }
+
+
+@router.get("/api/settings")
+async def get_settings_api() -> dict[str, object]:
+    settings = get_settings()
+    return serialize_settings(settings)
+
+
+class UpdateSettingsRequest(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+    access: dict[str, object] | None = None
+    assistant: dict[str, object] | None = None
+    event_log: dict[str, object] | None = None
+    leader: dict[str, object] | None = None
+    model: dict[str, object] | None = None
+    working_dir: str | None = None
+
+
+class UpdateTelegramSettingsRequest(BaseModel):
+    model_config = ConfigDict(extra="forbid")
+    bot_token: str | None = None
+
+
+@router.post("/api/settings")
+async def update_settings(req: UpdateSettingsRequest) -> dict[str, object]:
+    from flowent.graph_service import sync_assistant_role, sync_tab_leaders
+    from flowent.providers.gateway import gateway
+
+    source_settings = get_settings()
+    current = deepcopy(source_settings)
+    next_access_code: str | None = None
+    reauth_required = False
+
+    if req.access is not None:
+        raw_new_code = req.access.get("new_code", "")
+        raw_confirm_code = req.access.get("confirm_code", "")
+        if raw_new_code is not None and not isinstance(raw_new_code, str):
+            raise HTTPException(
+                status_code=400, detail="access.new_code must be a string"
+            )
+        if raw_confirm_code is not None and not isinstance(raw_confirm_code, str):
+            raise HTTPException(
+                status_code=400,
+                detail="access.confirm_code must be a string",
+            )
+        new_code = raw_new_code if isinstance(raw_new_code, str) else ""
+        confirm_code = raw_confirm_code if isinstance(raw_confirm_code, str) else ""
+        if new_code or confirm_code:
+            if not new_code.strip():
+                raise HTTPException(
+                    status_code=400,
+                    detail="access.new_code must not be empty",
+                )
+            if confirm_code != new_code:
+                raise HTTPException(
+                    status_code=400,
+                    detail="access.confirm_code must match access.new_code",
+                )
+            next_access_code = new_code
+
+    assistant_role_name: str | None = None
+    assistant_allow_network: object = MISSING
+    assistant_write_dirs: object = MISSING
+    if req.assistant is not None:
+        assistant_unknown_fields = sorted(
+            set(req.assistant) - {"role_name", "allow_network", "write_dirs"}
+        )
+        if assistant_unknown_fields:
+            raise HTTPException(
+                status_code=400,
+                detail=(
+                    "Unknown assistant fields: " + ", ".join(assistant_unknown_fields)
+                ),
+            )
+        if "allow_network" in req.assistant:
+            assistant_allow_network = req.assistant.get("allow_network")
+        if "write_dirs" in req.assistant:
+            assistant_write_dirs = req.assistant.get("write_dirs")
+        raw_role_name = req.assistant.get("role_name")
+        if isinstance(raw_role_name, str) and raw_role_name.strip():
+            assistant_role_name = raw_role_name
+
+    leader_role_name: str | None = None
+    if req.leader is not None:
+        raw_role_name = req.leader.get("role_name")
+        if isinstance(raw_role_name, str) and raw_role_name.strip():
+            leader_role_name = raw_role_name
+
+    timestamp_format: str | None = None
+    if req.event_log is not None:
+        raw_timestamp_format = req.event_log.get("timestamp_format")
+        if isinstance(raw_timestamp_format, str):
+            timestamp_format = raw_timestamp_format
+
+    active_provider_id: str | None = None
+    active_model: str | None = None
+    context_window_tokens: object = MISSING
+    input_image: object = MISSING
+    output_image: object = MISSING
+    timeout_ms: object = MISSING
+    retry_policy: object = MISSING
+    max_retries: object = MISSING
+    retry_initial_delay_seconds: object = MISSING
+    retry_max_delay_seconds: object = MISSING
+    retry_backoff_cap_retries: object = MISSING
+    auto_compact_token_limit: object = MISSING
+    model_params: object = MISSING
+    if req.model is not None:
+        raw_active_provider_id = req.model.get("active_provider_id")
+        if isinstance(raw_active_provider_id, str):
+            active_provider_id = raw_active_provider_id
+        raw_active_model = req.model.get("active_model")
+        if isinstance(raw_active_model, str):
+            active_model = raw_active_model
+        if "context_window_tokens" in req.model:
+            context_window_tokens = req.model.get("context_window_tokens")
+        if "input_image" in req.model:
+            input_image = req.model.get("input_image")
+        if "output_image" in req.model:
+            output_image = req.model.get("output_image")
+        if "timeout_ms" in req.model:
+            timeout_ms = req.model.get("timeout_ms")
+        if "retry_policy" in req.model:
+            retry_policy = req.model.get("retry_policy")
+        if "max_retries" in req.model:
+            max_retries = req.model.get("max_retries")
+        if "retry_initial_delay_seconds" in req.model:
+            retry_initial_delay_seconds = req.model.get("retry_initial_delay_seconds")
+        if "retry_max_delay_seconds" in req.model:
+            retry_max_delay_seconds = req.model.get("retry_max_delay_seconds")
+        if "retry_backoff_cap_retries" in req.model:
+            retry_backoff_cap_retries = req.model.get("retry_backoff_cap_retries")
+        if "auto_compact_token_limit" in req.model:
+            auto_compact_token_limit = req.model.get("auto_compact_token_limit")
+        if "params" in req.model:
+            model_params = req.model.get("params")
+
+    try:
+        resolved = resolve_settings_update(
+            current,
+            working_dir=req.working_dir,
+            assistant_role_name=assistant_role_name,
+            assistant_allow_network=assistant_allow_network,
+            assistant_write_dirs=assistant_write_dirs,
+            leader_role_name=leader_role_name,
+            active_provider_id=active_provider_id,
+            active_model=active_model,
+            context_window_tokens=context_window_tokens,
+            input_image=input_image,
+            output_image=output_image,
+            max_retries=max_retries,
+            retry_policy=retry_policy,
+            timeout_ms=timeout_ms,
+            retry_initial_delay_seconds=retry_initial_delay_seconds,
+            retry_max_delay_seconds=retry_max_delay_seconds,
+            retry_backoff_cap_retries=retry_backoff_cap_retries,
+            auto_compact_token_limit=auto_compact_token_limit,
+            model_params=model_params,
+            timestamp_format=timestamp_format,
+        )
+    except ValueError as exc:
+        raise HTTPException(status_code=400, detail=str(exc)) from exc
+
+    apply_resolved_settings_update(current, resolved)
+
+    if next_access_code is not None:
+        set_access_code(current, next_access_code)
+        reauth_required = True
+
+    save_settings(current)
+    source_settings.__dict__.clear()
+    source_settings.__dict__.update(deepcopy(current).__dict__)
+    initialize_live_access_signature()
+    try:
+        sync_assistant_role(reason="assistant settings updated")
+        sync_tab_leaders(reason="leader settings updated")
+        gateway.invalidate_cache()
+    except Exception:
+        logger.exception("Settings saved but runtime synchronization failed")
+    if reauth_required:
+        event_bus.close_all_connections(code=4001, reason="Access code rotated")
+    logger.info("Settings updated")
+    return {
+        "status": "saved",
+        "settings": serialize_settings(current),
+        "reauth_required": reauth_required,
+    }
+
+
+@router.get("/api/settings/telegram")
+async def get_telegram_settings() -> dict[str, object]:
+    settings = get_settings()
+    return serialize_telegram_settings(settings.telegram)
+
+
+@router.patch("/api/settings/telegram")
+async def update_telegram_settings(
+    req: UpdateTelegramSettingsRequest,
+) -> dict[str, object]:
+    from flowent.runtime import restart_telegram_channel
+
+    settings = get_settings()
+    previous_token = settings.telegram.bot_token
+
+    next_token = previous_token
+    if req.bot_token is not None:
+        next_token = req.bot_token.strip()
+
+    settings.telegram = TelegramSettings(
+        bot_token=next_token,
+        pending_chats=list(settings.telegram.pending_chats),
+        approved_chats=list(settings.telegram.approved_chats),
+    )
+    save_settings(settings)
+
+    if next_token != previous_token:
+        restart_telegram_channel()
+
+    logger.info("Telegram settings updated")
+    return {
+        "status": "saved",
+        "telegram": serialize_telegram_settings(settings.telegram),
+    }
+
+
+@router.post("/api/settings/telegram/approve/{chat_id}")
+async def approve_telegram_chat(chat_id: int) -> dict[str, object]:
+    settings = get_settings()
+    pending_chat = next(
+        (chat for chat in settings.telegram.pending_chats if chat.chat_id == chat_id),
+        None,
+    )
+    if pending_chat is None:
+        raise HTTPException(status_code=404, detail="Pending Telegram chat not found")
+
+    settings.telegram.pending_chats = [
+        chat for chat in settings.telegram.pending_chats if chat.chat_id != chat_id
+    ]
+    if not any(chat.chat_id == chat_id for chat in settings.telegram.approved_chats):
+        settings.telegram.approved_chats.append(
+            TelegramApprovedChat(
+                chat_id=pending_chat.chat_id,
+                username=pending_chat.username,
+                display_name=pending_chat.display_name,
+                approved_at=time.time(),
+            )
+        )
+    save_settings(settings)
+    logger.info("Telegram chat approved: {}", chat_id)
+    return {
+        "status": "approved",
+        "telegram": serialize_telegram_settings(settings.telegram),
+    }
+
+
+@router.delete("/api/settings/telegram/pending/{chat_id}")
+async def delete_pending_telegram_chat(chat_id: int) -> dict[str, object]:
+    settings = get_settings()
+    settings.telegram.pending_chats = [
+        chat for chat in settings.telegram.pending_chats if chat.chat_id != chat_id
+    ]
+    save_settings(settings)
+    logger.info("Pending Telegram chat removed: {}", chat_id)
+    return {
+        "status": "deleted",
+        "telegram": serialize_telegram_settings(settings.telegram),
+    }
+
+
+@router.delete("/api/settings/telegram/chat/{chat_id}")
+async def delete_telegram_chat(chat_id: int) -> dict[str, object]:
+    settings = get_settings()
+    settings.telegram.approved_chats = [
+        existing_chat
+        for existing_chat in settings.telegram.approved_chats
+        if existing_chat.chat_id != chat_id
+    ]
+    save_settings(settings)
+    logger.info("Telegram chat removed: {}", chat_id)
+    return {
+        "status": "deleted",
+        "telegram": serialize_telegram_settings(settings.telegram),
+    }

package/backend/src/flowent/routes/stats.py

@@ -0,0 +1,229 @@
+from __future__ import annotations
+
+import time
+
+from fastapi import APIRouter, HTTPException, Query
+
+from flowent.graph_service import is_tab_leader
+from flowent.mcp_service import mcp_service
+from flowent.registry import registry
+from flowent.settings import find_provider, find_role, get_settings
+from flowent.stats_service import stats_store
+from flowent.workspace_store import workspace_store
+
+router = APIRouter()
+
+RANGE_WINDOWS_SECONDS: dict[str, int] = {
+    "1h": 60 * 60,
+    "24h": 24 * 60 * 60,
+    "7d": 7 * 24 * 60 * 60,
+    "30d": 30 * 24 * 60 * 60,
+}
+
+
+def _resolve_range_window(range_value: str) -> int:
+    window = RANGE_WINDOWS_SECONDS.get(range_value)
+    if window is None:
+        raise HTTPException(
+            status_code=400,
+            detail="range must be one of: 1h, 24h, 7d, 30d",
+        )
+    return window
+
+
+def _build_node_label(
+    *,
+    name: str | None,
+    role_name: str | None,
+    node_type: str,
+    is_leader: bool,
+) -> str:
+    if name:
+        return name
+    if role_name:
+        return role_name
+    if node_type == "assistant":
+        return "Assistant"
+    if is_leader:
+        return "Leader"
+    return "Agent"
+
+
+def _serialize_workflow_snapshots() -> list[dict[str, object]]:
+    return [
+        {
+            "id": tab.id,
+            "title": tab.title,
+            "leader_id": tab.leader_id,
+            "created_at": tab.created_at,
+            "updated_at": tab.updated_at,
+        }
+        for tab in workspace_store.list_tabs()
+    ]
+
+
+def _resolve_current_model_source(role_name: str | None) -> dict[str, str | None]:
+    settings = get_settings()
+    provider_id = settings.model.active_provider_id
+    model = settings.model.active_model
+    role_cfg = find_role(settings, role_name) if role_name else None
+    if (
+        role_cfg is not None
+        and role_cfg.model is not None
+        and role_cfg.model.provider_id
+        and role_cfg.model.model
+    ):
+        provider_id = role_cfg.model.provider_id
+        model = role_cfg.model.model
+    if not provider_id:
+        return {
+            "provider_id": None,
+            "provider_name": None,
+            "provider_type": None,
+            "model": model or None,
+        }
+    provider = find_provider(settings, provider_id)
+    if provider is None:
+        return {
+            "provider_id": provider_id,
+            "provider_name": None,
+            "provider_type": None,
+            "model": model or None,
+        }
+    return {
+        "provider_id": provider.id,
+        "provider_name": provider.name,
+        "provider_type": provider.type,
+        "model": model or None,
+    }
+
+
+def _serialize_node_snapshots(
+    workflow_titles: dict[str, str],
+) -> list[dict[str, object]]:
+    nodes_by_id: dict[str, dict[str, object]] = {}
+
+    assistant = registry.get_assistant()
+    if assistant is not None:
+        model_source = _resolve_current_model_source(assistant.config.role_name)
+        nodes_by_id[assistant.uuid] = {
+            "id": assistant.uuid,
+            "label": _build_node_label(
+                name=assistant.config.name,
+                role_name=assistant.config.role_name,
+                node_type=assistant.config.node_type.value,
+                is_leader=False,
+            ),
+            "name": assistant.config.name,
+            "role_name": assistant.config.role_name,
+            "node_type": assistant.config.node_type.value,
+            "is_leader": False,
+            "state": assistant.state.value,
+            "workflow_id": None,
+            "workflow_title": None,
+            **model_source,
+        }
+
+    for record in workspace_store.list_node_records():
+        live = registry.get(record.id)
+        is_leader = is_tab_leader(node_id=record.id, tab_id=record.config.tab_id)
+        model_source = _resolve_current_model_source(record.config.role_name)
+        nodes_by_id[record.id] = {
+            "id": record.id,
+            "label": _build_node_label(
+                name=record.config.name,
+                role_name=record.config.role_name,
+                node_type=record.config.node_type.value,
+                is_leader=is_leader,
+            ),
+            "name": record.config.name,
+            "role_name": record.config.role_name,
+            "node_type": record.config.node_type.value,
+            "is_leader": is_leader,
+            "state": (live.state if live is not None else record.state).value,
+            "workflow_id": record.config.tab_id,
+            "workflow_title": (
+                workflow_titles.get(record.config.tab_id)
+                if record.config.tab_id is not None
+                else None
+            ),
+            **model_source,
+        }
+
+    for node in registry.get_all():
+        if node.uuid in nodes_by_id:
+            continue
+        is_leader = is_tab_leader(node_id=node.uuid, tab_id=node.config.tab_id)
+        model_source = _resolve_current_model_source(node.config.role_name)
+        nodes_by_id[node.uuid] = {
+            "id": node.uuid,
+            "label": _build_node_label(
+                name=node.config.name,
+                role_name=node.config.role_name,
+                node_type=node.config.node_type.value,
+                is_leader=is_leader,
+            ),
+            "name": node.config.name,
+            "role_name": node.config.role_name,
+            "node_type": node.config.node_type.value,
+            "is_leader": is_leader,
+            "state": node.state.value,
+            "workflow_id": node.config.tab_id,
+            "workflow_title": (
+                workflow_titles.get(node.config.tab_id)
+                if node.config.tab_id is not None
+                else None
+            ),
+            **model_source,
+        }
+
+    return list(nodes_by_id.values())
+
+
+def _serialize_stats_records(
+    records: list[dict[str, object]],
+) -> list[dict[str, object]]:
+    serialized: list[dict[str, object]] = []
+    for record in records:
+        item = dict(record)
+        if "tab_id" in item:
+            item["workflow_id"] = item.pop("tab_id")
+        if "tab_title" in item:
+            item["workflow_title"] = item.pop("tab_title")
+        serialized.append(item)
+    return serialized
+
+
+def _serialize_mcp_activity() -> list[dict[str, object]]:
+    serialized: list[dict[str, object]] = []
+    for activity in mcp_service.list_activities():
+        item = activity.serialize()
+        if "tab_id" in item:
+            item["workflow_id"] = item.pop("tab_id")
+        serialized.append(item)
+    return serialized
+
+
+@router.get("/api/stats")
+async def get_stats(range: str = Query("24h")) -> dict[str, object]:
+    window_seconds = _resolve_range_window(range)
+    now = time.time()
+    since = now - window_seconds
+    workflows = _serialize_workflow_snapshots()
+    workflow_titles: dict[str, str] = {}
+    for workflow in workflows:
+        workflow_id = workflow.get("id")
+        workflow_title = workflow.get("title")
+        if isinstance(workflow_id, str) and isinstance(workflow_title, str):
+            workflow_titles[workflow_id] = workflow_title
+
+    return {
+        "requested_at": now,
+        "range": range,
+        "workflows": workflows,
+        "nodes": _serialize_node_snapshots(workflow_titles),
+        "requests": _serialize_stats_records(stats_store.list_requests(since=since)),
+        "compacts": _serialize_stats_records(stats_store.list_compacts(since=since)),
+        "mcp_servers": mcp_service.list_server_payloads(),
+        "mcp_activity": _serialize_mcp_activity(),
+    }