fix: update CI workflow and fix tests for v1.1.1

Thordata · Thordata · commit 01e4bc90a15d · 2026-02-13T14:49:19.000+08:00
- Fix CI workflow: Update Python versions to 3.10+ (remove 3.9)
- Update import verification to include all 8 tools
- Fix test suite: Remove deprecated ThordataUniversalTool references
- Update tests to use new SDK API (serp_search_advanced, universal.scrape)
- Add tests for ThordataMarkdownTool and ThordataAccountTool
- Update version to 1.1.1
- Fix CHANGELOG dates
diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
@@ -12,7 +12,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        python-version: ["3.9", "3.10", "3.11", "3.12"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
       - uses: actions/checkout@v4
@@ -41,7 +41,7 @@ jobs:
 
       - name: Verify imports
         run: |
-          python -c "from thordata_langchain_tools import ThordataSerpTool, ThordataScrapeTool, ThordataUniversalTool, ThordataProxyTool; print('✅ All imports OK')"
+          python -c "from thordata_langchain_tools import ThordataSerpTool, ThordataScrapeTool, ThordataMarkdownTool, ThordataBatchScrapeTool, ThordataBatchSerpTool, ThordataProxyTool, ThordataTaskTool, ThordataAccountTool; print('✅ All imports OK')"
 
   build:
     runs-on: ubuntu-latest
@@ -64,4 +64,4 @@ jobs:
         run: python -m build
 
       - name: Check package
-        run: twine check dist/*
+        run: twine check dist/*
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,12 @@
 # Changelog
 
+## [1.1.1] - 2026-02-13
+
+### Fixed
+- Fixed CI workflow: Updated Python versions (3.10+), fixed import verification
+- Updated test suite to match current tool structure
+- Removed references to deprecated ThordataUniversalTool
+
 ## [1.1.0] - 2026-02-13
 
 ### Added
diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "thordata-langchain-tools"
-version = "1.1.0"
+version = "1.1.1"
 description = "LangChain tools powered by Thordata - SERP API, Web Scraping, and Proxy Network."
 readme = "README.md"
 requires-python = ">=3.10"
@@ -63,7 +63,7 @@ include = ["thordata_langchain_tools*"]
 
 [tool.black]
 line-length = 88
-target-version = ['py39', 'py310', 'py311', 'py312']
+target-version = ['py310', 'py311', 'py312']
 
 [tool.ruff]
 line-length = 88
diff --git a/test/test_tools.py b/test/test_tools.py
@@ -7,8 +7,9 @@
 from thordata_langchain_tools import (
     ThordataSerpTool,
     ThordataScrapeTool,
-    ThordataUniversalTool,
+    ThordataMarkdownTool,
     ThordataProxyTool,
+    ThordataAccountTool,
 )
 
 
@@ -36,7 +37,7 @@ def test_run_success(self, mock_client_class):
         """Test successful SERP search."""
         # Setup mock
         mock_client = MagicMock()
-        mock_client.serp_search.return_value = {
+        mock_client.serp_search_advanced.return_value = {
             "organic": [{"title": "Test Result", "link": "https://example.com"}]
         }
         mock_client_class.return_value = mock_client
@@ -48,13 +49,13 @@ def test_run_success(self, mock_client_class):
         # Verify
         assert "organic" in result
         assert len(result["organic"]) == 1
-        mock_client.serp_search.assert_called_once()
+        mock_client.serp_search_advanced.assert_called_once()
 
     @patch("thordata_langchain_tools.serp_tool.ThordataClient")
     def test_run_error_handling(self, mock_client_class):
         """Test error handling in SERP search."""
         mock_client = MagicMock()
-        mock_client.serp_search.side_effect = Exception("API Error")
+        mock_client.serp_search_advanced.side_effect = Exception("API Error")
         mock_client_class.return_value = mock_client
 
         tool = ThordataSerpTool()
@@ -80,20 +81,20 @@ def test_tool_description(self):
     def test_run_success(self, mock_client_class):
         """Test successful scrape."""
         mock_client = MagicMock()
-        mock_client.universal_scrape.return_value = "<html><body>Test</body></html>"
+        mock_client.universal.scrape.return_value = "<html><body>Test</body></html>"
         mock_client_class.return_value = mock_client
 
         tool = ThordataScrapeTool()
         result = tool._run(url="https://example.com", js_render=False, max_length=1000)
 
         assert "html" in result.lower()
-        mock_client.universal_scrape.assert_called_once()
+        mock_client.universal.scrape.assert_called_once()
 
     @patch("thordata_langchain_tools.scrape_tool.ThordataClient")
     def test_run_truncates_long_content(self, mock_client_class):
         """Test that long content is truncated."""
         mock_client = MagicMock()
-        mock_client.universal_scrape.return_value = "x" * 100000
+        mock_client.universal.scrape.return_value = "x" * 100000
         mock_client_class.return_value = mock_client
 
         tool = ThordataScrapeTool()
@@ -103,46 +104,31 @@ def test_run_truncates_long_content(self, mock_client_class):
         assert "truncated" in result.lower()
 
 
-class TestThordataUniversalTool:
-    """Tests for ThordataUniversalTool."""
+class TestThordataMarkdownTool:
+    """Tests for ThordataMarkdownTool."""
 
     def test_tool_name(self):
         """Test tool has correct name."""
-        tool = ThordataUniversalTool()
-        assert tool.name == "thordata_universal_scrape"
+        tool = ThordataMarkdownTool()
+        assert tool.name == "thordata_fetch_markdown"
 
     def test_tool_description(self):
         """Test tool has description."""
-        tool = ThordataUniversalTool()
-        assert (
-            "scrape" in tool.description.lower()
-            or "scraping" in tool.description.lower()
-        )
-
-    @patch("thordata_langchain_tools.universal_tool.ThordataClient")
-    def test_run_with_js_render(self, mock_client_class):
-        """Test scrape with JS rendering."""
+        tool = ThordataMarkdownTool()
+        assert "markdown" in tool.description.lower()
+
+    @patch("thordata_langchain_tools.markdown_tool.ThordataClient")
+    def test_run_success(self, mock_client_class):
+        """Test successful markdown conversion."""
         mock_client = MagicMock()
-        mock_client.universal_scrape.return_value = "<html>Rendered</html>"
+        mock_client.universal.scrape.return_value = "<html><body>Test</body></html>"
         mock_client_class.return_value = mock_client
 
-        tool = ThordataUniversalTool()
-        result = tool._run(
-            url="https://example.com",
-            js_render=True,
-            output_format="html",
-            country="us",
-            wait_for=".content",
-        )
+        tool = ThordataMarkdownTool()
+        result = tool._run(url="https://example.com", js_render=False, max_chars=1000)
 
-        assert "html" in result.lower()
-        mock_client.universal_scrape.assert_called_once_with(
-            url="https://example.com",
-            js_render=True,
-            output_format="html",
-            country="us",
-            wait_for=".content",
-        )
+        assert isinstance(result, str)
+        assert len(result) > 0
 
 
 class TestThordataProxyTool:
@@ -173,3 +159,31 @@ def test_run_basic_request(self, mock_client_class):
         result = tool._run(url="https://httpbin.org/ip")
 
         assert "1.2.3.4" in result
+
+
+class TestThordataAccountTool:
+    """Tests for ThordataAccountTool."""
+
+    def test_tool_name(self):
+        """Test tool has correct name."""
+        tool = ThordataAccountTool()
+        assert tool.name == "thordata_account_info"
+
+    def test_tool_description(self):
+        """Test tool has description."""
+        tool = ThordataAccountTool()
+        assert "account" in tool.description.lower()
+
+    @patch("thordata_langchain_tools.account_tool.ThordataClient")
+    def test_run_success(self, mock_client_class):
+        """Test successful account info retrieval."""
+        mock_client = MagicMock()
+        mock_client.get_traffic_balance.return_value = 1000.0
+        mock_client.account.get_usage_statistics.return_value = {"total": 500}
+        mock_client_class.return_value = mock_client
+
+        tool = ThordataAccountTool()
+        result = tool._run()
+
+        assert "traffic_balance" in result
+        assert result["traffic_balance"] == 1000.0
diff --git a/thordata_langchain_tools/__init__.py b/thordata_langchain_tools/__init__.py
@@ -26,7 +26,7 @@
     >>> markdown = markdown_tool.invoke({"url": "https://example.com"})
 """
 
-__version__ = "1.1.0"
+__version__ = "1.1.1"
 
 from .serp_tool import ThordataSerpTool
 from .scrape_tool import ThordataScrapeTool
