add chat_model test script

pyproject.toml

@@ -11,6 +11,10 @@ packages = [
     { include = "voice_for_nature_backend", from = "src"}
 ]
 
+include = [
+    "data/*",
+]
+
 [tool.poetry.urls]
 
 [tool.poetry.dependencies]

src/voice_for_nature_backend/chat_model.py

@@ -0,0 +1,11 @@
+from typing import List
+from llama_index.core import SimpleDirectoryReader
+from llama_index.core.schema import Document
+from . import __path__
+
+
+def get_data(rdir: str = None) -> List[Document]:
+    if rdir is None:
+        rdir = f"{__path__[0]}/data"
+    documents = SimpleDirectoryReader(rdir).load_data()
+    return documents