RYDE-WORK/aimingmed-ai
1
0
Fork 0
mirror of https://github.com/aimingmed/aimingmed-ai.git synced 2026-01-31 19:54:46 +08:00
Code Issues 6 Packages Projects Releases Wiki Activity

UPDATE

Browse Source
This commit is contained in:
leehk 2025-03-06 12:03:54 +08:00
parent 8d9ec8dc05
commit c76b7d0284
2 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
.github/workflows/app-testing.yml vendored
View File

@ -26,7 +26,6 @@ jobs:
with:
app-path: app/streamlit/Chatbot.py
ruff: true
skip-smoke: true
pytest-args: -v --junit-xml=test-results.xml
- if: always()
uses: pmeier/pytest-results-action@v0.6.0

30
app/streamlit/app_test.py
View File

@ -25,22 +25,22 @@ def create_chat_completion(response: str, role: str = "assistant") -> ChatComple
)
@patch("openai.resources.chat.Completions.create")
def test_Chatbot(openai_create):
at = AppTest.from_file("Chatbot.py").run()
assert not at.exception
at.chat_input[0].set_value("Do you know any jokes?").run()
assert at.info[0].value == "Please add your OpenAI API key to continue."
# @patch("openai.resources.chat.Completions.create")
# def test_Chatbot(openai_create):
# at = AppTest.from_file("Chatbot.py").run()
# assert not at.exception
# at.chat_input[0].set_value("Do you know any jokes?").run()
# assert at.info[0].value == "Please add your OpenAI API key to continue."
JOKE = "Why did the chicken cross the road? To get to the other side."
openai_create.return_value = create_chat_completion(JOKE)
at.text_input(key="chatbot_api_key").set_value("sk-...")
at.chat_input[0].set_value("Do you know any jokes?").run()
print(at)
assert at.chat_message[1].markdown[0].value == "Do you know any jokes?"
assert at.chat_message[2].markdown[0].value == JOKE
assert at.chat_message[2].avatar == "assistant"
assert not at.exception
# JOKE = "Why did the chicken cross the road? To get to the other side."
# openai_create.return_value = create_chat_completion(JOKE)
# at.text_input(key="chatbot_api_key").set_value("sk-...")
# at.chat_input[0].set_value("Do you know any jokes?").run()
# print(at)
# assert at.chat_message[1].markdown[0].value == "Do you know any jokes?"
# assert at.chat_message[2].markdown[0].value == JOKE
# assert at.chat_message[2].avatar == "assistant"
# assert not at.exception
@patch("langchain.llms.OpenAI.__call__")