From f18f484a87b5360b076756b5015f57b0c230a321 Mon Sep 17 00:00:00 2001
From: Saifeddine ALOUI <aloui.seifeddine@gmail.com>
Date: Mon, 20 Jan 2025 08:54:18 +0100
Subject: [PATCH] Update llm.py

---
 lightrag/llm.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/lightrag/llm.py b/lightrag/llm.py
index b6d7a2a8..daba06dd 100644
--- a/lightrag/llm.py
+++ b/lightrag/llm.py
@@ -350,7 +350,7 @@ async def ollama_model_if_cache(
     timeout = kwargs.pop("timeout", None)
     kwargs.pop("hashing_kv", None)
     api_key = kwargs.pop("api_key", None)
-    headers = {"Authorization": f"Bearer {api_key}"} if api_key else None
+    headers = {"Authorization": f"Bearer {api_key}"} if api_key else {}
     ollama_client = ollama.AsyncClient(host=host, timeout=timeout, headers=headers)
     messages = []
     if system_prompt:
@@ -383,7 +383,7 @@ async def lollms_model_if_cache(
 
     stream = True if kwargs.get("stream") else False
     api_key = kwargs.pop("api_key", None)
-    headers = {"Authorization": f"Bearer {api_key}"} if api_key else None
+    headers = {"Authorization": f"Bearer {api_key}"} if api_key else {}
 
     # Extract lollms specific parameters
     request_data = {