passing file execution id from backend to tools (#145)

* passing file execution id from backend to tools

* addressed pr comments to set default values

Author: muhammad-ali-e

src/unstract/sdk/constants.py

@@ -127,6 +127,7 @@ class MetadataKey:
     SOURCE_HASH = "source_hash"
     WORKFLOW_ID = "workflow_id"
     EXECUTION_ID = "execution_id"
+    FILE_EXECUTION_ID = "file_execution_id"
     ORG_ID = "organization_id"
     TOOL_META = "tool_metadata"
     TOOL_NAME = "tool_name"
@@ -168,3 +169,10 @@ class MimeType:
     PDF = "application/pdf"
     TEXT = "text/plain"
     JSON = "application/json"
+
+
+class UsageKwargs:
+    RUN_ID = "run_id"
+    FILE_NAME = "file_name"
+    WORKFLOW_ID = "workflow_id"
+    EXECUTION_ID = "execution_id"

src/unstract/sdk/tool/base.py

@@ -39,6 +39,8 @@ def __init__(self, log_level: LogLevel = LogLevel.INFO) -> None:
         self.variables = ToolConfigHelper.variables()
         self.workflow_id = ""
         self.execution_id = ""
+        self.file_execution_id = ""
+        self.source_file_name = ""
         self.org_id = ""
         self._exec_metadata = {}
         self.filestorage_provider = None
@@ -84,7 +86,11 @@ def from_tool_args(cls, args: list[str]) -> "BaseTool":
         if parsed_args.command not in Command.static_commands():
             tool._exec_metadata = tool._get_exec_metadata()
             tool.workflow_id = tool._exec_metadata.get(MetadataKey.WORKFLOW_ID)
-            tool.execution_id = tool._exec_metadata.get(MetadataKey.EXECUTION_ID)
+            tool.execution_id = tool._exec_metadata.get(MetadataKey.EXECUTION_ID, "")
+            tool.file_execution_id = tool._exec_metadata.get(
+                MetadataKey.FILE_EXECUTION_ID, ""
+            )
+            tool.source_file_name = tool._exec_metadata.get(MetadataKey.SOURCE_NAME, "")
             tool.org_id = tool._exec_metadata.get(MetadataKey.ORG_ID)
         return tool