From 53a526a4b68f12fe3e012afafd6cb01bc99d346f Mon Sep 17 00:00:00 2001
From: leavesster <11785335+leavesster@users.noreply.github.com>
Date: Fri, 7 Nov 2025 15:56:03 +0800
Subject: [PATCH] expose oomol token

---
 oocana/oocana/context.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/oocana/oocana/context.py b/oocana/oocana/context.py
index da19596..40f4cac 100644
--- a/oocana/oocana/context.py
+++ b/oocana/oocana/context.py
@@ -426,6 +426,13 @@ def write_pickle():
                     f"Output handle key: [{handle}] is defined as binary, but the file is not written."
                 )
         return value
+    
+    async def oomol_token(self) -> str:
+        """
+        get the oomol token from the mainframe.
+        :return: the oomol token
+        """
+        return os.getenv("OOMOL_TOKEN", "")
 
     def output(self, key: str, value: Any, *, to_node: list[ToNode] | None = None, to_flow: list[ToFlow] | None = None):
         """