Skip to content

fix(mistral): add tool spans for completions #381

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Abhijeet Prasad (AbhiPrasad) merged 1 commit into from
May 1, 2026
Merged

fix(mistral): add tool spans for completions #381

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
197 changes: 197 additions & 0 deletions ...t/integrations/mistral/cassettes/1.12.4/test_wrap_mistral_agents_complete_tool_spans.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,197 @@
interactions:
- request:
body: '{"model":"mistral-small-latest","name":"braintrust-test-agent-1777662640208","instructions":"You
are concise. Keep responses under five words."}'
headers:
Accept:
- application/json
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Content-Length:
- '144'
Host:
- api.mistral.ai
content-type:
- application/json
user-agent:
- mistral-client-python/1.12.4
method: POST
uri: https://api.mistral.ai/v1/agents
response:
body:
string: '{"instructions":"You are concise. Keep responses under five words.","tools":[],"completion_args":{"stop":null,"presence_penalty":null,"frequency_penalty":null,"temperature":null,"top_p":null,"max_tokens":null,"random_seed":null,"prediction":null,"response_format":null,"tool_choice":"auto","reasoning_effort":null},"guardrails":[],"model":"mistral-small-latest","name":"braintrust-test-agent-1777662640208","description":null,"handoffs":null,"metadata":null,"object":"agent","id":"ag_019de4f3414e7434ae6a78a856e4bafd","version":0,"versions":[],"created_at":"2026-05-01T19:10:40.491735Z","updated_at":"2026-05-01T19:10:40.491738Z","deployment_chat":false,"source":"api","version_message":null}'
headers:
CF-RAY:
- 9f51166de93746b5-YYZ
Connection:
- keep-alive
Content-Type:
- application/json
Date:
- Fri, 01 May 2026 19:10:40 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:
- '692'
mistral-correlation-id:
- 019de4f3-40f5-7b12-9317-dc66b898f3b1
set-cookie:
- __cf_bm=IHhDeC7rRK6YDtNcGbJe0_M_e2cgUmbG0oloO5VpkTs-1777662640.3053465-1.0.1.1-jT.JFGuSHJXanjc2Djujkxfx4rE0vcXBs0Hb7OW8W1aNfTIroDsCn_2VrY9kF.m44eU7H8YSwkH.Yh4HOfasl7nS_glA7HOQdvyn2KBdzbcrC0uZkNMzZqcczWqIiCkm;
HttpOnly; Secure; Path=/; Domain=mistral.ai; Expires=Fri, 01 May 2026 19:40:40
GMT
- _cfuvid=4.caIoynBFVOJU5CUbNRmeWN0aEZwc.d_BBtGYy0Ck8-1777662640.3053465-1.0.1.1-40ePzWEHHOdDI.y74_NIOYu42q283fv1RuXUVr3317E;
HttpOnly; SameSite=None; Secure; Path=/; Domain=mistral.ai
x-envoy-upstream-service-time:
- '95'
x-kong-proxy-latency:
- '41'
x-kong-request-id:
- 019de4f3-40f5-7b12-9317-dc66b898f3b1
x-kong-upstream-latency:
- '96'
status:
code: 200
message: OK
- request:
body: '{"messages":[{"content":"Use get_weather for Paris. Do not answer directly.","role":"user"}],"agent_id":"ag_019de4f3414e7434ae6a78a856e4bafd","max_tokens":100,"stream":false,"tools":[{"function":{"name":"get_weather","parameters":{"type":"object","properties":{"city":{"type":"string"}},"required":["city"]},"description":"Get
the weather for a city."},"type":"function"}],"tool_choice":"any"}'
headers:
Accept:
- application/json
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Content-Length:
- '393'
Cookie:
- __cf_bm=IHhDeC7rRK6YDtNcGbJe0_M_e2cgUmbG0oloO5VpkTs-1777662640.3053465-1.0.1.1-jT.JFGuSHJXanjc2Djujkxfx4rE0vcXBs0Hb7OW8W1aNfTIroDsCn_2VrY9kF.m44eU7H8YSwkH.Yh4HOfasl7nS_glA7HOQdvyn2KBdzbcrC0uZkNMzZqcczWqIiCkm;
_cfuvid=4.caIoynBFVOJU5CUbNRmeWN0aEZwc.d_BBtGYy0Ck8-1777662640.3053465-1.0.1.1-40ePzWEHHOdDI.y74_NIOYu42q283fv1RuXUVr3317E
Host:
- api.mistral.ai
content-type:
- application/json
user-agent:
- mistral-client-python/1.12.4
method: POST
uri: https://api.mistral.ai/v1/agents/completions
response:
body:
string: '{"id":"78cf63e6a47f4e9d8997f88f9d5d09a0","created":1777662640,"model":"mistral-small-latest","usage":{"prompt_tokens":100,"total_tokens":112,"completion_tokens":12,"prompt_tokens_details":{"cached_tokens":0}},"object":"chat.completion","choices":[{"index":0,"finish_reason":"tool_calls","message":{"role":"assistant","tool_calls":[{"id":"HpwsuA761","function":{"name":"get_weather","arguments":"{\"city\":
\"Paris\"}"},"index":0}],"content":""}}]}'
headers:
CF-RAY:
- 9f5116709c2cac7c-YYZ
Connection:
- keep-alive
Content-Type:
- application/json
Date:
- Fri, 01 May 2026 19:10:41 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:
- '447'
mistral-correlation-id:
- 019de4f3-42a6-7bcb-914d-147bd2dfa3ba
x-envoy-upstream-service-time:
- '285'
x-kong-proxy-latency:
- '13'
x-kong-request-id:
- 019de4f3-42a6-7bcb-914d-147bd2dfa3ba
x-kong-upstream-latency:
- '285'
x-ratelimit-limit-req-minute:
- '400'
x-ratelimit-limit-tokens-minute:
- '1500000'
x-ratelimit-remaining-req-minute:
- '394'
x-ratelimit-remaining-tokens-minute:
- '1499440'
x-ratelimit-tokens-query-cost:
- '112'
status:
code: 200
message: OK
- request:
body: ''
headers:
Accept:
- application/json
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Cookie:
- __cf_bm=IHhDeC7rRK6YDtNcGbJe0_M_e2cgUmbG0oloO5VpkTs-1777662640.3053465-1.0.1.1-jT.JFGuSHJXanjc2Djujkxfx4rE0vcXBs0Hb7OW8W1aNfTIroDsCn_2VrY9kF.m44eU7H8YSwkH.Yh4HOfasl7nS_glA7HOQdvyn2KBdzbcrC0uZkNMzZqcczWqIiCkm;
_cfuvid=4.caIoynBFVOJU5CUbNRmeWN0aEZwc.d_BBtGYy0Ck8-1777662640.3053465-1.0.1.1-40ePzWEHHOdDI.y74_NIOYu42q283fv1RuXUVr3317E
Host:
- api.mistral.ai
user-agent:
- mistral-client-python/1.12.4
method: DELETE
uri: https://api.mistral.ai/v1/agents/ag_019de4f3414e7434ae6a78a856e4bafd
response:
body:
string: ''
headers:
CF-RAY:
- 9f5116752c34f80e-YYZ
Connection:
- keep-alive
Date:
- Fri, 01 May 2026 19:10:41 GMT
Server:
- cloudflare
Strict-Transport-Security:
- max-age=15552000; includeSubDomains; preload
X-Content-Type-Options:
- nosniff
access-control-allow-origin:
- '*'
alt-svc:
- h3=":443"; ma=86400
cf-cache-status:
- DYNAMIC
mistral-correlation-id:
- 019de4f3-4579-71f1-8652-a0ef8f391026
x-envoy-upstream-service-time:
- '70'
x-kong-proxy-latency:
- '14'
x-kong-request-id:
- 019de4f3-4579-71f1-8652-a0ef8f391026
x-kong-upstream-latency:
- '71'
status:
code: 204
message: No Content
version: 1
194 changes: 194 additions & 0 deletions ...ust/integrations/mistral/cassettes/1.12.4/test_wrap_mistral_agents_stream_tool_spans.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,194 @@
interactions:
- request:
body: '{"model":"mistral-small-latest","name":"braintrust-test-agent-1777662671943","instructions":"You
are concise. Keep responses under five words."}'
headers:
Accept:
- application/json
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Content-Length:
- '144'
Host:
- api.mistral.ai
content-type:
- application/json
user-agent:
- mistral-client-python/1.12.4
method: POST
uri: https://api.mistral.ai/v1/agents
response:
body:
string: '{"instructions":"You are concise. Keep responses under five words.","tools":[],"completion_args":{"stop":null,"presence_penalty":null,"frequency_penalty":null,"temperature":null,"top_p":null,"max_tokens":null,"random_seed":null,"prediction":null,"response_format":null,"tool_choice":"auto","reasoning_effort":null},"guardrails":[],"model":"mistral-small-latest","name":"braintrust-test-agent-1777662671943","description":null,"handoffs":null,"metadata":null,"object":"agent","id":"ag_019de4f3bd267493989dd4f850ceb55b","version":0,"versions":[],"created_at":"2026-05-01T19:11:12.191782Z","updated_at":"2026-05-01T19:11:12.191786Z","deployment_chat":false,"source":"api","version_message":null}'
headers:
CF-RAY:
- 9f51173449b5cceb-YYZ
Connection:
- keep-alive
Content-Type:
- application/json
Date:
- Fri, 01 May 2026 19:11:12 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:
- '692'
mistral-correlation-id:
- 019de4f3-bce2-7629-b11e-cc23c1380f13
set-cookie:
- __cf_bm=mrhYeiViro6wHtnqEpjGd7hLPJ9mRg8F4nnNLW3ccTY-1777662672.0451913-1.0.1.1-0IRI6JbggwIc5aakVXrqZQm5PPHM.6xVcr90B638zB0obwP4lOxct1IbX4krwFGQFLq46M48D4NWvuD9p8r1RfcGdtWw4nZ.bXBhK9KkndJ_V5.E7_Kh8Kra7DFrwRjI;
HttpOnly; Secure; Path=/; Domain=mistral.ai; Expires=Fri, 01 May 2026 19:41:12
GMT
- _cfuvid=n4cckRYwtN0jFPxsF.RUuDU.CMxMRnq6LwMIFs0BFRU-1777662672.0451913-1.0.1.1-cGhPr72ahCa8TzVXUwb_f.ZXxxvaf3e3FaY7n6CAEOk;
HttpOnly; SameSite=None; Secure; Path=/; Domain=mistral.ai
x-envoy-upstream-service-time:
- '90'
x-kong-proxy-latency:
- '20'
x-kong-request-id:
- 019de4f3-bce2-7629-b11e-cc23c1380f13
x-kong-upstream-latency:
- '92'
status:
code: 200
message: OK
- request:
body: '{"messages":[{"content":"Use get_weather for Paris. Do not answer directly.","role":"user"}],"agent_id":"ag_019de4f3bd267493989dd4f850ceb55b","max_tokens":100,"stream":true,"tools":[{"function":{"name":"get_weather","parameters":{"type":"object","properties":{"city":{"type":"string"}},"required":["city"]},"description":"Get
the weather for a city."},"type":"function"}],"tool_choice":"any"}'
headers:
Accept:
- text/event-stream
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Content-Length:
- '392'
Cookie:
- __cf_bm=mrhYeiViro6wHtnqEpjGd7hLPJ9mRg8F4nnNLW3ccTY-1777662672.0451913-1.0.1.1-0IRI6JbggwIc5aakVXrqZQm5PPHM.6xVcr90B638zB0obwP4lOxct1IbX4krwFGQFLq46M48D4NWvuD9p8r1RfcGdtWw4nZ.bXBhK9KkndJ_V5.E7_Kh8Kra7DFrwRjI;
_cfuvid=n4cckRYwtN0jFPxsF.RUuDU.CMxMRnq6LwMIFs0BFRU-1777662672.0451913-1.0.1.1-cGhPr72ahCa8TzVXUwb_f.ZXxxvaf3e3FaY7n6CAEOk
Host:
- api.mistral.ai
content-type:
- application/json
user-agent:
- mistral-client-python/1.12.4
method: POST
uri: https://api.mistral.ai/v1/agents/completions
response:
body:
string: 'data: {"id":"f0045c73780a49f9baf96638d0c1635b","object":"chat.completion.chunk","created":1777662672,"model":"mistral-small-latest","choices":[{"index":0,"delta":{"role":"assistant","content":""},"finish_reason":null}]}


