Test (example/llm): Refactor and add basic tests for the LLM entry-point

Test (example/llm): Refactor and add basic tests for the LLM entry-point #1338