From 96fdb83abff69aa650a76cb847299c517caa7e1b Mon Sep 17 00:00:00 2001
From: Thuc Pham <51660321+thucpn@users.noreply.github.com>
Date: Tue, 28 May 2024 18:33:53 +0700
Subject: [PATCH] use logger warning
---
templates/types/streaming/fastapi/app/api/routers/chat.py | 4 +++-
1 file changed, 3 insertions(+), 1 deletion(-)
diff --git a/templates/types/streaming/fastapi/app/api/routers/chat.py b/templates/types/streaming/fastapi/app/api/routers/chat.py
index d3f2ce63..731a39a1 100644
--- a/templates/types/streaming/fastapi/app/api/routers/chat.py
+++ b/templates/types/streaming/fastapi/app/api/routers/chat.py
@@ -1,4 +1,5 @@
import os
+import logging
from pydantic import BaseModel
from typing import List, Any, Optional, Dict, Tuple
from fastapi import APIRouter, Depends, HTTPException, Request, status
@@ -50,7 +51,8 @@ class _SourceNodes(BaseModel):
file_name = metadata.get("file_name")
url_prefix = os.getenv("FILESERVER_URL_PREFIX")
if not url_prefix:
- print("Warning: FILESERVER_URL_PREFIX not set in environment variables")
+ logger = logging.getLogger("uvicorn")
+ logger.warning("Warning: FILESERVER_URL_PREFIX not set in environment variables")
if file_name and url_prefix:
url = f"{url_prefix}/data/{file_name}"
--
GitLab