Merge pull request #1558 from zhaoweiguo/patch-1

Update openai_api.py
geekan · Oct 31, 2024 · b5bb4d7 · b5bb4d7
2 parents 21d9c5a + 0c0cfe0
commit b5bb4d7
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py
@@ -114,9 +114,6 @@ async def _achat_completion_stream(self, messages: list[dict], timeout=USE_CONFI
                 elif hasattr(chunk.choices[0], "usage"):
                     # The usage of some services is an attribute of chunk.choices[0], such as Moonshot
                     usage = CompletionUsage(**chunk.choices[0].usage)
-                elif "openrouter.ai" in self.config.base_url and chunk_has_usage:
-                    # due to it get token cost from api
-                    usage = chunk.usage
                 has_finished = True
 
         log_llm_stream("\n")