Skip to content

Commit c9462e9

Browse files
committed
fix
1 parent 292c5ef commit c9462e9

File tree

1 file changed

+16
-1
lines changed

1 file changed

+16
-1
lines changed

tests/remote_server/remote_server.py

Lines changed: 16 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,8 +37,23 @@ def _worker():
3737
if not model:
3838
raise ValueError("model is required in completion_params")
3939

40+
# Convert Eval Protocol Message objects into OpenAI-compatible dicts,
41+
# excluding any None fields (Fireworks rejects extra keys even when null).
42+
messages_payload = []
43+
for m in req.messages:
44+
if hasattr(m, "dump_mdoel_for_chat_completion_request"):
45+
md = m.dump_mdoel_for_chat_completion_request() # type: ignore[attr-defined]
46+
elif hasattr(m, "model_dump"):
47+
md = m.model_dump(exclude_none=True) # type: ignore[call-arg]
48+
elif isinstance(m, dict):
49+
md = {k: v for k, v in m.items() if v is not None}
50+
else:
51+
md = {"role": getattr(m, "role", None), "content": getattr(m, "content", None)}
52+
md = {k: v for k, v in md.items() if v is not None}
53+
messages_payload.append(md)
54+
4055
# Spread all completion_params (model, temperature, max_tokens, etc.)
41-
completion_kwargs = {"messages": req.messages, **req.completion_params}
56+
completion_kwargs = {"messages": messages_payload, **req.completion_params}
4257

4358
if req.tools:
4459
completion_kwargs["tools"] = req.tools

0 commit comments

Comments
 (0)