Skip to content

Commit fccd5ba

Browse files
committed
chore: remove transmute in favor of type from wasi crate
1 parent 4747501 commit fccd5ba

File tree

3 files changed

+4
-5
lines changed

3 files changed

+4
-5
lines changed

Cargo.lock

Lines changed: 1 addition & 0 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

llm/bedrock/Cargo.toml

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@ aws-smithy-types = { version = "1.3.1" }
1919
aws-smithy-runtime-api = "1.8.3"
2020

2121
wasi-async-runtime = "0.1.2"
22+
wasi = "0.12.1+wasi-0.2.0"
2223

2324
# To infer mime types of downloaded images before passing to bedrock
2425
infer = { version = "0.19.0", default-features = false }

llm/bedrock/src/client.rs

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,8 +18,8 @@ use golem_llm::{
1818
config::{get_config_key, get_config_key_or_none},
1919
golem::llm::llm,
2020
};
21-
use golem_rust::bindings::wasi::clocks::monotonic_clock;
2221
use log::trace;
22+
use wasi::clocks::monotonic_clock;
2323

2424
#[derive(Debug)]
2525
pub struct Bedrock {
@@ -185,10 +185,7 @@ impl AsyncSleep for WasiSleep {
185185
let nanos = duration.as_nanos() as u64;
186186
let pollable = monotonic_clock::subscribe_duration(nanos);
187187

188-
reactor
189-
.clone()
190-
.wait_for(unsafe { std::mem::transmute(pollable) })
191-
.await;
188+
reactor.clone().wait_for(pollable).await;
192189
};
193190
Sleep::new(Box::pin(UnsafeFuture::new(fut)))
194191
}

0 commit comments

Comments
 (0)