rerank_text2.py

# Format kb_responses into a list of sources
kb_responses_text = []
for kb_response in kb_responses:
    kb_responses_text.append(
        [
            kb_response['text']
        ]
    )

# Flatten
kb_responses_text = [item[0] for item in kb_responses_text]

# Construct body
body = json.dumps(
    {
        "query": flat_conversation,
        "documents": kb_responses_text,
        "top_n": rerank_number_of_results,
    }
)