from __future__ import annotations

from typing import Any, AsyncGenerator

from app.services.agent_runtime.base import ChatRuntime, RuntimePreparedContext


class JarvisRuntimeAdapter(ChatRuntime):
    name = "jarvis"

    async def chat_stream(
        self,
        prepared: RuntimePreparedContext,
    ) -> AsyncGenerator[dict[str, Any], None]:
        raise NotImplementedError("Jarvis runtime is executed inside AgentService")

    async def chat_once(self, prepared: RuntimePreparedContext) -> tuple[str, str | None]:
        raise NotImplementedError("Jarvis runtime is executed inside AgentService")


jarvis_runtime_adapter = JarvisRuntimeAdapter()