from haystack import Pipeline
from haystack.dataclasses import ChatRole, ChatMessage
from haystack.components.builders import AnswerBuilder

builder = AnswerBuilder()
pipeline = Pipeline()
pipeline.add_component(name="builder", instance=builder)

# Test the pipeline
result = pipeline.run(
    data={
        "builder": {
            "query": "What's the answer?",
            "replies": [
                ChatMessage(
                    role=ChatRole.ASSISTANT,
                    content="This is the answer from the LLM.",
                    meta={"stuff": "meta data that always comes from the generator"},
                    name=None
                )
            ],
            "meta": [
                {"my_meta": "The meta data I want to add to the answer"}
            ]
        }
    }
)
print(result)
# {'builder':
#     {
#         'answers': [
#             GeneratedAnswer(
#                 data='This is the answer from the LLM.',
#                 query="What's the answer?",
#                 documents=[],
#                 meta={'stuff': 'meta data that always comes from the generator'})  # <-- Should also include meta I passed
#         ]
# }}