xinyin025/LangBot
1
0
Fork 0
mirror of https://github.com/langbot-app/LangBot.git synced 2026-08-14 14:31:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(telemetry): report workspace execution generation (#2374)

Browse Source 
Co-authored-by: dadachann <185672915+dadachann@users.noreply.github.com>
This commit is contained in:
Hyu
2026-08-01 02:10:59 +08:00
committed by GitHub
parent e3832ca536
commit d8ab0ba567
2 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/langbot/pkg/telemetry/heartbeat.py
+2
View File
@@ -37,6 +37,7 @@ class WorkspaceResourceSnapshot(typing.TypedDict):
extension_count: int extension_count: int
skill_count: int skill_count: int
adapters: list[str] adapters: list[str]
execution_generation: int
async def _count( async def _count(
@@ -81,6 +82,7 @@ async def _cloud_workspace_resource_counts(ap: core_app.Application, bindings) -
'extension_count': 0, 'extension_count': 0,
'skill_count': 0, 'skill_count': 0,
'adapters': [], 'adapters': [],
'execution_generation': binding.placement_generation,
} }
for binding in bindings for binding in bindings
} }
tests/unit_tests/telemetry/test_heartbeat.py
+4 -2
View File
@@ -139,8 +139,8 @@ class TestBuildHeartbeatPayload:
} }
ap.workspace_service.list_active_execution_bindings = AsyncMock( ap.workspace_service.list_active_execution_bindings = AsyncMock(
return_value=[ return_value=[
SimpleNamespace(workspace_uuid='workspace-a'), SimpleNamespace(workspace_uuid='workspace-a', placement_generation=7),
SimpleNamespace(workspace_uuid='workspace-b'), SimpleNamespace(workspace_uuid='workspace-b', placement_generation=9),
], ],
) )
ap.platform_mgr._bots_by_key[('instance-a', 'workspace-b', 'bot-b')] = SimpleNamespace( ap.platform_mgr._bots_by_key[('instance-a', 'workspace-b', 'bot-b')] = SimpleNamespace(
@@ -159,10 +159,12 @@ class TestBuildHeartbeatPayload:
assert by_workspace['workspace-a']['plugin_count'] == 2 assert by_workspace['workspace-a']['plugin_count'] == 2
assert by_workspace['workspace-a']['extension_count'] == 5 assert by_workspace['workspace-a']['extension_count'] == 5
assert by_workspace['workspace-a']['skill_count'] == 2 assert by_workspace['workspace-a']['skill_count'] == 2
assert by_workspace['workspace-a']['execution_generation'] == 7
assert by_workspace['workspace-a']['adapters'] == ['WorkspaceAAdapter'] assert by_workspace['workspace-a']['adapters'] == ['WorkspaceAAdapter']
assert by_workspace['workspace-b']['bot_count'] == 1 assert by_workspace['workspace-b']['bot_count'] == 1
assert by_workspace['workspace-b']['pipeline_count'] == 0 assert by_workspace['workspace-b']['pipeline_count'] == 0
assert by_workspace['workspace-b']['skill_count'] == 1 assert by_workspace['workspace-b']['skill_count'] == 1
assert by_workspace['workspace-b']['execution_generation'] == 9
assert by_workspace['workspace-b']['adapters'] == ['WorkspaceBAdapter'] assert by_workspace['workspace-b']['adapters'] == ['WorkspaceBAdapter']
assert 'workspace_resources' not in by_workspace['workspace-a'] assert 'workspace_resources' not in by_workspace['workspace-a']
ap.persistence_mgr.execute_async.assert_not_awaited() ap.persistence_mgr.execute_async.assert_not_awaited()