import llama_index

test_requirements.txt

@@ -2,6 +2,7 @@
 # For testing
 pytest==7.2.1
 pytest-dotenv==0.5.2
+pytest-asyncio
 # TODO: remove gpt_index after migration
 https://github.com/jerryjliu/gpt_index/archive/master.zip
 

tests/test_github_reader.py

@@ -1,4 +1,4 @@
-from gpt_index import Document
+from llama_index import Document
 import httpx
 import pytest
 import asyncio