refactor: Remove unused code in large_language_model.py (#5433)
This commit is contained in:
parent
39c14ec7c1
commit
142dc0afd7
4 changed files with 0 additions and 48 deletions
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue