formatting

llama_toolchain/agentic_system/api/datatypes.py

@@ -63,9 +63,9 @@ class ShieldCallStep(StepCommon):
 
 @json_schema_type
 class MemoryRetrievalStep(StepCommon):
-    step_type: Literal[
+    step_type: Literal[StepType.memory_retrieval.value] = (
         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[
+    event_type: Literal[AgenticSystemTurnResponseEventType.step_start.value] = (
         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[
+    event_type: Literal[AgenticSystemTurnResponseEventType.step_complete.value] = (
         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[
+    event_type: Literal[AgenticSystemTurnResponseEventType.step_progress.value] = (
         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[
+    event_type: Literal[AgenticSystemTurnResponseEventType.turn_start.value] = (
         AgenticSystemTurnResponseEventType.turn_start.value
-    ] = AgenticSystemTurnResponseEventType.turn_start.value
+    )
     turn_id: str
 
 
 @json_schema_type
 class AgenticSystemTurnResponseTurnCompletePayload(BaseModel):
-    event_type: Literal[
+    event_type: Literal[AgenticSystemTurnResponseEventType.turn_complete.value] = (
         AgenticSystemTurnResponseEventType.turn_complete.value
-    ] = AgenticSystemTurnResponseEventType.turn_complete.value
+    )
     turn: Turn
 
 

llama_toolchain/agentic_system/api/endpoints.py

@@ -67,36 +67,31 @@ class AgenticSystem(Protocol):
     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(
@@ -104,8 +99,7 @@ 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(
@@ -113,8 +107,7 @@ 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(
@@ -122,18 +115,15 @@ 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: ...