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

Format

Browse Source
This commit is contained in:
Alonso Guevara 2024-10-14 19:54:58 -06:00
parent 8cafb55769
commit 3fd40f539b
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
graphrag/query/structured_search/drift_search/action.py
View File

@ -86,9 +86,9 @@ class DriftAction:
generated_tokens = 0
else:
generated_tokens = num_tokens(self.answer, search_engine.token_encoder)
self.metadata.update(
{"token_ct": search_result.prompt_tokens + generated_tokens}
)
self.metadata.update({
"token_ct": search_result.prompt_tokens + generated_tokens
})
self.follow_ups = response.pop("follow_up_queries", [])
if not self.follow_ups: