From ac2b6af97e09b66022ff5a0c90a39c07cdea3b8c Mon Sep 17 00:00:00 2001 From: yangdx Date: Wed, 14 May 2025 10:57:05 +0800 Subject: [PATCH] Eliminate tenacity from dynamic import --- lightrag/llm/azure_openai.py | 2 -- lightrag/llm/bedrock.py | 2 -- lightrag/llm/hf.py | 4 ---- lightrag/llm/lmdeploy.py | 2 -- lightrag/llm/ollama.py | 3 --- 5 files changed, 13 deletions(-) diff --git a/lightrag/llm/azure_openai.py b/lightrag/llm/azure_openai.py index 3405d29e..1963b4fe 100644 --- a/lightrag/llm/azure_openai.py +++ b/lightrag/llm/azure_openai.py @@ -4,8 +4,6 @@ import pipmaster as pm # Pipmaster for dynamic library install # install specific modules if not pm.is_installed("openai"): pm.install("openai") -if not pm.is_installed("tenacity"): - pm.install("tenacity") from openai import ( AsyncAzureOpenAI, diff --git a/lightrag/llm/bedrock.py b/lightrag/llm/bedrock.py index 5f712603..d243983c 100644 --- a/lightrag/llm/bedrock.py +++ b/lightrag/llm/bedrock.py @@ -6,8 +6,6 @@ import pipmaster as pm # Pipmaster for dynamic library install if not pm.is_installed("aioboto3"): pm.install("aioboto3") -if not pm.is_installed("tenacity"): - pm.install("tenacity") import aioboto3 import numpy as np from tenacity import ( diff --git a/lightrag/llm/hf.py b/lightrag/llm/hf.py index 954a99b7..8a648ea9 100644 --- a/lightrag/llm/hf.py +++ b/lightrag/llm/hf.py @@ -9,12 +9,8 @@ if not pm.is_installed("transformers"): pm.install("transformers") if not pm.is_installed("torch"): pm.install("torch") -if not pm.is_installed("tenacity"): - pm.install("tenacity") if not pm.is_installed("numpy"): pm.install("numpy") -if not pm.is_installed("tenacity"): - pm.install("tenacity") from transformers import AutoTokenizer, AutoModelForCausalLM from tenacity import ( diff --git a/lightrag/llm/lmdeploy.py b/lightrag/llm/lmdeploy.py index acf51c23..50e60cd6 100644 --- a/lightrag/llm/lmdeploy.py +++ b/lightrag/llm/lmdeploy.py @@ -3,8 +3,6 @@ import pipmaster as pm # Pipmaster for dynamic library install # install specific modules if not pm.is_installed("lmdeploy"): pm.install("lmdeploy[all]") -if not pm.is_installed("tenacity"): - pm.install("tenacity") from lightrag.exceptions import ( APIConnectionError, diff --git a/lightrag/llm/ollama.py b/lightrag/llm/ollama.py index 3cf10511..c86aea1c 100644 --- a/lightrag/llm/ollama.py +++ b/lightrag/llm/ollama.py @@ -10,9 +10,6 @@ import pipmaster as pm # Pipmaster for dynamic library install # install specific modules if not pm.is_installed("ollama"): pm.install("ollama") -if not pm.is_installed("tenacity"): - pm.install("tenacity") - import ollama