Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 12 additions & 0 deletions crates/owhisper-interface/src/stream.rs
Original file line number Diff line number Diff line change
Expand Up @@ -191,6 +191,18 @@ impl StreamResponse {
}
}

pub fn set_provider(&mut self, provider: &str) {
if let StreamResponse::TranscriptResponse { metadata, .. } = self {
let entry = metadata
.extra
.get_or_insert_with(std::collections::HashMap::new);
entry.insert(
"provider".to_string(),
serde_json::Value::String(provider.to_string()),
);
}
}

pub fn remap_channel_index(&mut self, from: i32, to: i32) {
if let StreamResponse::TranscriptResponse { channel_index, .. } = self
&& !channel_index.is_empty()
Expand Down
7 changes: 6 additions & 1 deletion crates/transcribe-proxy/src/routes/streaming/hyprnote.rs
Original file line number Diff line number Diff line change
Expand Up @@ -78,8 +78,9 @@ fn build_response_transformer(
provider: Provider,
) -> impl Fn(&str) -> Option<String> + Send + Sync + 'static {
let mistral_adapter = MistralAdapter::default();
let provider_name = format!("{:?}", provider).to_lowercase();
move |raw: &str| {
let responses: Vec<owhisper_interface::stream::StreamResponse> = match provider {
let mut responses: Vec<owhisper_interface::stream::StreamResponse> = match provider {
Provider::Deepgram => DeepgramAdapter.parse_response(raw),
Provider::AssemblyAI => AssemblyAIAdapter.parse_response(raw),
Provider::Soniox => SonioxAdapter.parse_response(raw),
Expand All @@ -95,6 +96,10 @@ fn build_response_transformer(
return None;
}

for response in &mut responses {
response.set_provider(&provider_name);
}

if responses.len() == 1 {
return serde_json::to_string(&responses[0]).ok();
}
Expand Down
Loading