mirror of
https://github.com/microsoft/graphrag.git
synced 2025-07-21 16:04:52 +00:00
29 lines
2.3 KiB
Markdown
29 lines
2.3 KiB
Markdown
|
---
|
||
|
|
title: Question Generation ❔
|
||
|
|
navtitle: Question Generation
|
||
|
|
tags: [post, orchestration]
|
||
|
|
layout: page
|
||
|
|
date: 2024-03-28
|
||
|
|
---
|
||
|
|
|
||
|
|
## Entity-based Question Generation
|
||
|
|
|
||
|
|
The [question generation](https://github.com/microsoft/graphrag/blob/main//graphrag/query/question_gen/) method combines structured data from the knowledge graph with unstructured data from the input documents to generate candidate questions related to specific entities.
|
||
|
|
|
||
|
|
## Methodology
|
||
|
|
Given a list of prior user questions, the question generation method uses the same context-building approach employed in [local search](1-local_search.md) to extract and prioritize relevant structured and unstructured data, including entities, relationships, covariates, community reports and raw text chunks. These data records are then fitted into a single LLM prompt to generate candidate follow-up questions that represent the most important or urgent information content or themes in the data.
|
||
|
|
|
||
|
|
## Configuration
|
||
|
|
|
||
|
|
Below are the key parameters of the [Question Generation class](https://github.com/microsoft/graphrag/blob/main//graphrag/query/question_gen/local_gen.py):
|
||
|
|
* `llm`: OpenAI model object to be used for response generation
|
||
|
|
* `context_builder`: [context builder](https://github.com/microsoft/graphrag/blob/main//graphrag/query/structured_search/local_search/mixed_context.py) object to be used for preparing context data from collections of knowledge model objects, using the same context builder class as in local search
|
||
|
|
* `system_prompt`: prompt template used to generate candidate questions. Default template can be found at [system_prompt](https://github.com/microsoft/graphrag/blob/main//graphrag/query/question_gen/system_prompt.py)
|
||
|
|
* `llm_params`: a dictionary of additional parameters (e.g., temperature, max_tokens) to be passed to the LLM call
|
||
|
|
* `context_builder_params`: a dictionary of additional parameters to be passed to the [`context_builder`](https://github.com/microsoft/graphrag/blob/main//graphrag/query/structured_search/local_search/mixed_context.py) object when building context for the question generation prompt
|
||
|
|
* `callbacks`: optional callback functions, can be used to provide custom event handlers for LLM's completion streaming events
|
||
|
|
|
||
|
|
## How to Use
|
||
|
|
|
||
|
|
An example of the question generation function can be found in the following [notebook](../notebooks/local_search_nb).
|