Fix/list handling

promptolution/llms/local_llm.py

@@ -84,9 +84,7 @@ def _get_response(self, prompts: List[str], system_prompts: List[str]) -> List[s
         with torch.no_grad():
             response = self.pipeline(inputs, pad_token_id=self.eos_token_id)
 
-        if len(response) != 1:
-            response = [r[0] if isinstance(r, list) else r for r in response]
-
+        response = [r[0] if isinstance(r, list) else r for r in response]
         response = [r["generated_text"] for r in response]
         return response
 

tests/llms/test_local_llm.py

@@ -70,6 +70,16 @@ def test_local_llm_get_response(mock_local_dependencies):
     assert responses[1] == "Mock response 2"
 
 
+def test_local_llm_get_response_nested_single(mock_local_dependencies):
+    """Regression for #73: pipeline can return [[{...}]] for a single prompt; must be flattened."""
+    local_llm = LocalLLM(model_id="gpt2", batch_size=1)
+    mock_local_dependencies["pipeline_obj"].return_value = [[{"generated_text": "Mock response"}]]
+
+    responses = local_llm._get_response(["Hello, world!"], system_prompts=["System prompt"])
+
+    assert responses == ["Mock response"]
+
+
 @pytest.mark.parametrize(
     "model_id",
     [

tutorials/llm_as_judge_tutorial.ipynb

@@ -152,7 +152,7 @@
    "cell_type": "markdown",
    "metadata": {},
    "source": [
-    "### Creating Inital Prompts"
+    "### Creating Initial Prompts"
    ]
   },
   {