YanSte commited on
Commit
664ae11
·
1 Parent(s): d523251

added missing type

Browse files
Files changed (1) hide show
  1. lightrag/operate.py +1 -1
lightrag/operate.py CHANGED
@@ -573,7 +573,7 @@ async def kg_query(
573
  global_config: dict[str, str],
574
  hashing_kv: BaseKVStorage | None = None,
575
  system_prompt: str | None = None,
576
- ) -> str:
577
  # Handle cache
578
  use_model_func = global_config["llm_model_func"]
579
  args_hash = compute_args_hash(query_param.mode, query, cache_type="query")
 
573
  global_config: dict[str, str],
574
  hashing_kv: BaseKVStorage | None = None,
575
  system_prompt: str | None = None,
576
+ ) -> str | AsyncIterator[str]:
577
  # Handle cache
578
  use_model_func = global_config["llm_model_func"]
579
  args_hash = compute_args_hash(query_param.mode, query, cache_type="query")