Оценка Graph RAG с помощью RAGAS

Graph RAG с LangChain и Neo4j

Adam Cowley

Manager, Developer Education at Neo4j

Text-to-Cypher

 

Результаты добавляются в промпт и передаются в LLM

Векторный поиск

 

Отзыв оставлен пользователем с именем @emileifrem

Graph RAG с LangChain и Neo4j

Оценка Graph RAG

 

  1. Оценка сквозной генерации с помощью time
  2. Оценка использования токенов и затрат с помощью tiktoken
  3. Оценка качества вывода с помощью ragas
    • Context precision
    • Noise sensitivity

 

performance_tradeoffs.jpg

1 Изображение создано с помощью GPT-4o
Graph RAG с LangChain и Neo4j

Noise Sensitivity

  • Измеряет долю нерелевантной информации в извлечённых документах
  • Выше при использовании только векторного поиска
  • Возвращает оценку на основе релевантной информации
from ragas.metrics import NoiseSensitivity


metric = NoiseSensitivity(
    llm=evaluator_llm, 
    mode="irrelevant"
)
1 https://docs.ragas.io/en/stable/concepts/metrics/available_metrics/noise_sensitivity/
Graph RAG с LangChain и Neo4j

Context Precision

  • Измеряет долю релевантных фрагментов в извлечённых документах
  • Чем выше оценка, тем более релевантна извлечённая информация
from ragas.metrics import LLMContextPrecisionWithReference


metric = LLMContextPrecisionWithReference(
    llm=evaluator_llm, 
)
1 https://docs.ragas.io/en/stable/concepts/metrics/available_metrics/context_precision/#llm-based-context-precision
Graph RAG с LangChain и Neo4j

Context Precision

  • Измеряет долю релевантных фрагментов в извлечённых документах
  • Чем выше оценка, тем более релевантна извлечённая информация
from ragas.metrics import LLMContextPrecisionWithoutReference


metric = LLMContextPrecisionWithoutReference(
    llm=evaluator_llm, 
)
Graph RAG с LangChain и Neo4j

Структура результата Text-to-Cypher

cypher_result = {

    "user_input": "Who is Romeo's love?",

    "response": "Romeo loves Juliet",

    "retrieved_contexts": [
        {
            "source": "Romeo",
            "target": "Juliet",
            "relationship": "LOVES",
            "sentiment": 0.9837
        },
    ]

}
Graph RAG с LangChain и Neo4j

Структура результата Text-to-Cypher

cypher_result = {
    "user_input": "Who is Romeo's love?",
    "response": "Romeo loves Juliet",
    "retrieved_contexts": [
        json.dumps({
            "source": "Romeo",
            "target": "Juliet",
            "relationship": "LOVES",
            "sentiment": 0.9837
        }),
    ]
}
Graph RAG с LangChain и Neo4j

Структура результата только с векторным поиском

vector_result = { 
    "user_input": "Who is Romeo's love?",
    "response": "Romeo loves Juliet",
    "retrieved_contexts": [

        "But, soft! what light through yonder window breaks?..."
        "O, she doth teach the forches to burn bright!"

    ]
}
Graph RAG с LangChain и Neo4j

Структура гибридного результата

hybrid_result = {
    "user_input": "Who is Romeo's love?",
    "response": "Romeo loves Juliet",

    "retrieved_contexts": [

        json.dumps({
            "page_content": "But, soft! what light through yonder window breaks? ...",

            "metadata": {

              "act": 2, "scene": 2, "spoken_to": "Juliet"

            }

        }),

        # ...

    ]
}
Graph RAG с LangChain и Neo4j

Создание набора данных для оценки

cypher_result = {
    "user_input": "Who is Romeo's love?",
    "retrieved_contexts": [
        json.dumps({
            "source": "Romeo",
            "target": "Juliet",
            "relationship": "LOVES",
            "sentiment": 0.9837
        }),
    ]
}


cypher_dataset = EvaluationDataset.from_list([cypher_result])
Graph RAG с LangChain и Neo4j

Выбор LLM для оценки

from langchain.chat_models import init_chat_model

# Choose an LLM to perform the evaluation
llm = init_chat_model(
    "gpt-4o-mini",
    model_provider="openai",
    api_key="...",
    temperature=0
)


# Wrap in LangchainLLMWrapper
from ragas.llms import LangchainLLMWrapper
evaluator_llm = LangchainLLMWrapper(llm)
Graph RAG с LangChain и Neo4j

Оценка ответов

from ragas import evaluate, EvaluationDataset
from ragas.metrics import LLMContextPrecisionWithoutReference, NoiseSensitivity

cypher_scores = evaluate(

    dataset=cypher_dataset,

    metrics=[

        LLMContextPrecisionWithoutReference(llm=evaluator_llm),

        NoiseSensitivity(llm=evaluator_llm, mode="irrelevant")

    ]

)

{'llm_context_precision_without_reference': 1.0000,
'noise_sensitivity(mode=irrelevant)': 0.0000}
Graph RAG с LangChain и Neo4j

Давайте потренируемся!

Graph RAG с LangChain и Neo4j

Preparing Video For Download...