aboutsummaryrefslogtreecommitdiff
path: root/src/context.ts
diff options
context:
space:
mode:
Diffstat (limited to 'src/context.ts')
-rw-r--r--src/context.ts4
1 files changed, 3 insertions, 1 deletions
diff --git a/src/context.ts b/src/context.ts
index fab6653..f70a17d 100644
--- a/src/context.ts
+++ b/src/context.ts
@@ -39,7 +39,9 @@ export const createContext = (options: CtxOptions): Ctx => {
)
const getResultExpr = (resultKey?: string) =>
- `(${RESULT_TYPE_NAME}[${JSON.stringify(resultKey)})`
+ resultKey
+ ? `((${RESULT_TYPE_NAME}[${JSON.stringify(resultKey)})['output'])`
+ : 'undefined'
const addResult = (name: string, ty: string): Node | undefined =>
resultTypeNode