File size: 1,915 Bytes
bf41ce7 7bca2ed 1fac9b0 7bca2ed 1fac9b0 bf41ce7 7bca2ed bf41ce7 7bca2ed bf41ce7 7bca2ed bf41ce7 7bca2ed bf41ce7 7bca2ed bf41ce7 7bca2ed bf41ce7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 | import re
from Brain.src.firebase.cloudmessage import CloudMessage
from Brain.src.firebase.firebase import initialize_app
from Brain.src.model.req_model import ReqModel
from Brain.src.rising_plugin.risingplugin import getCompletion, getTextFromImage
from Brain.src.rising_plugin.image_embedding import query_image_text
TEST_IAMGE_NAME = "0ddffe51-3763-48d9-ab74-2086de529217"
TEST_UUID = "TEST_UUID"
test_setting = ReqModel(
data={
"token": "test-token",
"uuid": TEST_UUID,
"openai_key": "",
"pinecone_key": "",
"pinecone_env": "",
"firebase_key": "",
"settings": {"temperature": 0.6},
}
)
firebase_app = initialize_app(test_setting)
def test_langchain():
error = "Error happened while analyzing your prompt. Please ask me again"
result = getCompletion(
query="open three.js website", setting=test_setting, firebase_app=firebase_app
)
print(result)
assert result != error
def test_image2text():
error = "Error happened while analyzing your prompt. Please ask me again"
result = getTextFromImage(filename=TEST_IAMGE_NAME, firebase_app=firebase_app)
print(result)
assert result != error
def test_firebase_cloud_message():
cloud_message = CloudMessage(firebase_app=firebase_app)
token_list = cloud_message.get_tokens()
notification = {"title": "alert", "content": "test"}
pattern = r"send to \d+ devices, with \d+ successed, with \d+ failed."
state, value = cloud_message.send_message(notification, token_list)
assert re.match(pattern, value)
def test_query_image_text():
error = "Error happened in querying image & text"
result = query_image_text(
image_content=getTextFromImage(
filename=TEST_IAMGE_NAME, firebase_app=firebase_app
),
message="",
setting=test_setting,
)
print(result)
assert result != error
|