feat: ET-1594: add support for retrieving all prompts by category

lab_gen/services/conversation/conversation.py

@@ -46,6 +46,9 @@ def __init__(self, app: FastAPI, examples: dict[str, list[str]], prompts: dict[s
         self.app = app
         self.examples = examples
         self.prompts = prompts
+        self.all_prompts = {}
+        for key, value in self.prompts.items():
+            self.all_prompts[key] = value.input_variables
 
     def create_chain(self, llm: BaseLanguageModel, prompt: ChatPromptTemplate) -> RunnableWithMessageHistory:
         """
@@ -291,9 +294,9 @@ def end(self, user_id: str, conversation_id: str) -> None:
             return history.clear()
         raise NoConversationError(conversation_id)
 
-    def get_prompts(self) -> dict[str, list[str]]:
+    def get_prompts(self, categories: str) -> dict[str, list[str]]:
         """Gets the example prompts configured for this service."""
-        return self.examples
+        return self.all_prompts if categories else self.examples
 
     def get_prompt(self, prompt_id: str) -> StringPromptTemplate:
         """Gets the prompt template for the given prompt ID."""

lab_gen/web/api/prompts/views.py

@@ -1,4 +1,4 @@
-from fastapi import Depends, HTTPException
+from fastapi import Depends, HTTPException, Query
 from fastapi.routing import APIRouter
 
 from lab_gen.services.conversation.conversation import ConversationService
@@ -13,13 +13,14 @@ async def read_prompts(
     *,
     api_key: bool = Depends(get_api_key),  # noqa: ARG001
     conversation: ConversationService = Depends(conversation_provider),  # noqa: B008
+    show: str | None = Query(default=None, description="The category of prompts to show."),
 ) -> dict[str, list[str]]:
     """Returns available prompts.
 
     Returns:
         Mapping of prompt names to prompt texts.
     """
-    return conversation.get_prompts()
+    return conversation.get_prompts(show)
 
 @router.get("/prompts/{prompt_id}")
 async def read_prompt(  # noqa: D417