data: {"id":"f0045c73780a49f9baf96638d0c1635b","object":"chat.completion.chunk","created":1777662672,"model":"mistral-small-latest","choices":[{"index":0,"delta":{"tool_calls":[{"id":"ykk9sb1fw","function":{"name":"get_weather","arguments":"{\"city\":
\"Paris\"}"},"index":0}]},"finish_reason":"tool_calls"}],"usage":{"prompt_tokens":100,"total_tokens":112,"completion_tokens":12,"prompt_tokens_details":{"cached_tokens":80}},"p":"ab"}


data: [DONE]


'
headers:
CF-RAY:
- 9f5117364caf78a5-YYZ
Connection:
- keep-alive
Content-Type:
- text/event-stream; charset=utf-8
Date:
- Fri, 01 May 2026 19:11:13 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:
- 019de4f3-be2b-7b2e-8400-692abd2de685
x-envoy-upstream-service-time:
- '299'
x-kong-proxy-latency:
- '271'
x-kong-request-id:
- 019de4f3-be2b-7b2e-8400-692abd2de685
x-kong-upstream-latency:
- '300'
status:
code: 200
message: OK
- request:
body: ''
headers:
Accept:
- application/json
Accept-Encoding:
- gzip, deflate
Connection:
- keep-alive
Cookie:
- __cf_bm=mrhYeiViro6wHtnqEpjGd7hLPJ9mRg8F4nnNLW3ccTY-1777662672.0451913-1.0.1.1-0IRI6JbggwIc5aakVXrqZQm5PPHM.6xVcr90B638zB0obwP4lOxct1IbX4krwFGQFLq46M48D4NWvuD9p8r1RfcGdtWw4nZ.bXBhK9KkndJ_V5.E7_Kh8Kra7DFrwRjI;
_cfuvid=n4cckRYwtN0jFPxsF.RUuDU.CMxMRnq6LwMIFs0BFRU-1777662672.0451913-1.0.1.1-cGhPr72ahCa8TzVXUwb_f.ZXxxvaf3e3FaY7n6CAEOk
Host:
- api.mistral.ai
user-agent:
- mistral-client-python/1.12.4
method: DELETE
uri: https://api.mistral.ai/v1/agents/ag_019de4f3bd267493989dd4f850ceb55b
response:
body:
string: ''
headers:
CF-RAY:
- 9f51173f188cc109-YYZ
Connection:
- keep-alive
Date:
- Fri, 01 May 2026 19:11:13 GMT
Server:
- cloudflare
Strict-Transport-Security:
- max-age=15552000; includeSubDomains; preload
X-Content-Type-Options:
- nosniff
access-control-allow-origin:
- '*'
alt-svc:
- h3=":443"; ma=86400
cf-cache-status:
- DYNAMIC
mistral-correlation-id:
- 019de4f3-c3a8-7592-bba9-1a048f558172
x-envoy-upstream-service-time:
- '68'
x-kong-proxy-latency:
- '22'
x-kong-request-id:
- 019de4f3-c3a8-7592-bba9-1a048f558172
x-kong-upstream-latency:
- '69'
status:
code: 204
message: No Content
version: 1
Loading