apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-05-06 14:22:51 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

message extraction logic in ResponsesAPIRequest

Browse source
This commit is contained in:
Musa 2026-03-10 11:35:49 -07:00
parent 546ad1b8e1
commit fdef2b9d61
No known key found for this signature in database
1 changed files with 60 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

85
crates/hermesllm/src/apis/openai_responses.rs
View file

@ -1050,39 +1050,74 @@ impl ProviderRequest for ResponsesAPIRequest {
} }
fn extract_messages_text(&self) -> String { fn extract_messages_text(&self) -> String {
fn content_items_to_text(content_items: &[InputContent]) -> String {
content_items.iter().fold(String::new(), |acc, content| {
acc + " "
+ &match content {
InputContent::InputText { text } => text.clone(),
InputContent::InputImage { .. } => "[Image]".to_string(),
InputContent::InputFile { .. } => "[File]".to_string(),
InputContent::InputAudio { .. } => "[Audio]".to_string(),
}
})
}
fn message_content_to_text(content: &MessageContent) -> String {
match content {
MessageContent::Text(text) => text.clone(),
MessageContent::Items(content_items) => content_items_to_text(content_items),
}
}
match &self.input { match &self.input {
InputParam::Text(text) => text.clone(), InputParam::Text(text) => text.clone(),
InputParam::SingleItem(item) => { InputParam::SingleItem(item) => {
// Normalize single-item input for extraction behavior parity. // Normalize single-item input for extraction behavior parity.
match item { match item {
InputItem::Message(msg) => message_content_to_text(&msg.content), InputItem::Message(msg) => {
let mut extracted = String::new();
match &msg.content {
MessageContent::Text(text) => extracted.push_str(text),
MessageContent::Items(content_items) => {
for content in content_items {
// Preserve existing behavior: each content item is prefixed with a space.
extracted.push(' ');
match content {
InputContent::InputText { text } => {
extracted.push_str(text)
}
InputContent::InputImage { .. } => {
extracted.push_str("[Image]")
}
InputContent::InputFile { .. } => {
extracted.push_str("[File]")
}
InputContent::InputAudio { .. } => {
extracted.push_str("[Audio]")
}
}
}
}
}
extracted
}
_ => String::new(), _ => String::new(),
} }
} }
InputParam::Items(items) => items.iter().fold(String::new(), |acc, item| match item { InputParam::Items(items) => {
InputItem::Message(msg) => acc + " " + &message_content_to_text(&msg.content), let mut extracted = String::new();
// Skip non-message items (references, outputs, etc.) for item in items {
_ => acc, if let InputItem::Message(msg) = item {
}), // Preserve existing behavior: each message is prefixed with a space.
extracted.push(' ');
match &msg.content {
MessageContent::Text(text) => extracted.push_str(text),
MessageContent::Items(content_items) => {
for content in content_items {
// Preserve existing behavior: each content item is prefixed with a space.
extracted.push(' ');
match content {
InputContent::InputText { text } => {
extracted.push_str(text)
}
InputContent::InputImage { .. } => {
extracted.push_str("[Image]")
}
InputContent::InputFile { .. } => {
extracted.push_str("[File]")
}
InputContent::InputAudio { .. } => {
extracted.push_str("[Audio]")
}
}
}
}
}
}
}
extracted
}
} }
} }