From 4fa2b32046fa1bcfd1ac332a958dbc6852cd89c2 Mon Sep 17 00:00:00 2001
From: geekan <geekan@foxmail.com>
Date: Sun, 24 Dec 2023 17:23:59 +0800
Subject: [PATCH] refine setup process

---
 requirements-test.txt | 3 ---
 requirements.txt      | 2 --
 setup.py              | 2 ++
 3 files changed, 2 insertions(+), 5 deletions(-)
 delete mode 100644 requirements-test.txt

diff --git a/requirements-test.txt b/requirements-test.txt
deleted file mode 100644
index 0a34c35ea..000000000
--- a/requirements-test.txt
+++ /dev/null
@@ -1,3 +0,0 @@
--r requirements.txt
-pytest
-pytest-asyncio
\ No newline at end of file
diff --git a/requirements.txt b/requirements.txt
index 5144dc4a4..5cb01ab99 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -31,7 +31,6 @@ tenacity==8.2.2
 tiktoken==0.5.2
 tqdm==4.64.0
 #unstructured[local-inference]
-# playwright
 # selenium>4
 # webdriver_manager<3.9
 anthropic==0.3.6
@@ -58,6 +57,5 @@ gitignore-parser==0.1.9
 # connexion[swagger-ui]
 websockets~=12.0
 networkx~=3.2.1
-pylint~=3.0.3
 google-generativeai==0.3.1
 playwright==1.40.0
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 8ef2a6946..db326df71 100644
--- a/setup.py
+++ b/setup.py
@@ -49,6 +49,8 @@ setup(
         "search-ddg": ["duckduckgo-search==3.8.5"],
         "pyppeteer": ["pyppeteer>=1.0.2"],
         "ocr": ["paddlepaddle==2.4.2", "paddleocr>=2.0.1", "tabulate==0.9.0"],
+        "dev": ["pylint~=3.0.3", "black~=21.9b0", "isort~=5.9.3", "pre-commit~=2.15.0"],
+        "test": ["pytest", "pytest-cov", "pytest-asyncio", "pytest-mock"],
     },
     cmdclass={
         "install_mermaid": InstallMermaidCLI,