pre-commit fixes

Author: wdoppenberg

python-sdk/.pre-commit-config.yaml

@@ -1,26 +1,19 @@
 repos:
-  - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v5.0.0
-    hooks:
-      - id: trailing-whitespace
-        exclude_types: [ jupyter ]
-      - id: end-of-file-fixer
-        exclude_types: [ jupyter ]
-      - id: check-docstring-first
-      - id: debug-statements
-      - id: check-ast
   - repo: https://github.com/charliermarsh/ruff-pre-commit
     rev: v0.11.8
     hooks:
       - id: ruff
         args: [
           --fix
         ]
+        files: ^python-sdk/
       - id: ruff-format
+        files: ^python-sdk/
   - repo: https://github.com/pre-commit/mirrors-mypy
     rev: v1.15.0
     hooks:
       - id: mypy
+        files: ^python-sdk/
         args: [
           --python-version=3.12,
           --disallow-untyped-calls,
@@ -34,8 +27,4 @@ repos:
         additional_dependencies:
           - "types-pytz"
         exclude_types: [ jupyter ]
-        exclude: "tests"
-  - repo: https://github.com/kynan/nbstripout
-    rev: 0.8.1
-    hooks:
-      - id: nbstripout
+        exclude: "tests"

python-sdk/ag_ui/core/__init__.py

@@ -29,7 +29,7 @@
     RunErrorEvent,
     StepStartedEvent,
     StepFinishedEvent,
-    Event
+    Event,
 )
 
 from ag_ui.core.types import (

python-sdk/ag_ui/core/events.py

@@ -14,6 +14,7 @@ class EventType(str, Enum):
     """
     The type of event.
     """
+
     TEXT_MESSAGE_START = "TEXT_MESSAGE_START"
     TEXT_MESSAGE_CONTENT = "TEXT_MESSAGE_CONTENT"
     TEXT_MESSAGE_END = "TEXT_MESSAGE_END"
@@ -44,6 +45,7 @@ class BaseEvent(ConfiguredBaseModel):
     """
     Base event for all events in the Agent User Interaction Protocol.
     """
+
     type: EventType
     timestamp: Optional[int] = None
     raw_event: Optional[JSONValue] = None
@@ -53,6 +55,7 @@ class TextMessageStartEvent(BaseEvent):
     """
     Event indicating the start of a text message.
     """
+
     type: Literal[EventType.TEXT_MESSAGE_START] = EventType.TEXT_MESSAGE_START  # pyright: ignore[reportIncompatibleVariableOverride]
     message_id: str
     role: Literal["assistant"] = "assistant"
@@ -62,6 +65,7 @@ class TextMessageContentEvent(BaseEvent):
     """
     Event containing a piece of text message content.
     """
+
     type: Literal[EventType.TEXT_MESSAGE_CONTENT] = EventType.TEXT_MESSAGE_CONTENT  # pyright: ignore[reportIncompatibleVariableOverride]
     message_id: str
     delta: str = Field(min_length=1)
@@ -71,41 +75,58 @@ class TextMessageEndEvent(BaseEvent):
     """
     Event indicating the end of a text message.
     """
+
     type: Literal[EventType.TEXT_MESSAGE_END] = EventType.TEXT_MESSAGE_END  # pyright: ignore[reportIncompatibleVariableOverride]
     message_id: str
 
+
 class TextMessageChunkEvent(BaseEvent):
     """
     Event containing a chunk of text message content.
     """
+
     type: Literal[EventType.TEXT_MESSAGE_CHUNK] = EventType.TEXT_MESSAGE_CHUNK  # pyright: ignore[reportIncompatibleVariableOverride]
     message_id: Optional[str] = None
     role: Optional[Literal["assistant"]] = None
     delta: Optional[str] = None
 
+
 class ThinkingTextMessageStartEvent(BaseEvent):
     """
     Event indicating the start of a thinking text message.
     """
-    type: Literal[EventType.THINKING_TEXT_MESSAGE_START] = EventType.THINKING_TEXT_MESSAGE_START  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    type: Literal[EventType.THINKING_TEXT_MESSAGE_START] = (
+        EventType.THINKING_TEXT_MESSAGE_START
+    )  # pyright: ignore[reportIncompatibleVariableOverride]
+
 
 class ThinkingTextMessageContentEvent(BaseEvent):
     """
     Event indicating a piece of a thinking text message.
     """
-    type: Literal[EventType.THINKING_TEXT_MESSAGE_CONTENT] = EventType.THINKING_TEXT_MESSAGE_CONTENT  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    type: Literal[EventType.THINKING_TEXT_MESSAGE_CONTENT] = (
+        EventType.THINKING_TEXT_MESSAGE_CONTENT
+    )  # pyright: ignore[reportIncompatibleVariableOverride]
     delta: str = Field(min_length=1)
 
+
 class ThinkingTextMessageEndEvent(BaseEvent):
     """
     Event indicating the end of a thinking text message.
     """
-    type: Literal[EventType.THINKING_TEXT_MESSAGE_END] = EventType.THINKING_TEXT_MESSAGE_END  # pyright: ignore[reportIncompatibleVariableOverride]
+
+    type: Literal[EventType.THINKING_TEXT_MESSAGE_END] = (
+        EventType.THINKING_TEXT_MESSAGE_END
+    )  # pyright: ignore[reportIncompatibleVariableOverride]
+
 
 class ToolCallStartEvent(BaseEvent):
     """
     Event indicating the start of a tool call.
     """
+
     type: Literal[EventType.TOOL_CALL_START] = EventType.TOOL_CALL_START  # pyright: ignore[reportIncompatibleVariableOverride]
     tool_call_id: str
     tool_call_name: str
@@ -116,6 +137,7 @@ class ToolCallArgsEvent(BaseEvent):
     """
     Event containing tool call arguments.
     """
+
     type: Literal[EventType.TOOL_CALL_ARGS] = EventType.TOOL_CALL_ARGS  # pyright: ignore[reportIncompatibleVariableOverride]
     tool_call_id: str
     delta: str
@@ -125,46 +147,57 @@ class ToolCallEndEvent(BaseEvent):
     """
     Event indicating the end of a tool call.
     """
+
     type: Literal[EventType.TOOL_CALL_END] = EventType.TOOL_CALL_END  # pyright: ignore[reportIncompatibleVariableOverride]
     tool_call_id: str
 
+
 class ToolCallChunkEvent(BaseEvent):
     """
     Event containing a chunk of tool call content.
     """
+
     type: Literal[EventType.TOOL_CALL_CHUNK] = EventType.TOOL_CALL_CHUNK  # pyright: ignore[reportIncompatibleVariableOverride]
     tool_call_id: Optional[str] = None
     tool_call_name: Optional[str] = None
     parent_message_id: Optional[str] = None
     delta: Optional[str] = None
 
+
 class ToolCallResultEvent(BaseEvent):
     """
     Event containing the result of a tool call.
     """
+
     message_id: str
     type: Literal[EventType.TOOL_CALL_RESULT] = EventType.TOOL_CALL_RESULT  # pyright: ignore[reportIncompatibleVariableOverride]
     tool_call_id: str
     content: str
     role: Optional[Literal["tool"]] = None
 
+
 class ThinkingStartEvent(BaseEvent):
     """
     Event indicating the start of a thinking step event.
     """
+
     type: Literal[EventType.THINKING_START] = EventType.THINKING_START  # pyright: ignore[reportIncompatibleVariableOverride]
     title: Optional[str] = None
 
+
 class ThinkingEndEvent(BaseEvent):
     """
     Event indicating the end of a thinking step event.
     """
+
     type: Literal[EventType.THINKING_END] = EventType.THINKING_END  # pyright: ignore[reportIncompatibleVariableOverride]
 
+
 class StateSnapshotEvent(BaseEvent, Generic[AgentStateT]):
     """
     Event containing a snapshot of the state.
     """
+
     type: Literal[EventType.STATE_SNAPSHOT] = EventType.STATE_SNAPSHOT  # pyright: ignore[reportIncompatibleVariableOverride]
     snapshot: AgentStateT
 
@@ -173,6 +206,7 @@ class StateDeltaEvent(BaseEvent):
     """
     Event containing a delta of the state.
     """
+
     type: Literal[EventType.STATE_DELTA] = EventType.STATE_DELTA  # pyright: ignore[reportIncompatibleVariableOverride]
     delta: JSONValue  # JSON Patch (RFC 6902)
 
@@ -181,6 +215,7 @@ class MessagesSnapshotEvent(BaseEvent):
     """
     Event containing a snapshot of the messages.
     """
+
     type: Literal[EventType.MESSAGES_SNAPSHOT] = EventType.MESSAGES_SNAPSHOT  # pyright: ignore[reportIncompatibleVariableOverride]
     messages: List[Message]
 
@@ -189,6 +224,7 @@ class RawEvent(BaseEvent):
     """
     Event containing a raw event.
     """
+
     type: Literal[EventType.RAW] = EventType.RAW  # pyright: ignore[reportIncompatibleVariableOverride]
     event: JSONValue
     source: Optional[str] = None
@@ -198,6 +234,7 @@ class CustomEvent(BaseEvent):
     """
     Event containing a custom event.
     """
+
     type: Literal[EventType.CUSTOM] = EventType.CUSTOM  # pyright: ignore[reportIncompatibleVariableOverride]
     name: str
     value: JSONValue
@@ -207,6 +244,7 @@ class RunStartedEvent(BaseEvent):
     """
     Event indicating that a run has started.
     """
+
     type: Literal[EventType.RUN_STARTED] = EventType.RUN_STARTED  # pyright: ignore[reportIncompatibleVariableOverride]
     thread_id: str
     run_id: str
@@ -216,6 +254,7 @@ class RunFinishedEvent(BaseEvent):
     """
     Event indicating that a run has finished.
     """
+
     type: Literal[EventType.RUN_FINISHED] = EventType.RUN_FINISHED  # pyright: ignore[reportIncompatibleVariableOverride]
     thread_id: str
     run_id: str
@@ -226,6 +265,7 @@ class RunErrorEvent(BaseEvent):
     """
     Event indicating that a run has encountered an error.
     """
+
     type: Literal[EventType.RUN_ERROR] = EventType.RUN_ERROR  # pyright: ignore[reportIncompatibleVariableOverride]
     message: str
     code: Optional[str] = None
@@ -235,6 +275,7 @@ class StepStartedEvent(BaseEvent):
     """
     Event indicating that a step has started.
     """
+
     type: Literal[EventType.STEP_STARTED] = EventType.STEP_STARTED  # pyright: ignore[reportIncompatibleVariableOverride]
     step_name: str
 
@@ -243,6 +284,7 @@ class StepFinishedEvent(BaseEvent):
     """
     Event indicating that a step has finished.
     """
+
     type: Literal[EventType.STEP_FINISHED] = EventType.STEP_FINISHED  # pyright: ignore[reportIncompatibleVariableOverride]
     step_name: str
 
@@ -269,5 +311,5 @@ class StepFinishedEvent(BaseEvent):
         StepStartedEvent,
         StepFinishedEvent,
     ],
-    Field(discriminator="type")
+    Field(discriminator="type"),
 ]