(node, summary_token_threshold=200, model=None)
| 8 | from utils import * |
| 9 | |
| 10 | async def get_node_summary(node, summary_token_threshold=200, model=None): |
| 11 | node_text = node.get('text') |
| 12 | num_tokens = count_tokens(node_text, model=model) |
| 13 | if num_tokens < summary_token_threshold: |
| 14 | return node_text |
| 15 | else: |
| 16 | return await generate_node_summary(node, model=model) |
| 17 | |
| 18 | |
| 19 | async def generate_summaries_for_structure_md(structure, summary_token_threshold, model=None): |
no test coverage detected