RYDE-WORK/aimingmed-ai
1
0
Fork 0
mirror of https://github.com/aimingmed/aimingmed-ai.git synced 2026-02-08 08:13:20 +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_deepseek.ChatDeepSeek.__call__")
@patch("langchain_google_genai.ChatGoogleGenerativeAI.invoke") # @patch("langchain_google_genai.ChatGoogleGenerativeAI.invoke")
@patch("langchain_community.llms.moonshot.Moonshot.__call__") # @patch("langchain_community.llms.moonshot.Moonshot.__call__")
def test_Chatbot(moonshot_llm, gemini_llm, deepseek_llm): # def test_Chatbot(moonshot_llm, gemini_llm, deepseek_llm):
at = AppTest.from_file("Chatbot.py").run() # at = AppTest.from_file("Chatbot.py").run()
assert not at.exception # assert not at.exception
QUERY = "What is the best treatment for hypertension?" # QUERY = "What is the best treatment for hypertension?"
RESPONSE = "The best treatment for hypertension is..." # RESPONSE = "The best treatment for hypertension is..."
deepseek_llm.return_value.content = RESPONSE # deepseek_llm.return_value.content = RESPONSE
gemini_llm.return_value.content = RESPONSE # gemini_llm.return_value.content = RESPONSE
moonshot_llm.return_value = 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 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[1].markdown[0].value == QUERY
assert at.chat_message[2].markdown[0].value == RESPONSE # assert at.chat_message[2].markdown[0].value == RESPONSE
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__")