feat: remove prompt lab (#549)

Co-authored-by: Michał Pstrąg <47692610+micpst@users.noreply.github.com>

Author: mhordynski

docs/how-to/prompts/promptfoo.md

@@ -5,7 +5,7 @@ specified in the [promptfoo documentation](https://www.promptfoo.dev/docs/instal
 configuration files for all the prompts discovered by our autodiscover mechanism by running the following command:
 
 ```bash
-ragbits prompts generate-promptfoo-configs
+ragbits prompts promptfoo
 ```
 
 This command will generate a YAML files in the directory specified by `--target-path` (`promptfooconfigs` by

docs/how-to/prompts/prompts_lab.md

@@ -110,9 +110,6 @@ ragbits prompts exec song_prompt:SongPrompt --payload '{"subject": "unicorns", "
 
 Remember to change `song_prompt` to the name of the module where the prompt is defined and adjust the values of the placeholders to your liking.
 
-!!! tip
-    Ragbits also comes with a built-in GUI tool called Prompts Lab that allows you to manage and interact with prompts in a more user-friendly way. To learn more about using Prompts Lab, see the how-to article [How to Manage Prompts using GUI with Prompts Lab](../how-to/prompts/prompts_lab.md).
-
 ## Conclusion
 You now know how to define a prompt in Ragbits and how to use it with Large Language Models. You've also learned to make the prompt dynamic by using Pydantic models and the Jinja2 templating language. To learn more about defining prompts, such as configuring the desired output format, refer to the how-to article [How to define and use Prompts in Ragbits](../how-to/prompts/use_prompting.md).
 

docs/how-to/prompts/prompts_lab_input.png

@@ -10,11 +10,10 @@ nav:
       - quickstart/quickstart1_prompts.md
       - quickstart/quickstart2_rag.md
   - How-to:
-      - Prompting:
-          - "Define and use prompts": how-to/prompts/use_prompting.md
-          - "Use images in prompts": how-to/prompts/use_images_in_prompts.md
-          - "Manage prompts using GUI": how-to/prompts/prompts_lab.md
-          - "Test prompts with promptfoo": how-to/prompts/promptfoo.md
+      - "Prompts":
+        - "Define and use prompts": how-to/prompts/use_prompting.md
+        - "Use images in prompts": how-to/prompts/use_images_in_prompts.md
+        - "Test prompts with promptfoo": how-to/prompts/promptfoo.md
       - LLMs:
         - "Interact with LLMs": how-to/llms/use_llms.md
         - "Use local or self-hosted LLMs": how-to/llms/use_local_llms.md

docs/quickstart/quickstart1_prompts.md

@@ -2,6 +2,7 @@
 
 ## Unreleased
 
+- Remove prompt lab (#549)
 - Add batched() helper method to utils (#555)
 - Rename DocumentMeta create_text_document_from_literal to from_literal (#561)
 - Adjust typing for DocumentSearch (#554)

mkdocs.yml

@@ -60,9 +60,6 @@ local = [
 fastembed = [
     "fastembed>=0.4.2,<1.0.0"
 ]
-lab = [
-    "gradio~=4.44.0,<5.0.0",
-]
 promptfoo = [
     "PyYAML>=6.0.2,<7.0.0",
 ]

packages/ragbits-core/CHANGELOG.md

@@ -1,4 +1,5 @@
 import asyncio
+import inspect
 import json
 from importlib import import_module
 from pathlib import Path
@@ -9,6 +10,7 @@
 from ragbits.cli import print_output
 from ragbits.core.config import core_config
 from ragbits.core.llms.base import LLM, LLMType
+from ragbits.core.prompt.discovery import PromptDiscovery
 from ragbits.core.prompt.prompt import ChatFormat, Prompt
 
 prompts_app = typer.Typer(no_args_is_help=True)
@@ -21,6 +23,16 @@ class LLMResponseCliOutput(BaseModel):
     answer: str | BaseModel | None = None
 
 
+class PromptInfo(BaseModel):
+    """Information about a Prompt class for CLI display"""
+
+    name: str
+    import_path: str
+    description: str
+    input_type: str
+    output_type: str
+
+
 def _render(prompt_path: str, payload: str | None) -> Prompt:
     module_stringified, object_stringified = prompt_path.split(":")
     prompt_cls = getattr(import_module(module_stringified), object_stringified)
@@ -34,23 +46,7 @@ def _render(prompt_path: str, payload: str | None) -> Prompt:
 
 
 @prompts_app.command()
-def lab(
-    file_pattern: str = core_config.prompt_path_pattern,
-    llm_factory: str = core_config.llm_preference_factories[LLMType.TEXT],
-) -> None:
-    """
-    Launches the interactive application for listing, rendering, and testing prompts
-    defined within the current project.
-
-    For more information, see the [Prompts Lab documentation](../how-to/prompts/prompts_lab.md).
-    """
-    from ragbits.core.prompt.lab.app import lab_app
-
-    lab_app(file_pattern=file_pattern, llm_factory=llm_factory)
-
-
-@prompts_app.command()
-def generate_promptfoo_configs(
+def promptfoo(
     file_pattern: str = core_config.prompt_path_pattern,
     root_path: Path = Path.cwd(),  # noqa: B008
     target_path: Path = Path("promptfooconfigs"),
@@ -65,6 +61,25 @@ def generate_promptfoo_configs(
     generate_configs(file_pattern=file_pattern, root_path=root_path, target_path=target_path)
 
 
+@prompts_app.command()
+def search(file_pattern: str = core_config.prompt_path_pattern, root_path: Path = Path.cwd()) -> None:  # noqa: B008
+    """
+    Lists all available prompts that can be used with the 'render' and 'exec' commands.
+    """
+    prompt_classes = PromptDiscovery(file_pattern=file_pattern, root_path=root_path).discover()
+    prompt_infos = [
+        PromptInfo(
+            name=prompt_cls.__name__,
+            import_path=f"{prompt_cls.__module__}:{prompt_cls.__name__}",
+            description=inspect.getdoc(prompt_cls) or "",
+            input_type=getattr(prompt_cls.input_type, "__name__", str(prompt_cls.input_type)),
+            output_type=getattr(prompt_cls.output_type, "__name__", str(prompt_cls.output_type)),
+        )
+        for prompt_cls in prompt_classes
+    ]
+    print_output(prompt_infos)
+
+
 @prompts_app.command()
 def render(prompt_path: str, payload: str | None = None) -> None:
     """