eval_data_transformation.py

eval_data_transformation.py

from promptflow import tool

def transform_data(data: list, question: str, answer: str) -> list:
    transformed = [
        {
            "role": "user",
            "content": question
        },
        {
            "role": "assistant",
            "content": answer,
            "context": {
                "citations": [
                    {
                        "id": item["metadata"]["chunk_id"],
                        "title": item["metadata"]["title"],
                        "content": item["text"]
                    }
                    for item in data
                ]
            }
        }
    ]
    return transformed

@tool
def my_python_tool(source_data: list, question: str, answer: str) -> list:
    return transform_data(source_data, question, answer)