ragbits-core 0.16.0__py3-none-any.whl → 1.4.0.dev202512021005__py3-none-any.whl

ragbits/core/prompt/lab/app.py

@@ -1,262 +0,0 @@
-import asyncio
-from dataclasses import dataclass, field, replace
-from typing import Any, cast
-
-try:
-    import gradio as gr
-
-    HAS_GRADIO = True
-except ImportError:
-    HAS_GRADIO = False
-
-import jinja2
-from pydantic import BaseModel
-from rich.console import Console
-
-from ragbits.core.config import core_config
-from ragbits.core.llms import LLM
-from ragbits.core.llms.base import LLMType
-from ragbits.core.prompt import Prompt
-from ragbits.core.prompt.discovery import PromptDiscovery
-
-
-@dataclass(frozen=True)
-class PromptState:
-    """
-    Class to store the current state of the application.
-
-    This class holds various data structures used throughout the application's lifecycle.
-
-    Attributes:
-        prompts (list): A list containing discovered prompts.
-        rendered_prompt (Prompt): The most recently rendered Prompt instance.
-        llm (LLM): The LLM instance to be used for generating responses.
-    """
-
-    prompts: list = field(default_factory=list)
-    rendered_prompt: Prompt | None = None
-    llm: LLM | None = None
-
-
-def render_prompt(index: int, system_prompt: str, user_prompt: str, state: PromptState, *args: Any) -> PromptState:  # noqa: ANN401
-    """
-    Renders a prompt based on the provided key, system prompt, user prompt, and input variables.
-
-    This function constructs a Prompt object using the prompt constructor and input constructor
-    associated with the given key. It then updates the current prompt in the application state.
-
-    Args:
-        index (int): The index of the prompt to render in the prompts state.
-        system_prompt (str): The system prompt template for the prompt.
-        user_prompt (str): The user prompt template for the prompt.
-        state (PromptState): The application state object.
-        args (tuple): A tuple of input values for the prompt.
-
-    Returns:
-        PromptState: The updated application state object.
-    """
-    prompt_class = state.prompts[index]
-    prompt_class.system_prompt_template = jinja2.Template(system_prompt)
-    prompt_class.user_prompt_template = jinja2.Template(user_prompt)
-
-    input_type = prompt_class.input_type
-    input_fields = get_input_type_fields(input_type)
-    variables = {field["field_name"]: value for field, value in zip(input_fields, args, strict=False)}
-    input_data = input_type(**variables) if input_type is not None else None
-    prompt_object = prompt_class(input_data=input_data)
-    state = replace(state, rendered_prompt=prompt_object)
-
-    return state
-
-
-def list_prompt_choices(state: PromptState) -> list[tuple[str, int]]:
-    """
-    Returns a list of prompt choices based on the discovered prompts.
-
-    This function generates a list of tuples containing the names of discovered prompts and their
-    corresponding indices.
-
-    Args:
-        state (PromptState): The application state object.
-
-    Returns:
-        list[tuple[str, int]]: A list of tuples containing prompt names and their indices.
-    """
-    return [(prompt.__name__, idx) for idx, prompt in enumerate(state.prompts)]
-
-
-def send_prompt_to_llm(state: PromptState) -> str:
-    """
-    Sends the current prompt to the LLM and returns the response.
-
-    This function creates a LiteLLM client using the LLM model name and API key stored in the
-    application state. It then calls the LLM client to generate a response based on the current prompt.
-
-    Args:
-        state (PromptState): The application state object.
-
-    Returns:
-        str: The response generated by the LLM.
-
-    Raises:
-        ValueError: If the LLM model is not configured.
-    """
-    assert state.rendered_prompt is not None, "Prompt has not been rendered yet."  # noqa: S101
-
-    if state.llm is None:
-        raise ValueError("LLM model is not configured.")
-
-    try:
-        response = cast(str, asyncio.run(state.llm.generate_raw(prompt=state.rendered_prompt))["response"])
-    except Exception as e:  # pylint: disable=broad-except
-        response = str(e)
-
-    return response
-
-
-def get_input_type_fields(obj: BaseModel | None) -> list[dict]:
-    """
-    Retrieves the field names and default values from the input type of a prompt.
-
-    This function inspects the input type object associated with a prompt and extracts information
-    about its fields, including their names and default values.
-
-    Args:
-        obj (BaseModel): The input type object of the prompt.
-
-    Returns:
-        list[dict]: A list of dictionaries, each containing a field name and its default value.
-    """
-    if obj is None:
-        return []
-    return [
-        {"field_name": k, "field_default_value": v["schema"].get("default", None)}
-        for (k, v) in obj.__pydantic_core_schema__["schema"]["fields"].items()
-    ]
-
-
-def lab_app(  # pylint: disable=missing-param-doc
-    file_pattern: str = core_config.prompt_path_pattern,
-    llm_factory: str | None = core_config.llm_preference_factories[LLMType.TEXT],
-) -> None:
-    """
-    Launches the interactive application for listing, rendering, and testing prompts
-    defined within the current project.
-    """
-    if not HAS_GRADIO:
-        Console(stderr=True).print(
-            "To use Prompt Lab, you need the Gradio library. Please install it using the following command:\n"
-            r"[b]pip install ragbits-core\[lab][/b]"
-        )
-        return
-
-    prompts = PromptDiscovery(file_pattern=file_pattern).discover()
-
-    if not prompts:
-        Console(stderr=True).print(
-            f"""No prompts were found for the given file pattern: [b]{file_pattern}[/b].
-
-Please make sure that you are executing the command from the correct directory \
-or provide a custom file pattern using the [b]--file-pattern[/b] flag."""
-        )
-        return
-
-    with gr.Blocks() as gr_app:
-        prompts_state = gr.State(
-            PromptState(
-                prompts=list(prompts),
-                llm=LLM.subclass_from_factory(llm_factory) if llm_factory else None,
-            )
-        )
-
-        prompt_selection_dropdown = gr.Dropdown(
-            choices=list_prompt_choices(prompts_state.value),
-            value=0,
-            label="Select Prompt",
-        )
-
-        @gr.render(inputs=[prompt_selection_dropdown, prompts_state])
-        def show_split(index: int, state: gr.State) -> None:
-            prompt = state.prompts[index]
-            list_of_vars = []
-            with gr.Row():
-                with gr.Column(scale=1), gr.Tab("Inputs"):
-                    input_fields: list = get_input_type_fields(prompt.input_type)
-                    for entry in input_fields:
-                        with gr.Row():
-                            var = gr.Textbox(
-                                label=entry["field_name"],
-                                value=entry["field_default_value"],
-                                interactive=True,
-                            )
-                            list_of_vars.append(var)
-
-                    render_prompt_button = gr.Button(value="Render prompts")
-
-                with gr.Column(scale=4), gr.Tab("Prompt"):
-                    with gr.Row():
-                        with gr.Column():
-                            prompt_details_system_prompt = gr.Textbox(
-                                label="System Prompt",
-                                value=prompt.system_prompt,
-                                interactive=True,
-                            )
-
-                        with gr.Column():
-                            rendered_system_prompt = (
-                                state.rendered_prompt.rendered_system_prompt if state.rendered_prompt else ""
-                            )
-                            gr.Textbox(
-                                label="Rendered System Prompt",
-                                value=rendered_system_prompt,
-                                interactive=False,
-                            )
-
-                    with gr.Row():
-                        with gr.Column():
-                            prompt_details_user_prompt = gr.Textbox(
-                                label="User Prompt",
-                                value=prompt.user_prompt,
-                                interactive=True,
-                            )
-
-                        with gr.Column():
-                            rendered_user_prompt = (
-                                state.rendered_prompt.rendered_user_prompt if state.rendered_prompt else ""
-                            )
-                            gr.Textbox(
-                                label="Rendered User Prompt",
-                                value=rendered_user_prompt,
-                                interactive=False,
-                            )
-
-            llm_enabled = state.llm is not None
-            prompt_ready = state.rendered_prompt is not None
-            llm_request_button = gr.Button(
-                value="Send to LLM",
-                interactive=llm_enabled and prompt_ready,
-            )
-            gr.Markdown(
-                "To enable this button set an LLM factory function in CLI options or your pyproject.toml",
-                visible=not llm_enabled,
-            )
-            gr.Markdown(
-                "To enable this button, render a prompt first.",
-                visible=llm_enabled and not prompt_ready,
-            )
-            llm_prompt_response = gr.Textbox(lines=10, label="LLM response")
-
-            render_prompt_button.click(
-                render_prompt,
-                [
-                    prompt_selection_dropdown,
-                    prompt_details_system_prompt,
-                    prompt_details_user_prompt,
-                    prompts_state,
-                    *list_of_vars,
-                ],
-                [prompts_state],
-            )
-            llm_request_button.click(send_prompt_to_llm, prompts_state, llm_prompt_response)
-
-    gr_app.launch()

