|
14 | 14 |
|
15 | 15 | import os |
16 | 16 |
|
17 | | -from agent import AGENT, APP_NAME, SHORT_TERM_MEMORY |
| 17 | +from agent import agent, app_name, short_term_memory |
18 | 18 | from veadk.a2a.ve_a2a_server import init_app |
19 | 19 | from veadk.tracing.base_tracer import BaseTracer |
20 | 20 | from veadk.tracing.telemetry.opentelemetry_tracer import OpentelemetryTracer |
|
44 | 44 | TRACERS.append(OpentelemetryTracer(exporters=exporters)) |
45 | 45 |
|
46 | 46 |
|
47 | | -AGENT.tracers.extend(TRACERS) |
48 | | -if not getattr(AGENT, "before_model_callback", None): |
49 | | - AGENT.before_model_callback = [] |
50 | | -if not getattr(AGENT, "after_model_callback", None): |
51 | | - AGENT.after_model_callback = [] |
| 47 | +agent.tracers.extend(TRACERS) |
| 48 | +if not getattr(agent, "before_model_callback", None): |
| 49 | + agent.before_model_callback = [] |
| 50 | +if not getattr(agent, "after_model_callback", None): |
| 51 | + agent.after_model_callback = [] |
52 | 52 | for tracer in TRACERS: |
53 | | - if tracer.llm_metrics_hook not in AGENT.before_model_callback: |
54 | | - AGENT.before_model_callback.append(tracer.llm_metrics_hook) |
55 | | - if tracer.token_metrics_hook not in AGENT.after_model_callback: |
56 | | - AGENT.after_model_callback.append(tracer.token_metrics_hook) |
| 53 | + if tracer.llm_metrics_hook not in agent.before_model_callback: |
| 54 | + agent.before_model_callback.append(tracer.llm_metrics_hook) |
| 55 | + if tracer.token_metrics_hook not in agent.after_model_callback: |
| 56 | + agent.after_model_callback.append(tracer.token_metrics_hook) |
57 | 57 |
|
58 | 58 | # Tracer Config ================================================================ |
59 | 59 | # ============================================================================== |
60 | 60 |
|
61 | 61 | app = init_app( |
62 | 62 | server_url="0.0.0.0", # Automatic identification is not supported yet. |
63 | | - app_name=APP_NAME, |
64 | | - agent=AGENT, |
65 | | - short_term_memory=SHORT_TERM_MEMORY, |
| 63 | + app_name=app_name, |
| 64 | + agent=agent, |
| 65 | + short_term_memory=short_term_memory, |
66 | 66 | ) |
0 commit comments