Upload prometheus_example.py

examples/metric/prometheus_example.py

@@ -0,0 +1,55 @@
+import random
+import time
+from aworld.metrics.metric import MetricType
+from aworld.metrics.context_manager import MetricContext, ApiMetricTracker
+from aworld.metrics.template import MetricTemplate
+
+MetricContext.configure(
+    provider="prometheus",
+    backend="console"
+)
+
+# my_counter = get_metric_provider().create_counter("my_counter", "test_counter_desc", "count")
+
+# while 1:
+#     my_counter.add(random.randint(1, 10), )
+#     time.sleep(random.random())
+
+
+my_counter = MetricTemplate(
+    type=MetricType.COUNTER,
+    name="my_counter",
+    description="My custom counter",
+    unit="1"
+)
+
+my_gauge = MetricTemplate(
+    type=MetricType.GAUGE,
+    name="my_gauge"
+)
+
+my_histogram = MetricTemplate(
+    type=MetricType.HISTOGRAM,
+    name="my_histogram",
+    buckets=[2, 4, 6, 8, 10]
+)
+
+
+@ApiMetricTracker()
+def api():
+    time.sleep(random.uniform(0, 1))
+
+
+def custom_code():
+    with ApiMetricTracker("test_custom_code"):
+        time.sleep(random.uniform(0, 1))
+
+
+if __name__ == '__main__':
+    while 1:
+        MetricContext.count(my_counter, random.randint(1, 10))
+        MetricContext.gauge_set(my_gauge, random.randint(1, 10))
+        MetricContext.histogram_record(my_histogram, random.randint(1, 10))
+        api()
+        custom_code()
+        time.sleep(random.random())