hz/ebiz-dify-ai
1
0
Fork 0
mirror of http://112.124.100.131/huang.ze/ebiz-dify-ai.git synced 2025-12-13 12:56:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Remove unused code in large_language_model.py (#5433)

Browse Source
This commit is contained in:
-LAN-
2024-06-20 16:20:40 +08:00
committed by GitHub
parent 39c14ec7c1
commit 142dc0afd7
4 changed files with 0 additions and 48 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/tests/integration_tests/model_runtime/cohere/test_llm.py
View File

@@ -136,12 +136,6 @@ def test_invoke_chat_model():
assert isinstance(result, LLMResult)
assert len(result.message.content) > 0
for chunk in model._llm_result_to_stream(result):
assert isinstance(chunk, LLMResultChunk)
assert isinstance(chunk.delta, LLMResultChunkDelta)
assert isinstance(chunk.delta.message, AssistantPromptMessage)
assert len(chunk.delta.message.content) > 0 if chunk.delta.finish_reason is None else True
def test_invoke_stream_chat_model():
model = CohereLargeLanguageModel()