From 1f9234eee8c5fb258d4e133fa87dbff1e52f8716 Mon Sep 17 00:00:00 2001 From: better629 Date: Thu, 28 Dec 2023 09:34:51 +0800 Subject: [PATCH] fix client_kwargs due to previous PR delete sync client --- metagpt/provider/fireworks_api.py | 5 ++--- metagpt/provider/open_llm_api.py | 5 ++--- 2 files changed, 4 insertions(+), 6 deletions(-) diff --git a/metagpt/provider/fireworks_api.py b/metagpt/provider/fireworks_api.py index 5fe86fc1c..638b0703d 100644 --- a/metagpt/provider/fireworks_api.py +++ b/metagpt/provider/fireworks_api.py @@ -85,10 +85,9 @@ class FireworksLLM(OpenAILLM): self._init_client() self.model = self.config.fireworks_api_model # `self.model` should after `_make_client` to rewrite it - def _make_client_kwargs(self) -> (dict, dict): + def _make_client_kwargs(self) -> dict: kwargs = dict(api_key=self.config.fireworks_api_key, base_url=self.config.fireworks_api_base) - async_kwargs = kwargs.copy() - return kwargs, async_kwargs + return kwargs def _update_costs(self, usage: CompletionUsage): if self.config.calc_usage and usage: diff --git a/metagpt/provider/open_llm_api.py b/metagpt/provider/open_llm_api.py index 2893f5b30..976e95c57 100644 --- a/metagpt/provider/open_llm_api.py +++ b/metagpt/provider/open_llm_api.py @@ -48,10 +48,9 @@ class OpenLLMGPTAPI(OpenAILLM): self._init_client() self.model = self.config.open_llm_api_model # `self.model` should after `_make_client` to rewrite it - def _make_client_kwargs(self) -> (dict, dict): + def _make_client_kwargs(self) -> dict: kwargs = dict(api_key="sk-xxx", base_url=self.config.open_llm_api_base) - async_kwargs = kwargs.copy() - return kwargs, async_kwargs + return kwargs def _calc_usage(self, messages: list[dict], rsp: str) -> CompletionUsage: usage = CompletionUsage(prompt_tokens=0, completion_tokens=0, total_tokens=0)