assertEquals("(memory)", source.getLocation());

        source = new StringSource("Hello World!", "LOCATION");
        assertEquals("LOCATION", source.getLocation());
    }

    @Test
    void testGetContent() {
        StringSource source = new StringSource(null);
        assertEquals("", source.getContent());

        source = new StringSource("Hello World!", "LOCATION");

        assertEquals(Integer.valueOf(150), response.getTotalTokens());
        assertEquals("gpt-4", response.getModel());
    }

    @Test
    public void test_setContent() {
        final LlmChatResponse response = new LlmChatResponse();
        response.setContent("Test content");
        assertEquals("Test content", response.getContent());