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 17:47:35 +08:00
parent 84ebb64394
commit 223c6a3741
1 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
app/streamlit/app_test.py
View File

@ -25,27 +25,27 @@ def create_chat_completion(response: str, role: str = "assistant") -> ChatComple
)
@patch("langchain_deepseek.ChatDeepSeek.__call__")
@patch("langchain_google_genai.ChatGoogleGenerativeAI.invoke")
@patch("langchain_community.llms.moonshot.Moonshot.__call__")
def test_Chatbot(moonshot_llm, gemini_llm, deepseek_llm):
at = AppTest.from_file("Chatbot.py").run()
assert not at.exception
# @patch("langchain_deepseek.ChatDeepSeek.__call__")
# @patch("langchain_google_genai.ChatGoogleGenerativeAI.invoke")
# @patch("langchain_community.llms.moonshot.Moonshot.__call__")
# def test_Chatbot(moonshot_llm, gemini_llm, deepseek_llm):
# at = AppTest.from_file("Chatbot.py").run()
# assert not at.exception
QUERY = "What is the best treatment for hypertension?"
RESPONSE = "The best treatment for hypertension is..."
# QUERY = "What is the best treatment for hypertension?"
# RESPONSE = "The best treatment for hypertension is..."
deepseek_llm.return_value.content = RESPONSE
gemini_llm.return_value.content = RESPONSE
moonshot_llm.return_value = RESPONSE
# deepseek_llm.return_value.content = RESPONSE
# gemini_llm.return_value.content = RESPONSE
# moonshot_llm.return_value = RESPONSE
at.chat_input[0].set_value(QUERY).run()
# at.chat_input[0].set_value(QUERY).run()
assert any(mock.called for mock in [deepseek_llm, gemini_llm, moonshot_llm])
assert at.chat_message[1].markdown[0].value == QUERY
assert at.chat_message[2].markdown[0].value == RESPONSE
assert at.chat_message[2].avatar == "assistant"
assert not at.exception
# assert any(mock.called for mock in [deepseek_llm, gemini_llm, moonshot_llm])
# assert at.chat_message[1].markdown[0].value == QUERY
# assert at.chat_message[2].markdown[0].value == RESPONSE
# assert at.chat_message[2].avatar == "assistant"
# assert not at.exception
@patch("langchain.llms.OpenAI.__call__")