Questions on Query Behavior and Evaluation Metrics in Arize

class RerankerAttributes:
    """
    Attributes for a reranker
    """

    RERANKER_INPUT_DOCUMENTS = "reranker.input_documents"
    """
    List of documents as input to the reranker
    """
    RERANKER_OUTPUT_DOCUMENTS = "reranker.output_documents"
    """
    List of documents as output from the reranker
    """
    RERANKER_QUERY = "reranker.query"
    """
    Query string for the reranker
    """
    RERANKER_MODEL_NAME = "reranker.model_name"
    """
    Model name of the reranker
    """
    RERANKER_TOP_K = "reranker.top_k"
    """
    Top K parameter of the reranker
    """

query = SpanQuery().where(
    "span_kind" == "RERANKER",
).select(
    # input="reranker.query",
    model = "reranker.model_name",
).explode(
    "reranker.output_documents",
    reference = "document.content",
)
reranked_docs_df = px.active_session().query_spans(query)
reranked_docs_df

query = SpanQuery().where(
    "span_kind" == "RERANKER",
).select(
    reference = "reranker.output_documents",
)
reranked_docs_df = px.Client().query_spans(query)
print(type(reranked_docs_df.reference[0][0]))
print(reranked_docs_df.reference[0][0].keys())

query = SpanQuery().where(
    "span_kind" == "RERANKER",
).explode(
    "reranker.output_documents",
    reference = "document.content",
)
reranked_docs_df = px.active_session().query_spans(query)
reranked_docs_df