ragbits_core-0.16.0.dist-info/RECORD

@@ -1,72 +0,0 @@
-ragbits/core/__init__.py,sha256=j41mAvtD3DsKRDr3JZwOFMxtBv2Zq62oWN1G3BjqS-k,225
-ragbits/core/cli.py,sha256=dl6qCjT0VN4075Xf5CaxccavhPmuJIsDlpfLPdiEvao,489
-ragbits/core/config.py,sha256=jwhGQo5J6ZbPoIG-3Vx6OC9UWBkRsjDLaNKU5Vsjnyc,1832
-ragbits/core/options.py,sha256=TKdUaD2y-sCyQOJh6hxnxa70RYtsaE9rINJdAi4i_80,1666
-ragbits/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragbits/core/types.py,sha256=nmEhQFCwLv33g6Mj5ntXeo9_d6YZLdKf7Nk3hfzCYgE,757
-ragbits/core/audit/__init__.py,sha256=M_cUdFBhXNCWqP76RN9MIiuk-wFoQR65leRCnjbgBUM,4822
-ragbits/core/audit/base.py,sha256=wBPMOHY6jWBPTeekY9KcTru_lF2KM2-MRUZQHG0dS_E,13754
-ragbits/core/audit/cli.py,sha256=gChtMFC-AHEN0abRH3Jkjw2Ovf_hbB-vXvJXieaVF7g,9617
-ragbits/core/audit/otel.py,sha256=Iv09-UhkoodKMR1Z-qpegooH3ABN2LJYGd_qI72EZH4,2211
-ragbits/core/embeddings/__init__.py,sha256=LbVWIwkRBaMyLObmoZ2qxj8ooQk8tkonx4evb-YeUR8,451
-ragbits/core/embeddings/base.py,sha256=LZNTuWplpqUULRp48hF10YJp0bKzENSNGMgdWkUy1pk,2265
-ragbits/core/embeddings/exceptions.py,sha256=7DT-aT66Zd0l5K_HFvBmrs8_CuI4K46U4cQot0JQZcc,1222
-ragbits/core/embeddings/dense/__init__.py,sha256=gYtKXiGQ9LycnONWkxSuwlABzH4YDUF48q7gzR1fH-M,233
-ragbits/core/embeddings/dense/base.py,sha256=IYo-gX-QN-vy1Mbm-Tt2yW2uDOT5Uk2Pftyycbs_5Zc,1167
-ragbits/core/embeddings/dense/fastembed.py,sha256=dEgEnHrxUpROTkXDX5Eqddn_TE5ndpN4_5wUeZY4BNg,2426
-ragbits/core/embeddings/dense/litellm.py,sha256=JahshBZkpsoczwr9KPiY_L1KIco4pinD0QN3TXrXVvw,5763
-ragbits/core/embeddings/dense/local.py,sha256=Ct1YnBmAHZ0KUkKC1yj5Ul60FHL1945UgY5fgRLb_MQ,2562
-ragbits/core/embeddings/dense/noop.py,sha256=IrkFCobd6lvAFfxxSmT_uYOkwcSpvrNs3RNHhiRU4aY,4190
-ragbits/core/embeddings/dense/vertex_multimodal.py,sha256=Yb7mFHzxlHEh0FHO0aZbSPf2PqoHueDe5m4E-uUQ-zk,7293
-ragbits/core/embeddings/sparse/__init__.py,sha256=v04VqbGtr9TA3qpWJBtvTzGOlIbjadXdHe-Wv-w3q9w,280
-ragbits/core/embeddings/sparse/bag_of_tokens.py,sha256=5aK8N3V0kz3W_DtV4ilFLeO1OB9pJyzrB19sWIZ82DQ,2685
-ragbits/core/embeddings/sparse/base.py,sha256=GKt-U3Y5Tpn7DKH0zxjjhpBNHvpNgVtbxQqcUMYDJ6o,1242
-ragbits/core/embeddings/sparse/fastembed.py,sha256=8ruYkIiO0oL0Dvw68dAy8urlJnbRCbDNcptITrfb34w,2457
-ragbits/core/llms/__init__.py,sha256=mrpsrahvfXSvk1zarNd_Mf3xzfzma-CYI83_cY39oFk,169
-ragbits/core/llms/base.py,sha256=-5REsWIj5ztV2i_R1p07fFYLpijGMuAPvP_Sq5MuDoY,9352
-ragbits/core/llms/exceptions.py,sha256=aQkeiz-HireoYDjZhyLGxPcaE-Tc0A4jfsJnMX__qDM,1437
-ragbits/core/llms/factory.py,sha256=T-heeY_7Llp5bOl6u9ycKK8ZwFnUE1ry4Q4X-O68hEQ,1506
-ragbits/core/llms/litellm.py,sha256=DKBRzbLK16w8WPXqugwoGhtAwfiJrTUphc-ikdCaAbw,11976
-ragbits/core/llms/local.py,sha256=FK0NgFcqoA2k2V4HqGnpUyJHX6sUcITSHaK8agWaUig,5766
-ragbits/core/llms/mock.py,sha256=lRDgbdJUPugJIDOPqyCzMBpwln2sk3BE5uxO7hJjvuI,2410
-ragbits/core/prompt/__init__.py,sha256=FYbstvqrhW4g0FTwjfw-rKO_OjJPHTNQWFT1FE4wll8,94
-ragbits/core/prompt/_cli.py,sha256=TD7wrn8_RmVH6Q30FNgoSefHv9jXndg3qKhCbsVf-xg,3122
-ragbits/core/prompt/base.py,sha256=4jOgYj1HErvwO5SbSj_BZub5rY8Ys6X3Kivw3BQBatQ,2499
-ragbits/core/prompt/exceptions.py,sha256=-zOn5sbthLJWXrQ_1QBWFmb_046LL20TizdI58L5Dv4,558
-ragbits/core/prompt/parsers.py,sha256=UP8RDQN0fr0yWh9fdTTofm4ugX6rDJLrwChihPv3ig4,3005
-ragbits/core/prompt/prompt.py,sha256=2yPI84CN7mIv08b0HbtMzud7QiGZifxEAIqeRfN1_WU,14474
-ragbits/core/prompt/promptfoo.py,sha256=P7UQ8zF0e_jsb_D4lglL5xKZe9hmOz2S0LLPx9Su1oo,1768
-ragbits/core/prompt/discovery/__init__.py,sha256=0b_wbvKqHi--kRJ2BOuWLZ4QIpEQGow4Z2CuMrnaaHg,77
-ragbits/core/prompt/discovery/prompt_discovery.py,sha256=yWXdBWOelbkBP79pj72auHC6JZOVGz2Jcz0UVL1bwv0,2826
-ragbits/core/prompt/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-ragbits/core/prompt/lab/app.py,sha256=n9hgbRX8BX02kEUrjH8jLlYU-SbzXyRQ68p_NeDxJpc,9828
-ragbits/core/sources/__init__.py,sha256=c_zdvfk_JD1pVwEGcjWyPEJJ9_J5dsBmglTQL_3PTqU,590
-ragbits/core/sources/azure.py,sha256=C8tzC1PsuNfzLhWW5el19N_dqEE4Q_LND-m92jTBkIg,7512
-ragbits/core/sources/base.py,sha256=E_wHUr1Txrej9Q1KqLt2o7AE-sDMr459zl2mQykONJ4,6579
-ragbits/core/sources/exceptions.py,sha256=gv1d0kX304_kqLye6OfZ9szKTwaqLRszMYJCHRQuPRY,969
-ragbits/core/sources/gcs.py,sha256=8cLsB0CG3pEAN86jaxqXyutN_aUl5s8sLFXFGQ85KAM,4837
-ragbits/core/sources/git.py,sha256=Sf78EtKjwSLmJPt8ltj1MkvQRXsFAPBfN4wesTWCxoo,7608
-ragbits/core/sources/hf.py,sha256=HrxMUqEm-FYL-fi-nqCoiS2T_W9LZ0B9UPFtxCYKtSs,4042
-ragbits/core/sources/local.py,sha256=Nft4sBCsTtRv94LTGeZkkI0yqNRSB2MbfH315n-wdyU,2720
-ragbits/core/sources/s3.py,sha256=MWRlVa1XA_aQV5iJg6FG4uuEA74Kf5HENNaAXbXGC9c,7339
-ragbits/core/sources/web.py,sha256=daxifFHTBFK8QoCdtBWTWuoQ_D-LcQKoX2eHRje58DY,2717
-ragbits/core/utils/__init__.py,sha256=kn8ht1kb2aWZ0c6Z_bn4tOpQFfr-ctJxkpKBZkN7vZ8,66
-ragbits/core/utils/_pyproject.py,sha256=1I0VIDgUxsy4E7mNc09HVUrVDITSiytagIX6y0GfIYI,3608
-ragbits/core/utils/config_handling.py,sha256=Wz78cM9fM_ohm65mRas9bZyOnxt-5M8DrKPcF9Vzdzk,7841
-ragbits/core/utils/decorators.py,sha256=yZUwpI8cXLqNUEVMwN7JnlkvmhDvX4NNqgC6fKN-Pk4,1862
-ragbits/core/utils/dict_transformations.py,sha256=RK4ANrpXwXv3DjA1DcsDRYPHhNkt_StGxUJkLKW_psA,9103
-ragbits/core/utils/helpers.py,sha256=zTySnPGV8OEXHRYdLPtiOnHmazVFIPdJPSU2JlE_Dbs,349
-ragbits/core/utils/pydantic.py,sha256=Wlqn1Jt1kS2zNsPAfzaixw1XTOa_-FO_tsvE7r2pIr4,704
-ragbits/core/utils/secrets.py,sha256=Pjs6Tn1aBHY5vVerDkm9DBTgl8xawAuGPXX74xxs2Mw,1558
-ragbits/core/vector_stores/__init__.py,sha256=UUeJ2nat1lU3irP7svCx0R_1ffAlaWG4H-X88z0i-Dg,281
-ragbits/core/vector_stores/_cli.py,sha256=oI2JBp4HHtyCv85Ou5bZLBr12B2_0SjCQvrsR4KWdnk,3834
-ragbits/core/vector_stores/base.py,sha256=JQeEj6R8ngLbs0UI5YbuGMHmfJ5Q6097eQ1pNqbLQgg,10649
-ragbits/core/vector_stores/chroma.py,sha256=s96mP3XfhUPFxuQRUZDLHRYzk555QuFP1EleJvtqqB0,11319
-ragbits/core/vector_stores/hybrid.py,sha256=x8ak3jn6nHHokSIrdUzufZA1Jv6k7SRaEwR8K5ZyX0U,4085
-ragbits/core/vector_stores/hybrid_strategies.py,sha256=WreabPO5S7d1He-x8FjDJcFoIW77GIrdLeoYM-GwWrw,7554
-ragbits/core/vector_stores/in_memory.py,sha256=y74el3h6JV3uRfg8nu0w9Zseu64ejHfziViQFqWYxNM,5928
-ragbits/core/vector_stores/pgvector.py,sha256=xOPf51323oHTbxX-V1TwqYiiFsMas-qIiBrlAAndp-A,16805
-ragbits/core/vector_stores/qdrant.py,sha256=SejPISannqEqyMoz--mOrNk_To4QDCi4VtzL0TtKucA,12569
-ragbits_core-0.16.0.dist-info/METADATA,sha256=iwHBIzdkCKgybiL6lG6mSWVgZ5o1S43SJcVntLV7zgk,4289
-ragbits_core-0.16.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-ragbits_core-0.16.0.dist-info/RECORD,,