fix(mistral): add conversation tool spans

py/src/braintrust/integrations/mistral/cassettes/1.12.4/test_wrap_mistral_beta_conversations_start_stream_tool_spans.yaml

@@ -0,0 +1,115 @@
+interactions:
+- request:
+    body: '{"inputs":"Use the code interpreter to calculate 21 * 2. Return only the
+      result.","stream":true,"tools":[{"type":"code_interpreter"}],"model":"mistral-small-latest"}'
+    headers:
+      Accept:
+      - text/event-stream
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      Content-Length:
+      - '165'
+      Host:
+      - api.mistral.ai
+      content-type:
+      - application/json
+      user-agent:
+      - mistral-client-python/1.12.4
+    method: POST
+    uri: https://api.mistral.ai/v1/conversations
+  response:
+    body:
+      string: 'event: conversation.response.started
+
+        data: {"type":"conversation.response.started","created_at":"2026-05-04T18:17:38.536041Z","conversation_id":"conv_019df435c3a677148535d0889fa2490c"}
+
+
+        event: tool.execution.started
+
+        data: {"type":"tool.execution.started","created_at":"2026-05-04T18:17:38.540129Z","output_index":0,"id":"tool_exec_019df435c7ec7497b1feba55b9e05e09","model":"mistral-small-latest","name":"code_interpreter","arguments":"","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:38.550567Z","output_index":0,"id":"tool_exec_019df435c7ec7497b1feba55b9e05e09","name":"code_interpreter","arguments":"{\"code\":
+        \"","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:38.569375Z","output_index":0,"id":"tool_exec_019df435c7ec7497b1feba55b9e05e09","name":"code_interpreter","arguments":"21
+        * ","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:38.588794Z","output_index":0,"id":"tool_exec_019df435c7ec7497b1feba55b9e05e09","name":"code_interpreter","arguments":"2\"}","function":"code_interpreter"}
+
+
+        event: tool.execution.done
+
+        data: {"type":"tool.execution.done","created_at":"2026-05-04T18:17:39.246174Z","output_index":0,"id":"tool_exec_019df435c7ec7497b1feba55b9e05e09","name":"code_interpreter","function":"code_interpreter","info":{"code":"21
+        * 2","code_output":"42\n"}}
+
+
+        event: message.output.delta
+
+        data: {"type":"message.output.delta","created_at":"2026-05-04T18:17:39.496747Z","output_index":1,"id":"msg_019df435cba875c0be02cb5196bb0c6b","content_index":0,"model":"mistral-small-latest","role":"assistant","content":"4"}
+
+
+        event: message.output.delta
+
+        data: {"type":"message.output.delta","created_at":"2026-05-04T18:17:39.512024Z","output_index":1,"id":"msg_019df435cba875c0be02cb5196bb0c6b","content_index":0,"model":"mistral-small-latest","role":"assistant","content":"2"}
+
+
+        event: conversation.response.done
+
+        data: {"type":"conversation.response.done","created_at":"2026-05-04T18:17:39.534463Z","usage":{"prompt_tokens":86,"completion_tokens":19,"total_tokens":110,"connector_tokens":5,"connectors":{"code_interpreter":1}}}
+
+
+        '
+    headers:
+      CF-RAY:
+      - 9f6980d85e564d21-SJC
+      Connection:
+      - keep-alive
+      Content-Type:
+      - text/event-stream; charset=utf-8
+      Date:
+      - Mon, 04 May 2026 18:17:38 GMT
+      Server:
+      - cloudflare
+      Strict-Transport-Security:
+      - max-age=15552000; includeSubDomains; preload
+      Transfer-Encoding:
+      - chunked
+      X-Content-Type-Options:
+      - nosniff
+      access-control-allow-origin:
+      - '*'
+      alt-svc:
+      - h3=":443"; ma=86400
+      cf-cache-status:
+      - DYNAMIC
+      mistral-correlation-id:
+      - 019df435-c391-7b83-a2d1-5055d23b514f
+      set-cookie:
+      - __cf_bm=zMcNJeDI01Cs_2HPmgcbx8NK2vBd6q0te_lJkWy.Chg-1777918657.3398619-1.0.1.1-UHlbO0PlTlqBzpH16gQaciZIDn9Vo25APxvWZIsXdG5D5HBmLAro4wQfeyVtRwVUkk8hy_8aO5R90xX_XAIj0A.Ki1OeI.miQRj.WtJ71y1OK8CKJeoIPp2mVPaq7K8V;
+        HttpOnly; Secure; Path=/; Domain=mistral.ai; Expires=Mon, 04 May 2026 18:47:38
+        GMT
+      - _cfuvid=v6_zR1PTQDx2W0GmJI_6mxxXKChwk4Yf3Pe6CfQjgsI-1777918657.3398619-1.0.1.1-MYti7JhXogZfcgmc7gCEjnj57SSCgtvPropKKHhzXYw;
+        HttpOnly; SameSite=None; Secure; Path=/; Domain=mistral.ai
+      x-envoy-upstream-service-time:
+      - '1101'
+      x-kong-proxy-latency:
+      - '14'
+      x-kong-request-id:
+      - 019df435-c391-7b83-a2d1-5055d23b514f
+      x-kong-upstream-latency:
+      - '1102'
+    status:
+      code: 200
+      message: OK
+version: 1

py/src/braintrust/integrations/mistral/cassettes/1.12.4/test_wrap_mistral_beta_conversations_start_tool_spans.yaml

@@ -0,0 +1,86 @@
+interactions:
+- request:
+    body: '{"inputs":"Use the code interpreter to calculate 21 * 2. Return only the
+      result.","stream":false,"tools":[{"type":"code_interpreter"}],"model":"mistral-small-latest"}'
+    headers:
+      Accept:
+      - application/json
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      Content-Length:
+      - '166'
+      Host:
+      - api.mistral.ai
+      content-type:
+      - application/json
+      user-agent:
+      - mistral-client-python/1.12.4
+    method: POST
+    uri: https://api.mistral.ai/v1/conversations
+  response:
+    body:
+      string: '{"object":"conversation.response","conversation_id":"conv_019df4349547763ab5a0acfe1591236d","outputs":[{"object":"entry","type":"tool.execution","created_at":"2026-05-04T18:16:20.369317Z","completed_at":"2026-05-04T18:16:21.125072Z","model":"mistral-small-latest","id":"tool_exec_019df43496917039a5066b144b0542b6","name":"code_interpreter","arguments":"{\"code\":
+        \"21 * 2\"}","function":"code_interpreter","info":{"code":"21 * 2","code_output":"42\n"}},{"object":"entry","type":"message.output","created_at":"2026-05-04T18:16:21.242155Z","completed_at":"2026-05-04T18:16:21.272804Z","model":"mistral-small-latest","id":"msg_019df43499fa767a8fb220270fd7ce5b","role":"assistant","content":"42"}],"usage":{"prompt_tokens":86,"completion_tokens":19,"total_tokens":110,"connector_tokens":5,"connectors":{"code_interpreter":1}},"guardrails":null}'
+    headers:
+      CF-RAY:
+      - 9f697ef48c7849fc-SJC
+      Connection:
+      - keep-alive
+      Content-Type:
+      - application/json
+      Date:
+      - Mon, 04 May 2026 18:16:21 GMT
+      Server:
+      - cloudflare
+      Strict-Transport-Security:
+      - max-age=15552000; includeSubDomains; preload
+      Transfer-Encoding:
+      - chunked
+      X-Content-Type-Options:
+      - nosniff
+      access-control-allow-origin:
+      - '*'
+      alt-svc:
+      - h3=":443"; ma=86400
+      cf-cache-status:
+      - DYNAMIC
+      content-length:
+      - '841'
+      mistral-correlation-id:
+      - 019df434-952c-75b8-a13c-105b7be69de3
+      set-cookie:
+      - __cf_bm=srYdq.iW.XvPJCfCeOzjZi.Zj_lMoQjIq_WJsNR9CuE-1777918579.925641-1.0.1.1-tzEi4sdxAdz2irm_6.4lxf1GK.l8x4oyTxQEgn7FYZiSIUkpToQU.ZlVanS4aqidM4ENwJ1kO1d0ZLh3xj1jp.Pa.c2kaZCka52mZkBySQBH1u6_AETzHR.A0EAWJThk;
+        HttpOnly; Secure; Path=/; Domain=mistral.ai; Expires=Mon, 04 May 2026 18:46:21
+        GMT
+      - _cfuvid=Bs59mRiaoolnsjmFFfsJgMjpGWlBsQqzeA4B8Hpaztg-1777918579.925641-1.0.1.1-o3wi.3vNmTIzNBz5H38Uyb5lDmnUlSYDin8o1i1BWmM;
+        HttpOnly; SameSite=None; Secure; Path=/; Domain=mistral.ai
+      x-envoy-upstream-service-time:
+      - '1275'
+      x-kong-proxy-latency:
+      - '18'
+      x-kong-request-id:
+      - 019df434-952c-75b8-a13c-105b7be69de3
+      x-kong-upstream-latency:
+      - '1277'
+      x-ratelimit-limit-code-interpreter-day:
+      - '10'
+      x-ratelimit-limit-code-interpreter-minute:
+      - '3'
+      x-ratelimit-limit-tokens-minute:
+      - '500000'
+      x-ratelimit-limit-tokens-month:
+      - '1000000000'
+      x-ratelimit-remaining-code-interpreter-day:
+      - '9'
+      x-ratelimit-remaining-code-interpreter-minute:
+      - '2'
+      x-ratelimit-remaining-tokens-minute:
+      - '499890'
+      x-ratelimit-remaining-tokens-month:
+      - '999999392'
+    status:
+      code: 200
+      message: OK
+version: 1

