fix typings

CTY-git · CTY-git · commit d4dbfa115897 · 2025-02-05T13:18:44.000+08:00
diff --git a/patchwork/steps/SimplifiedLLMOnce/SimplifiedLLMOnce.py b/patchwork/steps/SimplifiedLLMOnce/SimplifiedLLMOnce.py
@@ -1,6 +1,6 @@
 from patchwork.step import Step
 from patchwork.steps.SimplifiedLLM.SimplifiedLLM import SimplifiedLLM
-from patchwork.steps.SimplifiedLLMOnce.typed import SimplifiedLLMOnceInputs
+from patchwork.steps.SimplifiedLLMOnce.typed import SimplifiedLLMOnceInputs, SimplifiedLLMOnceOutputs
 
 
 class SimplifiedLLMOnce(Step, input_class=SimplifiedLLMOnceInputs, output_class=SimplifiedLLMOnceOutputs):
diff --git a/patchwork/steps/SimplifiedLLMOnce/typed.py b/patchwork/steps/SimplifiedLLMOnce/typed.py
@@ -38,3 +38,8 @@ class SimplifiedLLMOnceInputs(__SimplifiedLLMOncePBInputsRequired, total=False):
     google_api_key: Annotated[
         str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "openai_api_key", "anthropic_api_key"])
     ]
+
+
+class SimplifiedLLMOnceOutputs(TypedDict):
+    request_tokens: int
+    response_tokens: int

Original file line number	Diff line number	Diff line change
`@@ -38,3 +38,8 @@ class SimplifiedLLMOnceInputs(__SimplifiedLLMOncePBInputsRequired, total=False):`
`38`	`38`	`google_api_key: Annotated[`
`39`	`39`	`str, StepTypeConfig(is_config=True, or_op=["patched_api_key", "openai_api_key", "anthropic_api_key"])`
`40`	`40`	`]`
	`41`	`+`
	`42`	`+`
	`43`	`+class SimplifiedLLMOnceOutputs(TypedDict):`
	`44`	`+ request_tokens: int`
	`45`	`+ response_tokens: int`