From 326057deeb747bac4c0b36518407a9b89c5ca9e0 Mon Sep 17 00:00:00 2001
From: zrguo <zrguo.bupt@qq.com>
Date: Mon, 20 Jan 2025 17:28:40 +0800
Subject: [PATCH] fixed linting

---
 lightrag/llm.py | 18 ++++++++++--------
 1 file changed, 10 insertions(+), 8 deletions(-)

diff --git a/lightrag/llm.py b/lightrag/llm.py
index 5ba2cfd4..f1c21dab 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -350,10 +350,11 @@ async def ollama_model_if_cache(
     timeout = kwargs.pop("timeout", None)
     kwargs.pop("hashing_kv", None)
     api_key = kwargs.pop("api_key", None)
-    headers = {
-                "Content-Type": "application/json",
-                "Authorization": f"Bearer {api_key}"
-                } if api_key else {"Content-Type": "application/json"}
+    headers = (
+        {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
+        if api_key
+        else {"Content-Type": "application/json"}
+    )
     ollama_client = ollama.AsyncClient(host=host, timeout=timeout, headers=headers)
     messages = []
     if system_prompt:
@@ -386,10 +387,11 @@ async def lollms_model_if_cache(
 
     stream = True if kwargs.get("stream") else False
     api_key = kwargs.pop("api_key", None)
-    headers = {
-                "Content-Type": "application/json",
-                "Authorization": f"Bearer {api_key}"
-                } if api_key else {"Content-Type": "application/json"}
+    headers = (
+        {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
+        if api_key
+        else {"Content-Type": "application/json"}
+    )
 
     # Extract lollms specific parameters
     request_data = {