22 lines
660 B
Python
22 lines
660 B
Python
|
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()
|