mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-19 00:01:33 +00:00
5fc57e7f4e
- Add BaseClient, BaseFileStore, BaseFileWatcher, BaseJob, BaseService, and BaseStep classes - Move component initialization logic from ApplicationContext to Application class - Add logo printing and logging initialization in Application startup - Create client module with base client implementation - Add file store base class with embedding resolution and validation - Implement file watcher base class with watchfiles integration - Add job base class for sequential step execution orchestration - Create service base class for job exposure mechanisms - Refactor BaseStep with LLM workflow execution capabilities - Add case converter utility for naming convention transformations - Update import structure and module organization - Add proper type hints and docstrings across all components - Implement component registry integration for dynamic loading - Add error handling for missing backend configurations
90 lines
3.3 KiB
Python
90 lines
3.3 KiB
Python
"""Runtime context for managing response states and asynchronous data streaming."""
|
|
|
|
import asyncio
|
|
|
|
from .application_context import ApplicationContext
|
|
from ..enumeration import ChunkEnum
|
|
from ..schema import Response, StreamChunk
|
|
|
|
|
|
class RuntimeContext:
|
|
"""Context for execution state, response metadata, and stream queues."""
|
|
|
|
def __init__(self, **kwargs):
|
|
"""Initialize the context with all keyword arguments stored in data."""
|
|
self.data: dict = kwargs
|
|
|
|
@property
|
|
def response(self) -> Response:
|
|
"""Get or create the response object."""
|
|
return self.data.setdefault("response", Response())
|
|
|
|
@property
|
|
def stream_queue(self) -> asyncio.Queue:
|
|
"""Get the stream queue."""
|
|
return self.data["stream_queue"]
|
|
|
|
@property
|
|
def application_context(self) -> ApplicationContext:
|
|
"""Get the application context."""
|
|
return self.data["application_context"]
|
|
|
|
@classmethod
|
|
def from_context(cls, context: "RuntimeContext | None" = None, **kwargs) -> "RuntimeContext":
|
|
"""Create a new context from an existing instance or keywords."""
|
|
if context is None:
|
|
return cls(**kwargs)
|
|
context.data.update(kwargs)
|
|
return context
|
|
|
|
async def _enqueue(self, chunk: StreamChunk) -> None:
|
|
"""Internal helper to put a chunk into the queue if it exists."""
|
|
if self.stream_queue:
|
|
await self.stream_queue.put(chunk)
|
|
|
|
async def add_stream_string(self, chunk: str, chunk_type: ChunkEnum) -> "RuntimeContext":
|
|
"""Enqueue a stream chunk from a raw string and type."""
|
|
await self._enqueue(StreamChunk(chunk_type=chunk_type, chunk=chunk))
|
|
return self
|
|
|
|
async def add_stream_chunk(self, stream_chunk: StreamChunk) -> "RuntimeContext":
|
|
"""Enqueue an existing stream chunk."""
|
|
await self._enqueue(stream_chunk)
|
|
return self
|
|
|
|
async def add_stream_done(self) -> "RuntimeContext":
|
|
"""Enqueue a termination chunk to signal the end of the stream."""
|
|
await self._enqueue(StreamChunk(chunk_type=ChunkEnum.DONE, chunk="", done=True))
|
|
return self
|
|
|
|
def add_response_error(self, e: Exception) -> "RuntimeContext":
|
|
"""Record an exception into the response object."""
|
|
self.response.success = False
|
|
self.response.answer = str(e)
|
|
return self
|
|
|
|
def apply_mapping(self, mapping: dict[str, str]) -> "RuntimeContext":
|
|
"""Copy internal values based on a source-to-target key map."""
|
|
if not mapping:
|
|
return self
|
|
|
|
for source, target in mapping.items():
|
|
if source in self.data:
|
|
self.data[target] = self.data[source]
|
|
return self
|
|
|
|
def validate_required_keys(
|
|
self,
|
|
required_keys: dict[str, bool],
|
|
context_name: str = "context",
|
|
) -> "RuntimeContext":
|
|
"""Ensure all required keys are present in the context.
|
|
|
|
Args:
|
|
required_keys: Dictionary mapping key names to boolean indicating if required
|
|
context_name: Name of the context for error messages (e.g., operator name)
|
|
"""
|
|
for key, is_required in required_keys.items():
|
|
if is_required and key not in self.data:
|
|
raise ValueError(f"{context_name}: missing required input '{key}'")
|
|
return self
|