Class SimpleCodec

python/tests/test_codecs.py:31–47

Test codec that treats LLMRequest.content as an OpenAI-like payload.

class SimpleCodec(LlmCodec):
    """Test codec that treats LLMRequest.content as an OpenAI-like payload."""

    def decode(self, request):
        content = request.content
        messages = content.get("messages", [])
        return AnnotatedLLMRequest(
            messages,
            model=content.get("model"),
            params={"temperature": content.get("temperature")} if "temperature" in content else None,
        )

    def encode(self, annotated, original):
        content = {**original.content, "messages": annotated.messages}
        if annotated.model is not None:
            content["model"] = annotated.model
        return LLMRequest(original.headers, content)


class AlternateCodec(LlmCodec):