Fix/filter audience 6703 local (#6773)

Signed-off-by: rabi <ramishra@redhat.com>
Co-authored-by: rabi <ramishra@redhat.com>
Co-authored-by: Douwe Osinga <douwe@squareup.com>
This commit is contained in:
Douwe Osinga
2026-01-28 15:49:52 -05:00
committed by GitHub
parent 14b3cd00e7
commit 2bc368948b
10 changed files with 40 additions and 43 deletions
@@ -44,12 +44,7 @@ fn format_tool_response(
match &response.tool_result {
Ok(call_result) => {
let abridged: Vec<_> = call_result
.content
.iter()
.filter(|c| c.audience().is_none_or(|a| a.contains(&Role::Assistant)))
.map(|c| c.raw.clone())
.collect();
let abridged: Vec<_> = call_result.content.iter().map(|c| c.raw.clone()).collect();
let mut tool_content = Vec::new();
let mut image_messages = Vec::new();
@@ -112,9 +107,10 @@ fn format_tool_response(
fn format_messages(messages: &[Message], image_format: &ImageFormat) -> Vec<DatabricksMessage> {
let mut result = Vec::new();
for message in messages.iter().filter(|m| m.is_agent_visible()) {
let filtered = message.agent_visible_content();
let mut converted = DatabricksMessage {
content: Value::Null,
role: match message.role {
role: match filtered.role {
Role::User => "user".to_string(),
Role::Assistant => "assistant".to_string(),
},
@@ -126,7 +122,7 @@ fn format_messages(messages: &[Message], image_format: &ImageFormat) -> Vec<Data
let mut has_tool_calls = false;
let mut has_multiple_content = false;
for content in &message.content {
for content in &filtered.content {
match content {
MessageContent::Text(text) => {
if !text.text.is_empty() {