docs: update LLM + K-means notebook to handle partial failures

docs: update LLM + K-means notebook to handle partial failures #2452