wehub-resource-sync
27 lines
663 B
Python
27 lines
663 B
Python
from dataclasses import dataclass
|
|
from types import SimpleNamespace
|
|
from typing import Any, Iterable
|
|
|
|
|
|
@dataclass
|
|
class FakeStructuredContent:
|
|
payload: dict[str, Any]
|
|
|
|
def model_dump(self, mode: str = "json") -> dict[str, Any]:
|
|
return self.payload
|
|
|
|
|
|
def content_event(chunk: str) -> Any:
|
|
return SimpleNamespace(type="content", chunk=chunk)
|
|
|
|
|
|
class FakeLLMClient:
|
|
def __init__(self, events: Iterable[Any]):
|
|
self._events = list(events)
|
|
self.generate_calls: list[dict[str, Any]] = []
|
|
|
|
def generate(self, **kwargs: Any):
|
|
self.generate_calls.append(kwargs)
|
|
for event in self._events:
|
|
yield event
|