andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-05 09:12:39 +00:00
Code Issues Actions 7 Packages Projects Releases Wiki Activity

fix: ruff format on llm.py and llm_result.py

Browse Source
This commit is contained in:
Joao Moura
2026-04-25 16:15:43 -07:00
parent c9a6955cd6
commit 54f5b7db2e
2 changed files with 26 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
lib/crewai/src/crewai/llm.py
View File

@@ -1827,8 +1827,12 @@ class LLM(BaseLLM):
"id": getattr(tc, "id", f"call_{i}"),
"type": "function",
"function": {
"name": getattr(tc.function, "name", "") if hasattr(tc, "function") else "",
"arguments": getattr(tc.function, "arguments", "{}") if hasattr(tc, "function") else "{}",
"name": getattr(tc.function, "name", "")
if hasattr(tc, "function")
else "",
"arguments": getattr(tc.function, "arguments", "{}")
if hasattr(tc, "function")
else "{}",
},
}
for i, tc in enumerate(raw)
@@ -1839,10 +1843,14 @@ class LLM(BaseLLM):
# Execute each tool call
for tc in raw:
func_name = sanitize_tool_name(
getattr(tc.function, "name", "") if hasattr(tc, "function") else ""
getattr(tc.function, "name", "")
if hasattr(tc, "function")
else ""
)
func_args_str = (
getattr(tc.function, "arguments", "{}") if hasattr(tc, "function") else "{}"
getattr(tc.function, "arguments", "{}")
if hasattr(tc, "function")
else "{}"
)
tool_call_id = getattr(tc, "id", f"call_{func_name}")
@@ -1872,7 +1880,9 @@ class LLM(BaseLLM):
fn = available_functions[func_name]
tool_output = fn(**func_args)
t1 = datetime.now()
record.output = str(tool_output) if tool_output is not None else ""
record.output = (
str(tool_output) if tool_output is not None else ""
)
record.duration_ms = (t1 - t0).total_seconds() * 1000
crewai_event_bus.emit(
self,
@@ -1908,11 +1918,13 @@ class LLM(BaseLLM):
result.tool_calls.append(record)
# Append tool result message for the model
conversation.append({
"role": "tool",
"tool_call_id": tool_call_id,
"content": record.output,
})
conversation.append(
{
"role": "tool",
"tool_call_id": tool_call_id,
"content": record.output,
}
)
else:
# Unexpected return type — treat as final text
result.text = str(raw)

8
lib/crewai/src/crewai/llm_result.py
View File

@@ -102,11 +102,11 @@ def _lookup_pricing(model: str) -> dict[str, float] | None:
return None
def estimate_cost_usd(
model: str, prompt_tokens: int, completion_tokens: int
) -> float:
def estimate_cost_usd(model: str, prompt_tokens: int, completion_tokens: int) -> float:
"""Estimate the cost in USD for a given model and token counts."""
pricing = _lookup_pricing(model)
if not pricing:
return 0.0
return (prompt_tokens * pricing["in"] + completion_tokens * pricing["out"]) / 1_000_000
return (
prompt_tokens * pricing["in"] + completion_tokens * pricing["out"]
) / 1_000_000