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
|
|
@ -156,12 +156,6 @@ def test_invoke_chat_model(setup_openai_mock):
|
|||
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
|
||||
|
||||
@pytest.mark.parametrize('setup_openai_mock', [['chat']], indirect=True)
|
||||
def test_invoke_chat_model_with_vision(setup_openai_mock):
|
||||
model = OpenAILargeLanguageModel()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue