dify/api/core/callback_handler/entity/llm_message.py

from pydantic import BaseModel


class LLMMessage(BaseModel):
    prompt: str = ''
    prompt_tokens: int = 0
    completion: str = ''
    completion_tokens: int = 0