From d25a5231751523694c5a4b820e3ce83a08659247 Mon Sep 17 00:00:00 2001 From: Saifeddine ALOUI Date: Wed, 19 Feb 2025 19:46:43 +0100 Subject: [PATCH] Update gremlin_impl.py --- lightrag/kg/gremlin_impl.py | 14 ++++++-------- 1 file changed, 6 insertions(+), 8 deletions(-) diff --git a/lightrag/kg/gremlin_impl.py b/lightrag/kg/gremlin_impl.py index 40a9f007..5d1ea47b 100644 --- a/lightrag/kg/gremlin_impl.py +++ b/lightrag/kg/gremlin_impl.py @@ -2,6 +2,7 @@ import asyncio import inspect import json import os +import pipmaster as pm from dataclasses import dataclass from typing import Any, Dict, List, final @@ -20,15 +21,12 @@ from lightrag.utils import logger from ..base import BaseGraphStorage -try: - from gremlin_python.driver import client, serializer - from gremlin_python.driver.aiohttp.transport import AiohttpTransport - from gremlin_python.driver.protocol import GremlinServerError -except ImportError as e: - raise ImportError( - "`gremlin` library is not installed. Please install it via pip: `pip install gremlin`." - ) from e +if not pm.is_installed("gremlinpython"): + pm.install("gremlinpython") +from gremlin_python.driver import client, serializer +from gremlin_python.driver.aiohttp.transport import AiohttpTransport +from gremlin_python.driver.protocol import GremlinServerError @final @dataclass