Class LLMResponse

web/pgadmin/llm/models.py:158–187

Represents a response from an LLM.

@dataclass
class LLMResponse:
    """Represents a response from an LLM."""
    content: str
    tool_calls: list[ToolCall] = field(default_factory=list)
    stop_reason: StopReason = StopReason.END_TURN
    model: str = ''
    usage: Usage = field(default_factory=Usage)
    raw_response: Optional[Any] = None

    @property
    def has_tool_calls(self) -> bool:
        """Check if the response contains tool calls."""
        return len(self.tool_calls) > 0

    def to_message(self) -> Message:
        """Convert response to an assistant message."""
        return Message.assistant(
            content=self.content,
            tool_calls=self.tool_calls
        )

    def to_dict(self) -> dict:
        """Convert to dictionary representation."""
        return {
            'content': self.content,
            'tool_calls': [tc.to_dict() for tc in self.tool_calls],
            'stop_reason': self.stop_reason.value,
            'model': self.model,
            'usage': self.usage.to_dict()
        }


@dataclass