Add simulator context manager and execution summary (#908)

Summary:
Pull Request resolved: #908

To make the simulator a bit easier to use for monarch test workloads:
* Add context manager to set up logging, tracing, sim event loop
* print execution details after run including execution ID, scuba links to tracing/metrics table

Reviewed By: eliothedeman

Differential Revision: D80366675

fbshipit-source-id: 22c6267bc2e024b06a42215978cd1d863895a5a0

Author: benjipelletier

monarch_hyperactor/src/telemetry.rs

@@ -113,6 +113,12 @@ pub fn use_sim_clock() -> PyResult<()> {
     Ok(())
 }
 
+/// Get the current execution ID
+#[pyfunction]
+pub fn get_execution_id() -> PyResult<String> {
+    Ok(hyperactor_telemetry::env::execution_id())
+}
+
 // opentelemetry requires that the names of counters etc are static for the lifetime of the program.
 // Since we are binding these classes from python to rust, we have to leak these strings in order to
 // ensure they live forever. This is fine, as these classes aren't dynamically created.
@@ -320,6 +326,13 @@ pub fn register_python_bindings(module: &Bound<'_, PyModule>) -> PyResult<()> {
     )?;
     module.add_function(use_sim_clock_fn)?;
 
+    let get_execution_id_fn = wrap_pyfunction!(get_execution_id, module)?;
+    get_execution_id_fn.setattr(
+        "__module__",
+        "monarch._rust_bindings.monarch_hyperactor.telemetry",
+    )?;
+    module.add_function(get_execution_id_fn)?;
+
     module.add_class::<PySpan>()?;
     module.add_class::<PyCounter>()?;
     module.add_class::<PyHistogram>()?;

python/monarch/_rust_bindings/monarch_hyperactor/telemetry.pyi

@@ -87,6 +87,20 @@ def use_sim_clock() -> None:
     """
     ...
 
+def get_execution_id() -> str:
+    """
+    Get the current execution ID.
+
+    Returns the execution ID that is set by the telemetry system. This ID is either:
+    - Set from the HYPERACTOR_EXECUTION_ID environment variable
+    - Set from the MAST_HPC_JOB_NAME environment variable (in MAST environments)
+    - Generated as a random string if neither environment variable is available
+
+    Returns:
+        str: The current execution ID
+    """
+    ...
+
 class PySpan:
     def __init__(self, name: str) -> None:
         """