return totalTokens;
    }

    /**
     * Sets the total number of tokens.
     *
     * @param totalTokens the total tokens count
     */
    public void setTotalTokens(final Integer totalTokens) {
        this.totalTokens = totalTokens;
    }

    /**
     * Gets the model name.
     *
     * @return the model name
     */
    public String getModel() {

        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());