kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-02-18 16:55:50 +08:00
Code Issues Actions 9 Packages Projects Releases Wiki Activity

accept drift context in search base

Browse Source
This commit is contained in:
Julian Whiting 2024-10-08 11:53:48 -04:00
parent 573731ce55
commit 9c4619af43
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
graphrag/query/structured_search/base.py
View File

@ -14,6 +14,7 @@ import tiktoken
from graphrag.query.context_builder.builders import (
GlobalContextBuilder,
LocalContextBuilder,
DRIFTContextBuilder,
)
from graphrag.query.context_builder.conversation_history import (
ConversationHistory,
@ -40,7 +41,7 @@ class BaseSearch(ABC):
def __init__(
self,
llm: BaseLLM,
context_builder: GlobalContextBuilder | LocalContextBuilder,
context_builder: GlobalContextBuilder | LocalContextBuilder | DRIFTContextBuilder,
token_encoder: tiktoken.Encoding | None = None,
llm_params: dict[str, Any] | None = None,
context_builder_params: dict[str, Any] | None = None,