Skip to content
Draft
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
14 changes: 14 additions & 0 deletions Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

3 changes: 2 additions & 1 deletion crates/devolutions-gateway-ai/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,8 @@ pub enum Provider {
}

impl Provider {
fn default_base_url(self) -> Option<&'static str> {
/// Base URL used when none is set on the builder.
pub fn default_base_url(self) -> Option<&'static str> {
match self {
Self::OpenAi => Some("https://api.openai.com/v1/"),
Self::Anthropic => Some("https://api.anthropic.com/v1/"),
Expand Down
21 changes: 21 additions & 0 deletions crates/job-queue-libsql/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -402,6 +402,27 @@ impl JobQueue for LibSqlJobQueue {

Ok(Some(scheduled_for))
}

async fn job_defs(&self, name: &str) -> anyhow::Result<Vec<String>> {
let sql_query = "SELECT json(def) FROM job_queue WHERE name = $1";
let params = [name];

trace!(%sql_query, ?params, "Listing job definitions");

let mut rows = self
.conn
.query(sql_query, params)
.await
.context("failed to execute SQL query")?;

let mut defs = Vec::new();

while let Some(row) = rows.next().await.context("failed to read the row")? {
defs.push(row.get::<String>(0).context("failed to read def value")?);
}

Ok(defs)
}
}

// Typically, migrations should not be modified once released, and we should only be appending to this list.
Expand Down
3 changes: 3 additions & 0 deletions crates/job-queue/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,9 @@ pub trait JobQueue: Send + Sync {

/// Retrieves the closest future scheduled date
async fn next_scheduled_date(&self) -> anyhow::Result<Option<OffsetDateTime>>;

/// Returns the JSON definition of every job with this name still in the queue
async fn job_defs(&self, name: &str) -> anyhow::Result<Vec<String>>;
}

pub struct JobCtx {
Expand Down
20 changes: 20 additions & 0 deletions crates/provisioner-task-store-libsql/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
[package]
name = "provisioner-task-store-libsql"
version = "0.0.0"
edition = "2024"
authors = ["Devolutions Inc. <infos@devolutions.net>"]
publish = false

[lints]
workspace = true

[dependencies]
anyhow = "1"
libsql = { version = "0.9", default-features = false, features = ["core"] }
tracing = "0.1"
uuid = "1.23"

[dev-dependencies]
tempfile = "3.24"
tokio = { version = "1", features = ["time", "macros", "rt", "rt-multi-thread"] }
uuid = { version = "1.23", features = ["v4"] }
Loading
Loading