formatting

llama_toolchain/agentic_system/api/datatypes.py

@@ -63,9 +63,9 @@ class ShieldCallStep(StepCommon):
 
 @json_schema_type
 class MemoryRetrievalStep(StepCommon):
-    step_type: Literal[StepType.memory_retrieval.value] = (
+    step_type: Literal[
         StepType.memory_retrieval.value
-    )
+    ] = StepType.memory_retrieval.value
     memory_bank_ids: List[str]
     documents: List[MemoryBankDocument]
     scores: List[float]
@@ -140,9 +140,9 @@ class AgenticSystemTurnResponseEventType(Enum):
 
 @json_schema_type
 class AgenticSystemTurnResponseStepStartPayload(BaseModel):
-    event_type: Literal[AgenticSystemTurnResponseEventType.step_start.value] = (
+    event_type: Literal[
         AgenticSystemTurnResponseEventType.step_start.value
-    )
+    ] = AgenticSystemTurnResponseEventType.step_start.value
     step_type: StepType
     step_id: str
     metadata: Optional[Dict[str, Any]] = Field(default_factory=dict)
@@ -150,9 +150,9 @@ class AgenticSystemTurnResponseStepStartPayload(BaseModel):
 
 @json_schema_type
 class AgenticSystemTurnResponseStepCompletePayload(BaseModel):
-    event_type: Literal[AgenticSystemTurnResponseEventType.step_complete.value] = (
+    event_type: Literal[
         AgenticSystemTurnResponseEventType.step_complete.value
-    )
+    ] = AgenticSystemTurnResponseEventType.step_complete.value
     step_type: StepType
     step_details: Step
 
@@ -161,9 +161,9 @@ class AgenticSystemTurnResponseStepCompletePayload(BaseModel):
 class AgenticSystemTurnResponseStepProgressPayload(BaseModel):
     model_config = ConfigDict(protected_namespaces=())
 
-    event_type: Literal[AgenticSystemTurnResponseEventType.step_progress.value] = (
+    event_type: Literal[
         AgenticSystemTurnResponseEventType.step_progress.value
-    )
+    ] = AgenticSystemTurnResponseEventType.step_progress.value
     step_type: StepType
     step_id: str
 
@@ -174,17 +174,17 @@ class AgenticSystemTurnResponseStepProgressPayload(BaseModel):
 
 @json_schema_type
 class AgenticSystemTurnResponseTurnStartPayload(BaseModel):
-    event_type: Literal[AgenticSystemTurnResponseEventType.turn_start.value] = (
+    event_type: Literal[
         AgenticSystemTurnResponseEventType.turn_start.value
-    )
+    ] = AgenticSystemTurnResponseEventType.turn_start.value
     turn_id: str
 
 
 @json_schema_type
 class AgenticSystemTurnResponseTurnCompletePayload(BaseModel):
-    event_type: Literal[AgenticSystemTurnResponseEventType.turn_complete.value] = (
+    event_type: Literal[
         AgenticSystemTurnResponseEventType.turn_complete.value
-    )
+    ] = AgenticSystemTurnResponseEventType.turn_complete.value
     turn: Turn
 
 

llama_toolchain/agentic_system/api/endpoints.py

@@ -63,36 +63,40 @@ class AgenticSystemStepResponse(BaseModel):
 
 
 class AgenticSystem(Protocol):
-
     @webmethod(route="/agentic_system/create")
     async def create_agentic_system(
         self,
         request: AgenticSystemCreateRequest,
-    ) -> AgenticSystemCreateResponse: ...
+    ) -> AgenticSystemCreateResponse:
+        ...
 
     @webmethod(route="/agentic_system/turn/create")
     async def create_agentic_system_turn(
         self,
         request: AgenticSystemTurnCreateRequest,
-    ) -> AgenticSystemTurnResponseStreamChunk: ...
+    ) -> AgenticSystemTurnResponseStreamChunk:
+        ...
 
     @webmethod(route="/agentic_system/turn/get")
     async def get_agentic_system_turn(
         self,
         agent_id: str,
         turn_id: str,
-    ) -> Turn: ...
+    ) -> Turn:
+        ...
 
     @webmethod(route="/agentic_system/step/get")
     async def get_agentic_system_step(
         self, agent_id: str, turn_id: str, step_id: str
-    ) -> AgenticSystemStepResponse: ...
+    ) -> AgenticSystemStepResponse:
+        ...
 
     @webmethod(route="/agentic_system/session/create")
     async def create_agentic_system_session(
         self,
         request: AgenticSystemSessionCreateRequest,
-    ) -> AgenticSystemSessionCreateResponse: ...
+    ) -> AgenticSystemSessionCreateResponse:
+        ...
 
     @webmethod(route="/agentic_system/memory_bank/attach")
     async def attach_memory_bank_to_agentic_system(
@@ -100,7 +104,8 @@ class AgenticSystem(Protocol):
         agent_id: str,
         session_id: str,
         memory_bank_ids: List[str],
-    ) -> None: ...
+    ) -> None:
+        ...
 
     @webmethod(route="/agentic_system/memory_bank/detach")
     async def detach_memory_bank_from_agentic_system(
@@ -108,7 +113,8 @@ class AgenticSystem(Protocol):
         agent_id: str,
         session_id: str,
         memory_bank_ids: List[str],
-    ) -> None: ...
+    ) -> None:
+        ...
 
     @webmethod(route="/agentic_system/session/get")
     async def get_agentic_system_session(
@@ -116,15 +122,18 @@ class AgenticSystem(Protocol):
         agent_id: str,
         session_id: str,
         turn_ids: Optional[List[str]] = None,
-    ) -> Session: ...
+    ) -> Session:
+        ...
 
     @webmethod(route="/agentic_system/session/delete")
     async def delete_agentic_system_session(
         self, agent_id: str, session_id: str
-    ) -> None: ...
+    ) -> None:
+        ...
 
     @webmethod(route="/agentic_system/delete")
     async def delete_agentic_system(
         self,
         agent_id: str,
-    ) -> None: ...
+    ) -> None:
+        ...