chore: cleanup

phil65 · Dec 15, 2024 · 25a4490 · 25a4490
1 parent 98b5a28
commit 25a4490
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/src/llmling/prompts/models.py b/src/llmling/prompts/models.py
@@ -356,18 +356,16 @@ def messages(self) -> list[PromptMessage]:
             case "text":
                 # Simple text format - whole file as user message
                 msg = MessageContent(type="text", content=content)
-                return [PromptMessage(role="user", content=msg)]
             case "markdown":
                 # TODO: Parse markdown sections into separate messages
                 msg = MessageContent(type="text", content=content)
-                return [PromptMessage(role="user", content=msg)]
             case "jinja2":
                 # Raw template - will be formatted during format()
                 msg = MessageContent(type="text", content=content)
-                return [PromptMessage(role="user", content=msg)]
             case _:
                 msg = f"Unsupported format: {self.fmt}"
                 raise ValueError(msg)
+        return [PromptMessage(role="user", content=msg)]
 
     async def format(
         self, arguments: dict[str, Any] | None = None