From 3125441505f8edd10578c40cc29dd1ae92ea1e91 Mon Sep 17 00:00:00 2001 From: geekan Date: Fri, 29 Dec 2023 02:02:49 +0800 Subject: [PATCH] fix --- requirements.txt | 2 +- tests/metagpt/provider/test_fireworks_api.py | 7 ++++++- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index cab719f24..832b4c1c8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ aiohttp==3.8.4 #azure_storage==0.37.0 channels==4.0.0 -chromadb==0.4.21 +# chromadb # Django==4.1.5 # docx==0.2.4 #faiss==1.5.3 diff --git a/tests/metagpt/provider/test_fireworks_api.py b/tests/metagpt/provider/test_fireworks_api.py index ebedb8000..b7f728e73 100644 --- a/tests/metagpt/provider/test_fireworks_api.py +++ b/tests/metagpt/provider/test_fireworks_api.py @@ -23,7 +23,12 @@ default_resp = ChatCompletion( object="chat.completion", created=1703300855, choices=[ - Choice(finish_reason="stop", index=0, message=ChatCompletionMessage(role="assistant", content=resp_content)) + Choice( + finish_reason="stop", + logprobs=None, + index=0, + message=ChatCompletionMessage(role="assistant", content=resp_content), + ) ], usage=CompletionUsage(completion_tokens=110, prompt_tokens=92, total_tokens=202), )