Timothy J. Baek 8 months ago
parent
commit
7b330d1490
1 changed files with 1 additions and 2 deletions
  1. 1 2
      backend/open_webui/apps/rag/utils.py

+ 1 - 2
backend/open_webui/apps/rag/utils.py

@@ -474,7 +474,6 @@ from typing import Optional, Sequence
 
 from langchain_core.callbacks import Callbacks
 from langchain_core.documents import BaseDocumentCompressor, Document
-from langchain_core.pydantic_v1 import Extra
 
 
 class RerankCompressor(BaseDocumentCompressor):
@@ -484,7 +483,7 @@ class RerankCompressor(BaseDocumentCompressor):
     r_score: float
 
     class Config:
-        extra = Extra.forbid
+        extra = "forbid"
         arbitrary_types_allowed = True
 
     def compress_documents(