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:
@@ -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() {
|
||||
|
||||
Reference in New Issue
Block a user