andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 08:38:30 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: test new approach

Browse Source
This commit is contained in:
Eduardo Chiarotti
2024-07-04 22:23:19 -03:00
parent 5cb5879d7c
commit 6b800b7fff
2 changed files with 15 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/crewai/agents/agent_builder/utilities/base_token_process.py
View File

@@ -8,7 +8,7 @@ class TokenProcess:
successful_requests: int = 0
def sum_prompt_tokens(self, tokens: int):
self.prompt_tokens = 10
self.prompt_tokens = self.prompt_tokens + tokens
self.total_tokens = self.total_tokens + tokens
def sum_completion_tokens(self, tokens: int):

26
src/crewai/crew.py
View File

@@ -133,6 +133,15 @@ class Crew(BaseModel):
default=False,
description="output_log_file",
)
total_usage_metrics: dict = Field(
description="Total usage metrics for the crew.",
default={
"total_tokens": 0,
"prompt_tokens": 0,
"completion_tokens": 0,
"successful_requests": 0,
},
)
@field_validator("id", mode="before")
@classmethod
@@ -595,25 +604,18 @@ class Crew(BaseModel):
def calculate_usage_metrics(self) -> Dict[str, int]:
"""Calculates and returns the usage metrics."""
total_usage_metrics = {
"total_tokens": 0,
"prompt_tokens": 5,
"completion_tokens": 0,
"successful_requests": 0,
}
for agent in self.agents:
if hasattr(agent, "_token_process"):
token_sum = agent._token_process.get_summary()
for key in total_usage_metrics:
total_usage_metrics[key] += token_sum.get(key, 0)
for key in self.total_usage_metrics:
self.total_usage_metrics[key] += token_sum.get(key, 0)
if self.manager_agent and hasattr(self.manager_agent, "_token_process"):
token_sum = self.manager_agent._token_process.get_summary()
for key in total_usage_metrics:
total_usage_metrics[key] += token_sum.get(key, 0)
for key in self.total_usage_metrics:
self.total_usage_metrics[key] += token_sum.get(key, 0)
return total_usage_metrics
return self.total_usage_metrics
def __repr__(self):
return f"Crew(id={self.id}, process={self.process}, number_of_agents={len(self.agents)}, number_of_tasks={len(self.tasks)})"