1  from llama_index.core.schema import NodeWithScore, TextNode
 2  
 3  from mlflow.entities import Document
 4  
 5  
 6  def test_from_llama_index_node_with_score():
 7      text_node = TextNode(text="Hello", metadata={"key": "value"})
 8      node_with_score = NodeWithScore(node=text_node, score=0.5)
 9      document = Document.from_llama_index_node_with_score(node_with_score)
10      assert document.page_content == "Hello"
11      assert document.metadata == {"score": 0.5, "key": "value"}
12      assert document.id == node_with_score.node_id