import yaml
from servc_agent.servc_send import process

with open("examples.yaml") as f:
    data = yaml.safe_load(f.read())

    for obj in data["examples"]:
        process(
            "agent-lake",
            "embeddings_insert",
            {
                "segment_id": data["segment"],
                "document_id": obj["id"],
                "embed_text": obj["inputfile"],
                "store_text": " -> ".join(
                    [f"input: {obj["inputfile"]}", obj["answer"]]
                ),
            },
            force=False,
            poll=False,
        )