Update utils.py (#1128)

不区分文件扩展名的大小写。
2023-08-16 22:40:05 +08:00 · 2023-08-16 22:40:05 +08:00 · 7cd7ca240b
parent 75d6feb55b
commit 7cd7ca240b
1 changed files with 1 additions and 1 deletions
--- a/server/knowledge_base/utils.py
+++ b/server/knowledge_base/utils.py
@ -69,7 +69,7 @@ class KnowledgeFile:
    ):
        self.kb_name = knowledge_base_name
        self.filename = filename
-        self.ext = os.path.splitext(filename)[-1]
+        self.ext = os.path.splitext(filename)[-1].lower()
        if self.ext not in SUPPORTED_EXTS:
            raise ValueError(f"暂未支持的文件格式 {self.ext}")
        self.filepath = get_file_path(knowledge_base_name, filename)