mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-14 20:16:48 +00:00
Skip trace work when disabled; share the agent span builder
With TRACES_ENABLED off, LLM calls record no GenAI metrics, and a streamed answer is joined into a preview only when a span keeps it. Agent runs and continuations build their invoke_agent span in one place, which now reads the model from model_id. The trace panel shows a stream's model time next to how long it was open. Adds missing type hints.
This commit is contained in:
1 parent
699906a015
commit
0bbbac0eb2
18 files changed
+88
-35
No files matched your search
@@ -437,6 +437,8 @@ class TestLogActivityTraceSpan:
|
||||
assert agent_span.attributes["gen_ai.operation.name"] == "invoke_agent"
|
||||
assert agent_span.attributes["gen_ai.agent.id"] == "agent-1"
|
||||
assert agent_span.attributes["docsgpt.answer_chars"] == 2
|
||||
# BaseAgent keeps its model in ``model_id``; the span reads it too.
|
||||
assert agent_span.attributes["gen_ai.request.model"] == "gpt-4o"
|
||||
assert agent_span.attributes["docsgpt.source_count"] == 1
|
||||
assert llm_span.parent_id == agent_span.id
|
||||
assert trace.activity_id is not None
|
||||
|
||||
Reference in new issue
Block a user