From 883d6b7cc7fd1df29717ab5858f194cfbf0e3246 Mon Sep 17 00:00:00 2001
From: "zhenjie.ye" <yezhenjie@outlook.de>
Date: Sat, 26 Oct 2024 16:12:10 +0800
Subject: [PATCH] Refactor code formatting in
 lightrag_api_openai_compatible_demo.py

---
 lightrag/lightrag.py |  4 +---
 lightrag/llm.py      |  4 +++-
 setup.py             | 31 ++++++++++++++++++++++++-------
 3 files changed, 28 insertions(+), 11 deletions(-)

diff --git a/lightrag/lightrag.py b/lightrag/lightrag.py
index 3004f5ed..b84e22ef 100644
--- a/lightrag/lightrag.py
+++ b/lightrag/lightrag.py
@@ -85,9 +85,7 @@ class LightRAG:
 
     # LLM
     llm_model_func: callable = gpt_4o_mini_complete  # hf_model_complete#
-    llm_model_name: str = (
-        "meta-llama/Llama-3.2-1B-Instruct"  #'meta-llama/Llama-3.2-1B'#'google/gemma-2-2b-it'
-    )
+    llm_model_name: str = "meta-llama/Llama-3.2-1B-Instruct"  #'meta-llama/Llama-3.2-1B'#'google/gemma-2-2b-it'
     llm_model_max_token_size: int = 32768
     llm_model_max_async: int = 16
 
diff --git a/lightrag/llm.py b/lightrag/llm.py
index bb0d6063..fd6b72d6 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -286,7 +286,9 @@ async def hf_model_if_cache(
     output = hf_model.generate(
         **input_ids, max_new_tokens=512, num_return_sequences=1, early_stopping=True
     )
-    response_text = hf_tokenizer.decode(output[0][len(inputs["input_ids"][0]):], skip_special_tokens=True)
+    response_text = hf_tokenizer.decode(
+        output[0][len(inputs["input_ids"][0]) :], skip_special_tokens=True
+    )
     if hashing_kv is not None:
         await hashing_kv.upsert({args_hash: {"return": response_text, "model": model}})
     return response_text
diff --git a/setup.py b/setup.py
index bdf49f02..1b1f65f0 100644
--- a/setup.py
+++ b/setup.py
@@ -1,6 +1,7 @@
 import setuptools
 from pathlib import Path
 
+
 # Reading the long description from README.md
 def read_long_description():
     try:
@@ -8,6 +9,7 @@ def read_long_description():
     except FileNotFoundError:
         return "A description of LightRAG is currently unavailable."
 
+
 # Retrieving metadata from __init__.py
 def retrieve_metadata():
     vars2find = ["__author__", "__version__", "__url__"]
@@ -17,18 +19,26 @@ def retrieve_metadata():
             for line in f.readlines():
                 for v in vars2find:
                     if line.startswith(v):
-                        line = line.replace(" ", "").replace('"', "").replace("'", "").strip()
+                        line = (
+                            line.replace(" ", "")
+                            .replace('"', "")
+                            .replace("'", "")
+                            .strip()
+                        )
                         vars2readme[v] = line.split("=")[1]
     except FileNotFoundError:
         raise FileNotFoundError("Metadata file './lightrag/__init__.py' not found.")
-    
+
     # Checking if all required variables are found
     missing_vars = [v for v in vars2find if v not in vars2readme]
     if missing_vars:
-        raise ValueError(f"Missing required metadata variables in __init__.py: {missing_vars}")
-    
+        raise ValueError(
+            f"Missing required metadata variables in __init__.py: {missing_vars}"
+        )
+
     return vars2readme
 
+
 # Reading dependencies from requirements.txt
 def read_requirements():
     deps = []
@@ -36,9 +46,12 @@ def read_requirements():
         with open("./requirements.txt") as f:
             deps = [line.strip() for line in f if line.strip()]
     except FileNotFoundError:
-        print("Warning: 'requirements.txt' not found. No dependencies will be installed.")
+        print(
+            "Warning: 'requirements.txt' not found. No dependencies will be installed."
+        )
     return deps
 
+
 metadata = retrieve_metadata()
 long_description = read_long_description()
 requirements = read_requirements()
@@ -51,7 +64,9 @@ setuptools.setup(
     description="LightRAG: Simple and Fast Retrieval-Augmented Generation",
     long_description=long_description,
     long_description_content_type="text/markdown",
-    packages=setuptools.find_packages(exclude=("tests*", "docs*")),  # Automatically find packages
+    packages=setuptools.find_packages(
+        exclude=("tests*", "docs*")
+    ),  # Automatically find packages
     classifiers=[
         "Development Status :: 4 - Beta",
         "Programming Language :: Python :: 3",
@@ -66,6 +81,8 @@ setuptools.setup(
     project_urls={  # Additional project metadata
         "Documentation": metadata.get("__url__", ""),
         "Source": metadata.get("__url__", ""),
-        "Tracker": f"{metadata.get('__url__', '')}/issues" if metadata.get("__url__") else ""
+        "Tracker": f"{metadata.get('__url__', '')}/issues"
+        if metadata.get("__url__")
+        else "",
     },
 )