Remove summary length check for entity relations
- Summary now determined by num_fragment
This commit is contained in:
@@ -121,8 +121,11 @@ async def _handle_entity_relation_summary(
|
|||||||
)
|
)
|
||||||
|
|
||||||
tokens = tokenizer.encode(description)
|
tokens = tokenizer.encode(description)
|
||||||
if len(tokens) < summary_max_tokens: # No need for summary
|
|
||||||
return description
|
### summarize is not determined here anymore (It's determined by num_fragment now)
|
||||||
|
# if len(tokens) < summary_max_tokens: # No need for summary
|
||||||
|
# return description
|
||||||
|
|
||||||
prompt_template = PROMPTS["summarize_entity_descriptions"]
|
prompt_template = PROMPTS["summarize_entity_descriptions"]
|
||||||
use_description = tokenizer.decode(tokens[:llm_max_tokens])
|
use_description = tokenizer.decode(tokens[:llm_max_tokens])
|
||||||
context_base = dict(
|
context_base = dict(
|
||||||
|
Reference in New Issue
Block a user