feat: fix issues

Author: NicolasGirardot

edgee/__init__.py

@@ -209,6 +209,7 @@ def tool_calls(self) -> list | None:
 @dataclass
 class StreamToolCallDelta:
     """Partial tool call in a stream."""
+
     index: int
     id: str | None = None
     type: str | None = None
@@ -266,20 +267,23 @@ def tool_call_deltas(self) -> list[dict] | None:
 @dataclass
 class ChunkEvent:
     """A chunk of streamed content."""
+
     type: str = "chunk"
     chunk: StreamChunk = None
 
 
 @dataclass
 class ToolStartEvent:
     """Tool execution is starting."""
+
     type: str = "tool_start"
     tool_call: dict = None
 
 
 @dataclass
 class ToolResultEvent:
     """Tool execution completed."""
+
     type: str = "tool_result"
     tool_call_id: str = None
     tool_name: str = None
@@ -289,6 +293,7 @@ class ToolResultEvent:
 @dataclass
 class IterationCompleteEvent:
     """One iteration of the tool loop completed."""
+
     type: str = "iteration_complete"
     iteration: int = 0
 
@@ -424,8 +429,12 @@ def _send_simple(
                     total_usage.prompt_tokens += response.usage.prompt_tokens
                     total_usage.completion_tokens += response.usage.completion_tokens
                     total_usage.total_tokens += response.usage.total_tokens
-                    total_usage.input_tokens_details.cached_tokens += response.usage.input_tokens_details.cached_tokens
-                    total_usage.output_tokens_details.reasoning_tokens += response.usage.output_tokens_details.reasoning_tokens
+                    total_usage.input_tokens_details.cached_tokens += (
+                        response.usage.input_tokens_details.cached_tokens
+                    )
+                    total_usage.output_tokens_details.reasoning_tokens += (
+                        response.usage.output_tokens_details.reasoning_tokens
+                    )
 
             # No tool calls? We're done - return final response
             if not response.tool_calls:
@@ -453,11 +462,13 @@ def _send_simple(
                     result = {"error": f"Unknown tool: {tool_name}"}
 
                 # Add tool result to messages
-                messages.append({
-                    "role": "tool",
-                    "tool_call_id": tool_call["id"],
-                    "content": result if isinstance(result, str) else json.dumps(result),
-                })
+                messages.append(
+                    {
+                        "role": "tool",
+                        "tool_call_id": tool_call["id"],
+                        "content": result if isinstance(result, str) else json.dumps(result),
+                    }
+                )
 
             # Loop continues - model will process tool results
 
@@ -536,7 +547,9 @@ def _handle_non_streaming_response(self, request: Request) -> SendResponse:
                     cached_tokens=usage_data.get("input_tokens_details", {}).get("cached_tokens", 0)
                 ),
                 output_tokens_details=OutputTokenDetails(
-                    reasoning_tokens=usage_data.get("output_tokens_details", {}).get("reasoning_tokens", 0)
+                    reasoning_tokens=usage_data.get("output_tokens_details", {}).get(
+                        "reasoning_tokens", 0
+                    )
                 ),
             )
 
@@ -746,11 +759,13 @@ def _stream_simple(
                 )
 
                 # Add tool result to messages
-                messages.append({
-                    "role": "tool",
-                    "tool_call_id": tool_call["id"],
-                    "content": result if isinstance(result, str) else json.dumps(result),
-                })
+                messages.append(
+                    {
+                        "role": "tool",
+                        "tool_call_id": tool_call["id"],
+                        "content": result if isinstance(result, str) else json.dumps(result),
+                    }
+                )
 
             # Yield iteration complete event
             yield IterationCompleteEvent(iteration=iteration)

examples/stream_tools.py

@@ -28,9 +28,7 @@ def get_weather(params: WeatherParams) -> dict:
         "London": {"temperature": 12, "condition": "rainy"},
         "New York": {"temperature": 22, "condition": "sunny"},
     }
-    data = weather_data.get(
-        params.location, {"temperature": 20, "condition": "unknown"}
-    )
+    data = weather_data.get(params.location, {"temperature": 20, "condition": "unknown"})
     return {
         "location": params.location,
         "temperature": data["temperature"],
@@ -62,7 +60,12 @@ def calculate(params: CalculatorParams) -> dict:
     }
     op = ops.get(params.operation)
     if op:
-        return {"operation": params.operation, "a": params.a, "b": params.b, "result": op(params.a, params.b)}
+        return {
+            "operation": params.operation,
+            "a": params.a,
+            "b": params.b,
+            "result": op(params.a, params.b),
+        }
     return {"error": f"Unknown operation: {params.operation}"}
 
 

examples/tools_auto.py

@@ -29,9 +29,7 @@ def get_weather(params: WeatherParams) -> dict:
         "London": {"temperature": 12, "condition": "rainy"},
         "New York": {"temperature": 22, "condition": "sunny"},
     }
-    data = weather_data.get(
-        params.location, {"temperature": 20, "condition": "unknown"}
-    )
+    data = weather_data.get(params.location, {"temperature": 20, "condition": "unknown"})
     return {
         "location": params.location,
         "temperature": data["temperature"],

examples/tools_multiple.py

@@ -26,9 +26,7 @@ def get_weather(params: WeatherParams) -> dict:
         "London": {"temperature": 12, "condition": "rainy"},
         "New York": {"temperature": 22, "condition": "sunny"},
     }
-    data = weather_data.get(
-        params.location, {"temperature": 20, "condition": "unknown"}
-    )
+    data = weather_data.get(params.location, {"temperature": 20, "condition": "unknown"})
     return {
         "location": params.location,
         "temperature": data["temperature"],