From db9b4dc841ae4030a776260eaf8a8a48c4915f7d Mon Sep 17 00:00:00 2001
From: yangdx <yangdx@znipower.com>
Date: Thu, 6 Feb 2025 01:00:49 +0800
Subject: [PATCH] Added environment variable loading with dotenv in Ollama API

---
 lightrag/api/ollama_api.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/lightrag/api/ollama_api.py b/lightrag/api/ollama_api.py
index e2637db0..23b09f55 100644
--- a/lightrag/api/ollama_api.py
+++ b/lightrag/api/ollama_api.py
@@ -11,6 +11,11 @@ from fastapi.responses import StreamingResponse
 import asyncio
 from ascii_colors import trace_exception
 from lightrag import LightRAG, QueryParam
+from dotenv import load_dotenv
+
+
+# Load environment variables
+load_dotenv()
 
 
 class OllamaServerInfos: