agent_workers_oss proxy file

Author: diegoimbert

backend/windmill-api/src/agent_workers_oss.rs

@@ -0,0 +1,31 @@
+/*
+ * Author: Ruben Fiszel
+ * Copyright: Windmill Labs, Inc 2042
+ * This file and its contents are licensed under the AGPLv3 License.
+ * Please see the included NOTICE for copyright information and
+ * LICENSE-AGPL for a copy of the license.
+ */
+
+use crate::db::DB;
+
+use axum::Router;
+
+use serde::{Deserialize, Serialize};
+
+pub fn global_service() -> Router {
+    crate::agent_workers_ee::global_service()
+}
+
+pub fn workspaced_service(
+    db: DB,
+    _base_internal_url: String,
+) -> (
+    Router,
+    Vec<tokio::task::JoinHandle<()>>,
+    Option<windmill_worker::JobCompletedSender>,
+) {
+    crate::agent_workers_ee::workspaced_service()
+}
+
+pub use crate::agent_workers_ee::AgentAuth;
+pub use crate::agent_workers_ee::AgentCache;

backend/windmill-api/src/lib.rs

@@ -28,7 +28,7 @@ use crate::{
 };
 
 #[cfg(feature = "agent_worker_server")]
-use agent_workers_ee::AgentCache;
+use agent_workers_oss::AgentCache;
 
 use anyhow::Context;
 use argon2::Argon2;
@@ -62,7 +62,9 @@ use crate::scim_ee::has_scim_token;
 use windmill_common::error::AppError;
 
 #[cfg(feature = "agent_worker_server")]
-mod agent_workers_ee;
+mod agent_workers_oss;
+#[cfg(feature = "agent_worker_server")]
+mod agent_workers_oss;
 mod ai;
 mod apps;
 pub mod args;
@@ -497,7 +499,7 @@ pub async fn run_server(
     #[cfg(feature = "agent_worker_server")]
     let (agent_workers_router, agent_workers_bg_processor, agent_workers_killpill_tx) =
         if server_mode {
-            agent_workers_ee::workspaced_service(db.clone(), _base_internal_url.clone())
+            agent_workers_oss::workspaced_service(db.clone(), _base_internal_url.clone())
         } else {
             (Router::new(), vec![], None)
         };
@@ -621,7 +623,7 @@ pub async fn run_server(
                 .nest("/agent_workers", {
                     #[cfg(feature = "agent_worker_server")]
                     {
-                        agent_workers_ee::global_service().layer(Extension(agent_cache.clone()))
+                        agent_workers_oss::global_service().layer(Extension(agent_cache.clone()))
                     }
                     #[cfg(not(feature = "agent_worker_server"))]
                     {