py/src/braintrust/integrations/mistral/cassettes/latest/test_wrap_mistral_beta_conversations_start_stream_tool_spans.yaml

@@ -0,0 +1,115 @@
+interactions:
+- request:
+    body: '{"inputs":"Use the code interpreter to calculate 21 * 2. Return only the
+      result.","stream":true,"tools":[{"type":"code_interpreter"}],"model":"mistral-small-latest"}'
+    headers:
+      Accept:
+      - text/event-stream
+      Accept-Encoding:
+      - gzip, deflate
+      Connection:
+      - keep-alive
+      Content-Length:
+      - '165'
+      Host:
+      - api.mistral.ai
+      content-type:
+      - application/json
+      user-agent:
+      - mistral-client-python/2.4.4
+    method: POST
+    uri: https://api.mistral.ai/v1/conversations
+  response:
+    body:
+      string: 'event: conversation.response.started
+
+        data: {"type":"conversation.response.started","created_at":"2026-05-04T18:17:31.868571Z","conversation_id":"conv_019df435acbe72fe8159393921079b2b"}
+
+
+        event: tool.execution.started
+
+        data: {"type":"tool.execution.started","created_at":"2026-05-04T18:17:31.872991Z","output_index":0,"id":"tool_exec_019df435ade0730c8aea545a915fb65e","model":"mistral-small-latest","name":"code_interpreter","arguments":"","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:31.877128Z","output_index":0,"id":"tool_exec_019df435ade0730c8aea545a915fb65e","name":"code_interpreter","arguments":"{\"code\":
+        \"","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:31.892107Z","output_index":0,"id":"tool_exec_019df435ade0730c8aea545a915fb65e","name":"code_interpreter","arguments":"21
+        * ","function":"code_interpreter"}
+
+
+        event: tool.execution.delta
+
+        data: {"type":"tool.execution.delta","created_at":"2026-05-04T18:17:31.910950Z","output_index":0,"id":"tool_exec_019df435ade0730c8aea545a915fb65e","name":"code_interpreter","arguments":"2\"}","function":"code_interpreter"}
+
+
+        event: tool.execution.done
+
+        data: {"type":"tool.execution.done","created_at":"2026-05-04T18:17:32.570027Z","output_index":0,"id":"tool_exec_019df435ade0730c8aea545a915fb65e","name":"code_interpreter","function":"code_interpreter","info":{"code":"21
+        * 2","code_output":"42\n"}}
+
+
+        event: message.output.delta
+
+        data: {"type":"message.output.delta","created_at":"2026-05-04T18:17:32.707363Z","output_index":1,"id":"msg_019df435b12372eabb016e1e2f5d3d97","content_index":0,"model":"mistral-small-latest","role":"assistant","content":"4"}
+
+
+        event: message.output.delta
+
+        data: {"type":"message.output.delta","created_at":"2026-05-04T18:17:32.721768Z","output_index":1,"id":"msg_019df435b12372eabb016e1e2f5d3d97","content_index":0,"model":"mistral-small-latest","role":"assistant","content":"2"}
+
+
+        event: conversation.response.done
+
+        data: {"type":"conversation.response.done","created_at":"2026-05-04T18:17:32.748780Z","usage":{"prompt_tokens":86,"completion_tokens":19,"total_tokens":110,"connector_tokens":5,"connectors":{"code_interpreter":1}}}
+
+
+        '
+    headers:
+      CF-RAY:
+      - 9f6980b3a8d3d59a-SJC
+      Connection:
+      - keep-alive
+      Content-Type:
+      - text/event-stream; charset=utf-8
+      Date:
+      - Mon, 04 May 2026 18:17:31 GMT
+      Server:
+      - cloudflare
+      Strict-Transport-Security:
+      - max-age=15552000; includeSubDomains; preload
+      Transfer-Encoding:
+      - chunked
+      X-Content-Type-Options:
+      - nosniff
+      access-control-allow-origin:
+      - '*'
+      alt-svc:
+      - h3=":443"; ma=86400
+      cf-cache-status:
+      - DYNAMIC
+      mistral-correlation-id:
+      - 019df435-aca5-76ef-af89-b68d546fb35c
+      set-cookie:
+      - __cf_bm=MiLFLA_gRlztjcWsGqmI3aQqgrwWwIBCZOWsIkRfSEs-1777918651.470252-1.0.1.1-Xhd7DGqnK0C5DI9xz_5WdP6KBerfk.TGvtYM.FCrA098iP3Sq5xf.gZqeP1ZLGo3ovGnxMrfUkuUZqjoD5a2PSSeLmJXETg4G65ng9Mz5lRZg1QvRjK.2Z.Uh8ubZp0Q;
+        HttpOnly; Secure; Path=/; Domain=mistral.ai; Expires=Mon, 04 May 2026 18:47:31
+        GMT
+      - _cfuvid=UmqrIl_V_PrtzRN4zlVFwwLc7ynKa1oHPQqEjo1ytj0-1777918651.470252-1.0.1.1-Qc_29.Z00umjFznSjum_jqFURSIenwOKOGFMm.g7KnY;
+        HttpOnly; SameSite=None; Secure; Path=/; Domain=mistral.ai
+      x-envoy-upstream-service-time:
+      - '300'
+      x-kong-proxy-latency:
+      - '16'
+      x-kong-request-id:
+      - 019df435-aca5-76ef-af89-b68d546fb35c
+      x-kong-upstream-latency:
+      - '301'
+    status:
+      code: 200
+      message: OK
+version: 1