AmbigNLG: チュートリアル

				
					from jinja2 import Template


class AdditionalInstructions(BaseModel):
    category: AmbiguityCategory
    additional_instructions: list[str]

def generate_additional_instructions(instruction: str, input_text: str, ambiguity_category: str, model: str = "gpt-4o-mini-2024-07-18"):
    template = Template("""To resolve the specified ambiguity in the instruction, provide multiple additional instructions as the infilled templates. Each additional instruction strictly adheres to the template format.
Ensure this added information aligns with the primary objective of the task, supports understanding of complex concepts, or aids in narrowing down the scope to generate more precise responses.

# Input Text:
{{ input_text }}

# Instruction:
{{ instruction }}

# Ambiguity to Resolve:
{{ ambiguity_category }}

# Template to Infill:
{% if ambiguity_category == 'Context' %}Additional context: <paragraph>
{% elif ambiguity_category == 'Keywords' %}Include <keywords> in your response.
{% elif ambiguity_category == 'Length' %}Answer with <number> words.
{% elif ambiguity_category == 'Planning' %}Please generate the output based on the following outline: 1. <topic1> 2. <topic2> ...
{% elif ambiguity_category == 'Style' %}Write in a <style> style.
{% elif ambiguity_category == 'Theme' %}Primarily discuss the following theme: <theme>
{% else %}No template found for this category.
{% endif %}""")

    prompt = template.render(
        input_text=input_text,
        instruction=instruction,
        ambiguity_category=ambiguity_category
    )
    response = client.beta.chat.completions.parse(
        model=model,
        messages=[{"role": "user", "content": prompt}],
        temperature=0,
        response_format=AdditionalInstructions,
    )

    return response.choices[0].message.parsed.additional_instructions


def get_user_selected_instruction(candidates: list[str]):
    print("Generated candidate instructions:")
    for idx, candidate in enumerate(candidates, start=1):
        print(f"{idx}: {candidate}")

    user_input = input(f"\nSelect an instruction (1-{len(candidates)}), or type 'skip' to skip, or provide your own: ")

    if user_input.isdigit():
        selection = int(user_input)
        if 1 <= selection <= len(candidates):
            print(f"Selected instruction: {candidates[selection - 1]}")
            return candidates[selection - 1]
    elif user_input.lower() == 'skip':
        print("Skipping instruction selection.")
        return None
    else:
        print(f"Custom instruction provided: {user_input}")
        return user_input

    return None


def refine_instructions(sample_instruction: str, sample_input: str, ambiguities: list[str]):
    additional_instructions = []

    for category in ambiguities:
        category = category.strip()
        print(f"\nSuggested additional instructions for {category}:")

        candidates = generate_additional_instructions(sample_instruction, sample_input, category)
        selected_instruction = get_user_selected_instruction(candidates)

        if selected_instruction:
            additional_instructions.append(selected_instruction)

    refined_instruction = "\n".join([sample_instruction] + additional_instructions)

    return refined_instruction

refined_instruction = refine_instructions(sample_instruction, sample_input, ambiguities)
print("\nFinal Refined Instruction:\n")
print(f"```{refined_instruction}```")

Suggested additional instructions for Length:
Generated candidate instructions:
1: Answer with 100 words.
2: Answer with 200 words.
3: Answer with 300 words.
Selected instruction: Answer with 100 words.

Suggested additional instructions for Context:
Generated candidate instructions:
1: Provide a brief overview of the history of sushi, highlighting its evolution from narezushi to modern forms like nigirizushi and conveyor belt sushi.
2: Summarize the different types of sushi mentioned in the article, including chirashizushi, inarizushi, makizushi, and their regional variations.
3: Explain the significance of sushi in Japanese culture, including its traditional preparation methods and etiquette associated with eating sushi.
Selected instruction: Explain the significance of sushi in Japanese culture, including its traditional preparation methods and etiquette associated with eating sushi.

Suggested additional instructions for Theme:
Generated candidate instructions:
1: Primarily discuss the following theme: The historical evolution of sushi from its origins to modern variations.
2: Primarily discuss the following theme: The different types of sushi and their unique characteristics.
3: Primarily discuss the following theme: The cultural significance of sushi in Japanese cuisine and its global influence.
4: Primarily discuss the following theme: The ingredients used in sushi and their preparation methods.
Skipping instruction selection.

Final Refined Instruction:

```Summarize this article!
Answer with 100 words.
Explain the significance of sushi in Japanese culture, including its traditional preparation methods and etiquette associated with eating sushi.```