Fix ManageEngine typed.py

Author: whoisarpit

patchwork/steps/ManageEngineAgent/typed.py

@@ -1,40 +1,19 @@
-from typing import Dict, Optional, TypedDict
-
-from typing_extensions import Annotated
+from typing_extensions import Annotated, Dict, List, Optional, TypedDict
 
 from patchwork.common.utils.step_typing import StepTypeConfig
 
 
 class ManageEngineAgentInputs(TypedDict, total=False):
-    """
-    Inputs for the ManageEngine agentic step
-    """
-
-    # Required inputs
     me_access_token: str
-
-    # Optional configuration
     max_agent_calls: int
     openai_api_key: Annotated[
-        str,
-        StepTypeConfig(
-            is_config=True,
-            or_op=["patched_api_key", "google_api_key", "anthropic_api_key"],
-        ),
+        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "anthropic_api_key"])
     ]
     anthropic_api_key: Annotated[
-        str,
-        StepTypeConfig(
-            is_config=True,
-            or_op=["patched_api_key", "google_api_key", "openai_api_key"],
-        ),
+        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "google_api_key", "openai_api_key"])
     ]
     google_api_key: Annotated[
-        str,
-        StepTypeConfig(
-            is_config=True,
-            or_op=["patched_api_key", "openai_api_key", "anthropic_api_key"],
-        ),
+        str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "openai_api_key", "anthropic_api_key"])
     ]
 
     # Prompt and strategy configuration
@@ -43,11 +22,8 @@ class ManageEngineAgentInputs(TypedDict, total=False):
     example_json: Optional[Dict]
 
 
-class ManageEngineAgentOutputs(TypedDict, total=False):
-    """
-    Outputs from the ManageEngine agentic step
-    """
-
-    result: Dict
-    usage: Dict
-    error: Optional[str]
+class ManageEngineAgentOutputs(TypedDict):
+    conversation_history: List[Dict]
+    tool_records: List[Dict]
+    request_tokens: int
+    response_tokens: int