langchain.chains.example_generator — 🦜🔗 LangChain 0.2.0

from typing import List

from langchain_core.language_models import BaseLanguageModel
from langchain_core.prompts.few_shot import FewShotPromptTemplate
from langchain_core.prompts.prompt import PromptTemplate

from langchain.chains.llm import LLMChain

TEST_GEN_TEMPLATE_SUFFIX = "Add another example."


[docs]def generate_example(
    examples: List[dict], llm: BaseLanguageModel, prompt_template: PromptTemplate
) -> str:
    """给定一个提示的示例列表，返回另一个示例。"""
    prompt = FewShotPromptTemplate(
        examples=examples,
        suffix=TEST_GEN_TEMPLATE_SUFFIX,
        input_variables=[],
        example_prompt=prompt_template,
    )
    chain = LLMChain(llm=llm, prompt=prompt)
    return chain.predict()

Source code for langchain.chains.example_generator