ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter9/gaia-experience/AWorld/tests/trace/metric/prometheus/prometheus_example.py

@@ -0,0 +1,48 @@
+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 = 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())