Class DeepXivAsyncClient

tests/test_retrieval_disabled_mode.py:204–237

    calls: list[dict] = []

    class DeepXivAsyncClient:
        def __init__(self, *args, **kwargs):
            pass

        async def __aenter__(self):
            return self

        async def __aexit__(self, exc_type, exc, tb):
            return None

        async def get(self, url, **kwargs):
            calls.append({'url': url, **kwargs})
            if str(url).endswith('/stats/usage'):
                return _json_response(200, {'daily_limit': 1000, 'usage_history': []}, url)
            return _json_response(
                200,
                {
                    'total_count': 1,
                    'result': [
                        {
                            'arxiv_id': '1706.03762',
                            'title': 'Attention Is All You Need',
                            'abstract': 'Transformer abstract',
                            'url': 'https://arxiv.org/pdf/1706.03762.pdf',
                            'date': '2017-06-12T17:57:34Z',
                            'citation_count': 100,
                            'categories': ['cs.CL'],
                            'authors': [{'name': 'Ashish Vaswani', 'orgs': ['Google']}],
                            'score': 0.99,
                        }
                    ],
                },
                url,
            )

    monkeypatch.setattr(paper_search_module.httpx, 'AsyncClient', DeepXivAsyncClient)