fix: azure openai stream response usage missing (#1998)

2024-01-11 17:34:58 +08:00
parent c9e4147b11
commit 5e97eb1840
3 changed files with 41 additions and 32 deletions
--- a/api/tests/integration_tests/model_runtime/azure_openai/test_llm.py
+++ b/api/tests/integration_tests/model_runtime/azure_openai/test_llm.py
@@ -190,7 +190,6 @@ def test_invoke_stream_chat_model(setup_openai_mock):
        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
        if chunk.delta.finish_reason is not None:
            assert chunk.delta.usage is not None
            assert chunk.delta.usage.completion_tokens > 0