From 17916b84f696c28a8e2579da62b4cffb14a5c1ed Mon Sep 17 00:00:00 2001 From: Shashank Harinath <9397524+ShankHarinath@users.noreply.github.com> Date: Thu, 19 Oct 2023 00:52:53 -0700 Subject: [PATCH 1/2] Fix indentation --- metagpt/provider/openai_api.py | 26 +++++++++++++------------- 1 file changed, 13 insertions(+), 13 deletions(-) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 7e865f288..4c71e1077 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -111,19 +111,19 @@ class CostManager(metaclass=Singleton): return self.total_completion_tokens -def get_total_cost(self): - """ - Get the total cost of API calls. - - Returns: - float: The total cost of API calls. - """ - return self.total_cost - - -def get_costs(self) -> Costs: - """Get all costs""" - return Costs(self.total_prompt_tokens, self.total_completion_tokens, self.total_cost, self.total_budget) + def get_total_cost(self): + """ + Get the total cost of API calls. + + Returns: + float: The total cost of API calls. + """ + return self.total_cost + + + def get_costs(self) -> Costs: + """Get all costs""" + return Costs(self.total_prompt_tokens, self.total_completion_tokens, self.total_cost, self.total_budget) def log_and_reraise(retry_state): From 53fa9ef83844ac27a0c0c0d8a6d598428c183f94 Mon Sep 17 00:00:00 2001 From: Shashank Harinath Date: Thu, 19 Oct 2023 00:59:32 -0700 Subject: [PATCH 2/2] Rename file --- metagpt/provider/openai_api.py | 4 +--- metagpt/roles/__init__.py | 2 +- metagpt/roles/{seacher.py => searcher.py} | 2 +- 3 files changed, 3 insertions(+), 5 deletions(-) rename metagpt/roles/{seacher.py => searcher.py} (99%) diff --git a/metagpt/provider/openai_api.py b/metagpt/provider/openai_api.py index 4c71e1077..f0110b148 100644 --- a/metagpt/provider/openai_api.py +++ b/metagpt/provider/openai_api.py @@ -110,7 +110,6 @@ class CostManager(metaclass=Singleton): """ return self.total_completion_tokens - def get_total_cost(self): """ Get the total cost of API calls. @@ -119,8 +118,7 @@ class CostManager(metaclass=Singleton): float: The total cost of API calls. """ return self.total_cost - - + def get_costs(self) -> Costs: """Get all costs""" return Costs(self.total_prompt_tokens, self.total_completion_tokens, self.total_cost, self.total_budget) diff --git a/metagpt/roles/__init__.py b/metagpt/roles/__init__.py index 1768b786c..f033a5dfa 100644 --- a/metagpt/roles/__init__.py +++ b/metagpt/roles/__init__.py @@ -12,7 +12,7 @@ from metagpt.roles.project_manager import ProjectManager from metagpt.roles.product_manager import ProductManager from metagpt.roles.engineer import Engineer from metagpt.roles.qa_engineer import QaEngineer -from metagpt.roles.seacher import Searcher +from metagpt.roles.searcher import Searcher from metagpt.roles.sales import Sales from metagpt.roles.customer_service import CustomerService diff --git a/metagpt/roles/seacher.py b/metagpt/roles/searcher.py similarity index 99% rename from metagpt/roles/seacher.py rename to metagpt/roles/searcher.py index 0b6e089da..3a19f32f2 100644 --- a/metagpt/roles/seacher.py +++ b/metagpt/roles/searcher.py @@ -3,7 +3,7 @@ """ @Time : 2023/5/23 17:25 @Author : alexanderwu -@File : seacher.py +@File : searcher.py """ from metagpt.actions import ActionOutput, SearchAndSummarize from metagpt.logs import logger