From 3fd40f539bc577685dd0de27518a13d0eaa8d51a Mon Sep 17 00:00:00 2001 From: Alonso Guevara Date: Mon, 14 Oct 2024 19:54:58 -0600 Subject: [PATCH] Format --- graphrag/query/structured_search/drift_search/action.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/graphrag/query/structured_search/drift_search/action.py b/graphrag/query/structured_search/drift_search/action.py index eb820603..1e90a9a1 100644 --- a/graphrag/query/structured_search/drift_search/action.py +++ b/graphrag/query/structured_search/drift_search/action.py @@ -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: