added missing type
Browse files- 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")
|