refactor: Refactor proxy LLM (#1064)

dbgpt/rag/knowledge/tests/test_csv.py

@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import MagicMock, mock_open, patch
 
+import pytest
+
 from dbgpt.rag.knowledge.csv import CSVKnowledge
 
 MOCK_CSV_DATA = "id,name,age\n1,John Doe,30\n2,Jane Smith,25\n3,Bob Johnson,40"

dbgpt/rag/knowledge/tests/test_docx.py

@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import MagicMock, patch
 
+import pytest
+
 from dbgpt.rag.knowledge.docx import DocxKnowledge
 
 

dbgpt/rag/knowledge/tests/test_html.py

@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import mock_open, patch
 
+import pytest
+
 from dbgpt.rag.knowledge.html import HTMLKnowledge
 
 MOCK_HTML_CONTENT = b"""

dbgpt/rag/knowledge/tests/test_markdown.py

@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import mock_open, patch
 
+import pytest
+
 from dbgpt.rag.knowledge.markdown import MarkdownKnowledge
 
 MOCK_MARKDOWN_DATA = """# Header 1

dbgpt/rag/knowledge/tests/test_pdf.py

@@ -1,5 +1,6 @@
+from unittest.mock import MagicMock, mock_open, patch
+
 import pytest
-from unittest.mock import MagicMock, patch, mock_open
 
 from dbgpt.rag.knowledge.pdf import PDFKnowledge
 

dbgpt/rag/knowledge/tests/test_txt.py

@@ -1,6 +1,7 @@
-import pytest
 from unittest.mock import mock_open, patch
 
+import pytest
+
 from dbgpt.rag.knowledge.txt import TXTKnowledge
 
 MOCK_TXT_CONTENT = b"Sample text content for testing.\nAnother line of text."