Add test for CI (#3114)

### What problem does this PR solve?

Add test for CI

### Type of change

- [x] New Feature (non-breaking change which adds functionality)

Co-authored-by: liuhua <10215101452@stu.ecun.edu.cn>
This commit is contained in:
liuhua
2024-10-31 14:07:23 +08:00
committed by GitHub
parent 5590a823c6
commit 9aeb07d830
13 changed files with 286 additions and 172 deletions

View File

@@ -1,6 +1,7 @@
import os
from ragflow_sdk import RAGFlow, DataSet, Document, Chunk
HOST_ADDRESS = 'http://127.0.0.1:9380'
HOST_ADDRESS = os.getenv('HOST_ADDRESS', 'http://127.0.0.1:9380')
def test_upload_document_with_success(get_api_key_fixture):
@@ -66,7 +67,7 @@ def test_download_document_with_success(get_api_key_fixture):
# Check if the retrieved document is of type Document
if isinstance(doc, Document):
# Download the document content and save it to a file
with open("./ragflow.txt", "wb+") as file:
with open("ragflow.txt", "wb+") as file:
file.write(doc.download())
# Print the document object for debugging
print(doc)
@@ -144,7 +145,7 @@ def test_parse_and_cancel_document(get_api_key_fixture):
# Define the document name and path
name3 = 'westworld.pdf'
path = './test_data/westworld.pdf'
path = 'test_data/westworld.pdf'
# Create a document in the dataset using the file path
ds.upload_documents({"name": name3, "blob": open(path, "rb").read()})