andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-09 08:08:32 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

fix: use proper ArxivPaperTool import (#408)

Browse Source
This commit is contained in:
Lucas Gomide
2025-08-05 10:46:11 -03:00
committed by GitHub
parent 707c8583f4
commit 9220cfba28
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai_tools/tools/arxiv_paper_tool/arxiv_paper_tool_test.py
View File

@@ -3,7 +3,7 @@ import urllib.error
from unittest.mock import patch, MagicMock, mock_open from unittest.mock import patch, MagicMock, mock_open
from pathlib import Path from pathlib import Path
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
from crewai_tools.tools.arxiv_paper_tool import ArxivPaperTool from crewai_tools import ArxivPaperTool
@pytest.fixture @pytest.fixture
def tool(): def tool():
@@ -97,7 +97,7 @@ def test_invalid_xml_response(mock_urlopen, tool):
with pytest.raises(ET.ParseError): with pytest.raises(ET.ParseError):
tool.fetch_arxiv_data("quantum", 1) tool.fetch_arxiv_data("quantum", 1)
@patch.object(ArxivPaperTool, "fetch_arxiv_data") @patch.object(ArxivPaperTool, "fetch_arxiv_data")
def test_run_with_max_results(mock_fetch, tool): def test_run_with_max_results(mock_fetch, tool):
mock_fetch.return_value = [{ mock_fetch.return_value = [{