Skip to content

Commit cdcde62

Browse files
committed
update server
1 parent 1024b82 commit cdcde62

File tree

1 file changed

+5
-1
lines changed
  • eval_protocol/quickstart/svg_agent/vercel_svg_server/api

1 file changed

+5
-1
lines changed

eval_protocol/quickstart/svg_agent/vercel_svg_server/api/init.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,10 +67,14 @@ def do_POST(self):
6767
self._send_error(400, error_msg)
6868
return
6969

70+
model = req.completion_params.get("model")
71+
if model and isinstance(model, str) and model.startswith("fireworks_ai/"):
72+
model = model[len("fireworks_ai/") :]
73+
7074
# Prepare completion arguments
7175
completion_kwargs = {
7276
"messages": req.messages,
73-
"model": req.completion_params.get("model"),
77+
"model": model,
7478
"temperature": req.completion_params.get("temperature"),
7579
"max_tokens": req.completion_params.get("max_tokens"),
7680
}

0 commit comments

Comments
 (0)