632ea8c032
Co-authored-by: Antoine <33425718+anth2o@users.noreply.github.com> Co-authored-by: Bastien <bastien.baret@gmail.com> Co-authored-by: Clément Sirieix <clement.sirieix@mistral.ai> Co-authored-by: Kim-Adeline Miguel <51720070+kimadeline@users.noreply.github.com> Co-authored-by: Mathias Gesbert <mathias.gesbert@mistral.ai> Co-authored-by: Maxime Dolores <maxime.dolores@ext.mistral.ai> Co-authored-by: Michel Thomazo <51709227+michelTho@users.noreply.github.com> Co-authored-by: Nelson PROIA <144663685+Nelson-PROIA@users.noreply.github.com> Co-authored-by: Pierre Rossinès <pierre.rossines@mistral.ai> Co-authored-by: Quentin <quentin.torroba@mistral.ai> Co-authored-by: Robin Gullo <robin.gullo@mistral.ai> Co-authored-by: Mistral Vibe <vibe@mistral.ai>
65 lines
1.7 KiB
Python
65 lines
1.7 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import cast
|
|
|
|
from vibe.core.telemetry.types import (
|
|
AgentEntrypoint,
|
|
EntrypointMetadata,
|
|
TelemetryBaseMetadata,
|
|
TelemetryCallType,
|
|
TelemetryRequestMetadata,
|
|
)
|
|
|
|
|
|
def build_base_metadata(
|
|
*,
|
|
entrypoint_metadata: EntrypointMetadata | None,
|
|
session_id: str | None,
|
|
parent_session_id: str | None = None,
|
|
) -> dict[str, str]:
|
|
entrypoint_payload = (
|
|
entrypoint_metadata.model_dump() if entrypoint_metadata is not None else {}
|
|
)
|
|
return cast(
|
|
dict[str, str],
|
|
TelemetryBaseMetadata(
|
|
session_id=session_id,
|
|
parent_session_id=parent_session_id,
|
|
**entrypoint_payload,
|
|
).model_dump(exclude_none=True),
|
|
)
|
|
|
|
|
|
def build_request_metadata(
|
|
*,
|
|
entrypoint_metadata: EntrypointMetadata | None,
|
|
session_id: str | None,
|
|
parent_session_id: str | None = None,
|
|
call_type: TelemetryCallType,
|
|
message_id: str | None = None,
|
|
) -> TelemetryRequestMetadata:
|
|
entrypoint_payload = (
|
|
entrypoint_metadata.model_dump() if entrypoint_metadata is not None else {}
|
|
)
|
|
return TelemetryRequestMetadata(
|
|
session_id=session_id,
|
|
parent_session_id=parent_session_id,
|
|
call_type=call_type,
|
|
message_id=message_id,
|
|
**entrypoint_payload,
|
|
)
|
|
|
|
|
|
def build_entrypoint_metadata(
|
|
*,
|
|
agent_entrypoint: AgentEntrypoint,
|
|
agent_version: str,
|
|
client_name: str,
|
|
client_version: str,
|
|
) -> EntrypointMetadata:
|
|
return EntrypointMetadata(
|
|
agent_entrypoint=agent_entrypoint,
|
|
agent_version=agent_version,
|
|
client_name=client_name,
|
|
client_version=client_version,
|
|
)
|