Skip to content

Commit d8361d0

Browse files
authored
fix(ai): don't check the content-length header (#523)
1 parent 62051c8 commit d8361d0

File tree

1 file changed

+11
-27
lines changed

1 file changed

+11
-27
lines changed

ext/ai/utils.rs

Lines changed: 11 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
use std::{hash::Hasher, path::PathBuf};
22

3-
use anyhow::{bail, Context};
3+
use anyhow::Context;
44
use deno_core::error::AnyError;
55
use futures::{io::AllowStdIo, StreamExt};
66
use reqwest::Url;
77
use tokio::io::AsyncWriteExt;
88
use tokio_util::compat::FuturesAsyncWriteCompatExt;
9-
use tracing::{debug, error, info, info_span, instrument, trace, Instrument};
9+
use tracing::{info, info_span, instrument, trace, Instrument};
1010
use xxhash_rust::xxh3::Xxh3;
1111

1212
#[instrument(fields(%kind, url = %url))]
@@ -92,17 +92,6 @@ pub async fn fetch_and_cache_from_url(
9292
.await
9393
.context("failed to download")?;
9494

95-
let len = resp
96-
.headers()
97-
.get(header::CONTENT_LENGTH)
98-
.map(|it| it.to_str().map_err(AnyError::new))
99-
.transpose()?
100-
.map(|it| it.parse::<usize>().map_err(AnyError::new))
101-
.transpose()?
102-
.context("invalid Content-Length header")?;
103-
104-
debug!(total_bytes = len);
105-
10695
let file = tokio::fs::File::create(&filepath)
10796
.await
10897
.context("failed to create file")?;
@@ -131,23 +120,18 @@ pub async fn fetch_and_cache_from_url(
131120
faster_hex::hex_string(&hasher.finish().to_be_bytes())
132121
};
133122

134-
if written == len as u64 {
135-
info!({ bytes_written = written, checksum = &checksum_str }, "done");
123+
info!({ bytes_written = written, checksum = &checksum_str }, "done");
136124

137-
let mut checksum_file = tokio::fs::File::create(&checksum_path)
138-
.await
139-
.context("failed to create checksum file")?;
125+
let mut checksum_file = tokio::fs::File::create(&checksum_path)
126+
.await
127+
.context("failed to create checksum file")?;
140128

141-
let _ = checksum_file
142-
.write(checksum_str.as_bytes())
143-
.await
144-
.context("failed to write checksum to file system")?;
129+
let _ = checksum_file
130+
.write(checksum_str.as_bytes())
131+
.await
132+
.context("failed to write checksum to file system")?;
145133

146-
Ok(filepath)
147-
} else {
148-
error!({ expected = len, got = written }, "bytes mismatch");
149-
bail!("error copying data to file: expected {len} length, but got {written}");
150-
}
134+
Ok(filepath)
151135
}
152136
}
153137
.instrument(span)

0 commit comments

Comments
 (0)