return completionTokens;
    }

    /**
     * Sets the number of completion tokens.
     *
     * @param completionTokens the completion tokens count
     */
    public void setCompletionTokens(final Integer completionTokens) {
        this.completionTokens = completionTokens;
    }

    /**
     * Gets the total number of tokens.
     *
     * @return the total tokens count
     */

        final LlmChatResponse response = new LlmChatResponse();

        response.setContent("Response content");
        response.setFinishReason("stop");
        response.setPromptTokens(100);
        response.setCompletionTokens(50);
        response.setTotalTokens(150);
        response.setModel("gpt-4");

        assertEquals("Response content", response.getContent());
        assertEquals("stop", response.getFinishReason());