From 42b3690e71703c570abe1935371299b2f5b415b9 Mon Sep 17 00:00:00 2001 From: Samuel Chan Date: Fri, 27 Dec 2024 12:01:50 +0800 Subject: [PATCH] Fix the async issue while running on Windows --- lightrag/kg/age_impl.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/lightrag/kg/age_impl.py b/lightrag/kg/age_impl.py index 5bd8198c..2a97bc37 100644 --- a/lightrag/kg/age_impl.py +++ b/lightrag/kg/age_impl.py @@ -1,7 +1,7 @@ import asyncio import inspect import json -import os +import os, sys from contextlib import asynccontextmanager from dataclasses import dataclass from typing import Any, Dict, List, NamedTuple, Optional, Tuple, Union @@ -20,6 +20,9 @@ from lightrag.utils import logger from ..base import BaseGraphStorage +if sys.platform.startswith("win"): + import asyncio.windows_events + asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy()) class AGEQueryException(Exception): """Exception for the AGE queries."""