RYDE-WORK/aimingmed-ai
1
0
Fork 0
mirror of https://github.com/aimingmed/aimingmed-ai.git synced 2026-02-08 16:37:29 +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: with:
app-path: app/streamlit/Chatbot.py app-path: app/streamlit/Chatbot.py
ruff: true ruff: true
skip-smoke: true
pytest-args: -v --junit-xml=test-results.xml pytest-args: -v --junit-xml=test-results.xml
- if: always() - if: always()
uses: pmeier/pytest-results-action@v0.6.0 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") # @patch("openai.resources.chat.Completions.create")
def test_Chatbot(openai_create): # def test_Chatbot(openai_create):
at = AppTest.from_file("Chatbot.py").run() # at = AppTest.from_file("Chatbot.py").run()
assert not at.exception # assert not at.exception
at.chat_input[0].set_value("Do you know any jokes?").run() # 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." # 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." # JOKE = "Why did the chicken cross the road? To get to the other side."
openai_create.return_value = create_chat_completion(JOKE) # openai_create.return_value = create_chat_completion(JOKE)
at.text_input(key="chatbot_api_key").set_value("sk-...") # at.text_input(key="chatbot_api_key").set_value("sk-...")
at.chat_input[0].set_value("Do you know any jokes?").run() # at.chat_input[0].set_value("Do you know any jokes?").run()
print(at) # print(at)
assert at.chat_message[1].markdown[0].value == "Do you know any jokes?" # 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].markdown[0].value == JOKE
assert at.chat_message[2].avatar == "assistant" # assert at.chat_message[2].avatar == "assistant"
assert not at.exception # assert not at.exception
@patch("langchain.llms.OpenAI.__call__") @patch("langchain.llms.OpenAI.__call__")