Skip to content

Few Shot Strategy

FewShot(llm_client=None, api_key=None, model=None, temperature=None, **kwargs)

Bases: BaseStrategy

Refines a prompt by following few shot prompt technique.

Source code in promptrefiner/base.py 
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
def __init__(
    self,
    llm_client=None,
    api_key=None,
    model=None,
    temperature=None,
    **kwargs,
):
    """
    Initializes with OpenAI API credentials (defaults to global config).
    """
    if llm_client:
        self.llm_client = llm_client
    else:
        config = load_config(api_key, model, temperature)
        self.llm_client = get_llm_client(
            config["api_key"], config["model"], temperature, **kwargs
        )