trodo-python 2.4.1__tar.gz → 2.5.0__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {trodo_python-2.4.1 → trodo_python-2.5.0}/PKG-INFO +1 -1
- {trodo_python-2.4.1 → trodo_python-2.5.0}/pyproject.toml +1 -1
- trodo_python-2.5.0/tests/test_anon_distinct_id.py +100 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/wrap_agent.py +43 -7
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo_python.egg-info/PKG-INFO +1 -1
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo_python.egg-info/SOURCES.txt +1 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/README.md +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/setup.cfg +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_auto_instrument_fixes.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_cross_process_session.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_end_run.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_processor_methods.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_register_otel.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_start_run.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/tests/test_wrap_agent_unchanged.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/api/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/api/async_client.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/api/endpoints.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/api/http_client.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/auto/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/auto/auto_event_manager.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/client.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/managers/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/managers/group_manager.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/managers/people_manager.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/auto_instrument.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/context.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/helpers.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/processor.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/register.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/otel/transport.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/queue/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/queue/batch_flusher.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/queue/event_queue.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/session/__init__.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/session/server_session.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/session/session_manager.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/types.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo/user_context.py +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo_python.egg-info/dependency_links.txt +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo_python.egg-info/requires.txt +0 -0
- {trodo_python-2.4.1 → trodo_python-2.5.0}/trodo_python.egg-info/top_level.txt +0 -0
|
@@ -0,0 +1,100 @@
|
|
|
1
|
+
"""Anonymous distinct_id minting on agent surfaces.
|
|
2
|
+
|
|
3
|
+
When the caller doesn't pass ``distinct_id``, the SDK mints an
|
|
4
|
+
``anon_<ts>_python_<uuid>_<rand>`` id so:
|
|
5
|
+
|
|
6
|
+
* the agent_runs row always lands with a non-null distinct_id,
|
|
7
|
+
* the RunHandle exposes the same id so callers can bind
|
|
8
|
+
``trodo.feedback(distinct_id=...)`` later,
|
|
9
|
+
* older backends without identity-resolution still attribute the row.
|
|
10
|
+
"""
|
|
11
|
+
from __future__ import annotations
|
|
12
|
+
|
|
13
|
+
import re
|
|
14
|
+
|
|
15
|
+
from trodo.otel.wrap_agent import _mint_anon_distinct_id, start_run, wrap_agent
|
|
16
|
+
|
|
17
|
+
|
|
18
|
+
ANON_RE = re.compile(r"^anon_\d+_python_")
|
|
19
|
+
|
|
20
|
+
|
|
21
|
+
def test_mint_anon_distinct_id_shape():
|
|
22
|
+
a = _mint_anon_distinct_id()
|
|
23
|
+
assert ANON_RE.match(a)
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
def test_mint_anon_distinct_id_unique_under_load():
|
|
27
|
+
ids = {_mint_anon_distinct_id() for _ in range(1000)}
|
|
28
|
+
assert len(ids) == 1000
|
|
29
|
+
|
|
30
|
+
|
|
31
|
+
def test_wrap_agent_mints_anon_when_distinct_id_omitted(processor, http):
|
|
32
|
+
observed = {}
|
|
33
|
+
with wrap_agent(
|
|
34
|
+
processor=processor,
|
|
35
|
+
team_site_id="site-x",
|
|
36
|
+
agent_name="chat",
|
|
37
|
+
) as run:
|
|
38
|
+
observed["distinct_id"] = run.distinct_id
|
|
39
|
+
run.set_output("done")
|
|
40
|
+
|
|
41
|
+
assert ANON_RE.match(observed["distinct_id"])
|
|
42
|
+
assert len(http.run_ingest) == 1
|
|
43
|
+
assert http.run_ingest[0]["run"]["distinct_id"] == observed["distinct_id"]
|
|
44
|
+
|
|
45
|
+
|
|
46
|
+
def test_wrap_agent_respects_explicit_distinct_id(processor, http):
|
|
47
|
+
with wrap_agent(
|
|
48
|
+
processor=processor,
|
|
49
|
+
team_site_id="site-x",
|
|
50
|
+
agent_name="chat",
|
|
51
|
+
distinct_id="user-42",
|
|
52
|
+
) as run:
|
|
53
|
+
assert run.distinct_id == "user-42"
|
|
54
|
+
|
|
55
|
+
assert http.run_ingest[0]["run"]["distinct_id"] == "user-42"
|
|
56
|
+
|
|
57
|
+
|
|
58
|
+
def test_wrap_agent_mints_different_anon_ids_across_calls(processor, http):
|
|
59
|
+
seen = []
|
|
60
|
+
for _ in range(3):
|
|
61
|
+
with wrap_agent(
|
|
62
|
+
processor=processor,
|
|
63
|
+
team_site_id="site-x",
|
|
64
|
+
agent_name="chat",
|
|
65
|
+
) as run:
|
|
66
|
+
seen.append(run.distinct_id)
|
|
67
|
+
assert len(set(seen)) == 3
|
|
68
|
+
|
|
69
|
+
|
|
70
|
+
def test_wrap_agent_still_mints_anon_on_error(processor, http):
|
|
71
|
+
import pytest
|
|
72
|
+
with pytest.raises(ValueError):
|
|
73
|
+
with wrap_agent(
|
|
74
|
+
processor=processor,
|
|
75
|
+
team_site_id="site-x",
|
|
76
|
+
agent_name="chat",
|
|
77
|
+
) as _run:
|
|
78
|
+
raise ValueError("boom")
|
|
79
|
+
|
|
80
|
+
assert len(http.run_ingest) == 1
|
|
81
|
+
payload = http.run_ingest[0]["run"]
|
|
82
|
+
assert payload["status"] == "error"
|
|
83
|
+
assert ANON_RE.match(payload["distinct_id"])
|
|
84
|
+
|
|
85
|
+
|
|
86
|
+
def test_start_run_mints_anon_when_distinct_id_omitted(processor, http):
|
|
87
|
+
start_run(processor=processor, agent_name="external_session")
|
|
88
|
+
assert len(http.run_start) == 1
|
|
89
|
+
run = http.run_start[0]["run"]
|
|
90
|
+
assert ANON_RE.match(run["distinct_id"])
|
|
91
|
+
|
|
92
|
+
|
|
93
|
+
def test_start_run_respects_explicit_distinct_id(processor, http):
|
|
94
|
+
start_run(
|
|
95
|
+
processor=processor,
|
|
96
|
+
agent_name="external_session",
|
|
97
|
+
distinct_id="user-7",
|
|
98
|
+
)
|
|
99
|
+
run = http.run_start[0]["run"]
|
|
100
|
+
assert run["distinct_id"] == "user-7"
|
|
@@ -114,12 +114,37 @@ def _aggregate(spans: list[TrodoSpan]) -> Dict[str, Any]:
|
|
|
114
114
|
}
|
|
115
115
|
|
|
116
116
|
|
|
117
|
+
def _mint_anon_distinct_id() -> str:
|
|
118
|
+
"""Mint a server-side anonymous distinct_id for an agent run.
|
|
119
|
+
|
|
120
|
+
Mirrors UserIdentity.generateAnonymousDistinctId on the backend
|
|
121
|
+
(``anon_<ts>_<scope>_<uuid>_<rand>``) so server-SDK anon runs and
|
|
122
|
+
browser-SDK anon sessions share one prefix the dashboard already
|
|
123
|
+
filters on.
|
|
124
|
+
|
|
125
|
+
Minted client-side (not just server-side) so:
|
|
126
|
+
* the RunHandle / OTLP span attribute carries the id immediately and
|
|
127
|
+
downstream ``trodo.feedback(distinct_id=...)`` calls have something
|
|
128
|
+
to bind to,
|
|
129
|
+
* users running against an older backend still get a real attributable
|
|
130
|
+
distinct_id on every ``agent_runs`` row, instead of NULL.
|
|
131
|
+
"""
|
|
132
|
+
import random
|
|
133
|
+
import string
|
|
134
|
+
ts = int(time.time() * 1000)
|
|
135
|
+
rand = "".join(random.choices(string.ascii_lowercase + string.digits, k=16))
|
|
136
|
+
return f"anon_{ts}_python_{uuid.uuid4()}_{rand}"
|
|
137
|
+
|
|
138
|
+
|
|
117
139
|
class RunHandle:
|
|
118
140
|
"""Handle returned by wrap_agent for setting input/output and getting run_id."""
|
|
119
141
|
|
|
120
|
-
def __init__(self, run_id: str, agent_name: str) -> None:
|
|
142
|
+
def __init__(self, run_id: str, agent_name: str, distinct_id: str) -> None:
|
|
121
143
|
self.run_id = run_id
|
|
122
144
|
self.agent_name = agent_name
|
|
145
|
+
# Always populated — wrap_agent mints anon if caller didn't pass one
|
|
146
|
+
# so downstream ``trodo.feedback(distinct_id=...)`` always has a target.
|
|
147
|
+
self.distinct_id = distinct_id
|
|
123
148
|
self.input: Optional[str] = None
|
|
124
149
|
self.output: Optional[str] = None
|
|
125
150
|
self.metadata: Dict[str, Any] = {}
|
|
@@ -209,10 +234,16 @@ def start_run(
|
|
|
209
234
|
to add spans — they flush incrementally via ``append_spans``.
|
|
210
235
|
"""
|
|
211
236
|
rid = run_id or str(uuid.uuid4())
|
|
237
|
+
# Mint anon when missing so the run row is attributable. The minted id
|
|
238
|
+
# is stamped onto the TrodoRun payload but is not surfaced to the caller —
|
|
239
|
+
# start_run's signature returns just run_id for backward compat. Callers
|
|
240
|
+
# who need the distinct_id should pass their own (or use wrap_agent,
|
|
241
|
+
# whose RunHandle exposes it via handle.distinct_id).
|
|
242
|
+
effective_distinct_id = distinct_id or _mint_anon_distinct_id()
|
|
212
243
|
run = TrodoRun(
|
|
213
244
|
run_id=rid,
|
|
214
245
|
agent_name=agent_name,
|
|
215
|
-
distinct_id=
|
|
246
|
+
distinct_id=effective_distinct_id,
|
|
216
247
|
conversation_id=conversation_id,
|
|
217
248
|
parent_run_id=parent_run_id,
|
|
218
249
|
status="running",
|
|
@@ -278,7 +309,11 @@ class wrap_agent:
|
|
|
278
309
|
self._processor = processor
|
|
279
310
|
self._team_site_id = team_site_id
|
|
280
311
|
self._agent_name = agent_name
|
|
281
|
-
|
|
312
|
+
# Mint anon when caller didn't pass one. From this point on every
|
|
313
|
+
# internal path uses self._distinct_id, never the raw constructor
|
|
314
|
+
# argument, so the run row, OTLP span attribute, and RunHandle all
|
|
315
|
+
# agree on a single non-null value.
|
|
316
|
+
self._distinct_id = distinct_id or _mint_anon_distinct_id()
|
|
282
317
|
self._conversation_id = conversation_id
|
|
283
318
|
self._parent_run_id = parent_run_id
|
|
284
319
|
self._metadata = metadata
|
|
@@ -312,7 +347,7 @@ class wrap_agent:
|
|
|
312
347
|
self._started_iso = _now_iso()
|
|
313
348
|
self._started_ms = time.time() * 1000.0
|
|
314
349
|
|
|
315
|
-
self.handle = RunHandle(run_id, self._agent_name)
|
|
350
|
+
self.handle = RunHandle(run_id, self._agent_name, self._distinct_id)
|
|
316
351
|
ctx = ActiveSpanContext(
|
|
317
352
|
run_id=run_id,
|
|
318
353
|
span_id=root_span_id,
|
|
@@ -387,8 +422,9 @@ class wrap_agent:
|
|
|
387
422
|
run_id = _hex_to_uuid(trace_id_hex)
|
|
388
423
|
|
|
389
424
|
otel_span.set_attribute("trodo.agent_name", self._agent_name)
|
|
390
|
-
|
|
391
|
-
|
|
425
|
+
# self._distinct_id is always set (anon-minted in __init__ when caller
|
|
426
|
+
# didn't pass one), so the attribute always lands on the OTLP span.
|
|
427
|
+
otel_span.set_attribute("trodo.distinct_id", str(self._distinct_id))
|
|
392
428
|
if self._conversation_id:
|
|
393
429
|
otel_span.set_attribute("trodo.conversation_id", str(self._conversation_id))
|
|
394
430
|
if self._parent_run_id:
|
|
@@ -398,7 +434,7 @@ class wrap_agent:
|
|
|
398
434
|
otel_span.set_attribute(f"trodo.metadata.{k}", _serialize_attr(v))
|
|
399
435
|
|
|
400
436
|
self._otel_span = otel_span
|
|
401
|
-
self.handle = RunHandle(run_id, self._agent_name)
|
|
437
|
+
self.handle = RunHandle(run_id, self._agent_name, self._distinct_id)
|
|
402
438
|
return self.handle
|
|
403
439
|
|
|
404
440
|
def _exit_otel(self, exc_type, exc, tb) -> None:
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|