Skip to content

Commit 8e3ea83

Browse files
committed
avoid double serialization
1 parent c1efd4a commit 8e3ea83

File tree

3 files changed

+10
-2
lines changed

3 files changed

+10
-2
lines changed

src/balancer/chunk_forwarding_session_controller/mod.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ where
4040
{
4141
async fn send_response(&mut self, message: OutgoingMessage) -> anyhow::Result<()> {
4242
let transformed_message = self.transformer.transform(message).await?;
43-
let stringified_message = serde_json::to_string(&transformed_message)?;
43+
let stringified_message = self.transformer.stringify(&transformed_message)?;
4444

4545
self.chunk_tx.send(stringified_message)?;
4646

src/balancer/chunk_forwarding_session_controller/transforms_outgoing_message.rs

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -9,4 +9,8 @@ pub trait TransformsOutgoingMessage {
99
type TransformedMessage: Serialize;
1010

1111
async fn transform(&self, message: OutgoingMessage) -> Result<Self::TransformedMessage>;
12+
13+
fn stringify(&self, message: &Self::TransformedMessage) -> Result<String> {
14+
Ok(serde_json::to_string(message)?)
15+
}
1216
}

src/balancer/compatibility/openai_service/http_route/post_chat_completions.rs

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -143,9 +143,13 @@ impl TransformsOutgoingMessage for OpenAICombinedResponseTransfomer {
143143
_ => Err(anyhow!("Unexpected message type: {:?}", message)),
144144
}
145145
}
146+
147+
fn stringify(&self, message: &Self::TransformedMessage) -> anyhow::Result<String> {
148+
Ok(message.clone())
149+
}
146150
}
147151

148-
#[post("/v1/chat_completions")]
152+
#[post("/v1/chat/completions")]
149153
async fn respond(
150154
app_data: web::Data<AppData>,
151155
openai_params: web::Json<OpenAICompletionRequestParams>,

0 commit comments

Comments
 (0)