Spaces:
Sleeping
Sleeping
File size: 2,324 Bytes
fb49ac2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 |
import os # noqa
# os.environ["START_TRACE_SERVER"] = "false" # noqa
os.environ["MONITOR_SERVICE_NAME"] = "otlp_example" # noqa
# os.environ["OTLP_TRACES_ENDPOINT"] = "http://localhost:4318/v1/traces"
# os.environ["METRICS_SYSTEM_ENABLED"] = "true"
# os.environ["LOGFIRE_WRITE_TOKEN"] = (
# "Your logfire write token, "
# "create guide refer to "
# "https://logfire.pydantic.dev/docs/how-to-guides/create-write-tokens/"
# )
import aworld.trace as trace # noqa
from aworld.logs.util import logger, trace_logger
from aworld.trace.server import get_trace_server
trace.configure()
@trace.func_span(span_name="test_func", attributes={"test_attr": "test_value"}, extract_args=["param1"], add_attr="add_attr_value")
def traced_func(param1: str = None, param2: int = None):
trace_logger.info("this is a traced func")
traced_func2(param1="func2_param1_value", param2=222)
traced_func3(param1="func3_param1_value", param2=333)
@trace.func_span(span_name="test_func_2", add_attr="add_attr_value")
def traced_func2(param1: str = None, param2: int = None):
name = 'func2'
trace_logger.info(f"this is a traced {name}")
raise Exception("this is a traced func2 exception")
@trace.func_span
def traced_func3(param1: str = None, param2: int = None):
trace_logger.info("this is a traced func3")
def main():
logger.info("this is a no trace log")
trace.auto_tracing("examples.trace.*", 0.01)
with trace.span("hello") as span:
span.set_attribute("parent_test_attr", "pppppp")
logger.info("hello aworld")
trace_logger.info("trace hello aworld")
with trace.span("child hello") as span2:
span2.set_attribute("child_test_attr", "cccccc")
logger.info("child hello aworld")
current_span = trace.get_current_span()
logger.info("trace_id=%s", current_span.get_trace_id())
try:
traced_func(param1="func1_param1_value", param2=111)
except Exception as e:
logger.error(f"exception: {e}")
# from examples.trace.autotrace_demo import TestClassB
# b = TestClassB()
# b.classb_function_1()
# b.classb_function_2()
# b.classb_function_1()
# b.classb_function_2()
if get_trace_server():
get_trace_server().join()
if __name__ == "__main__":
main()
|