geoffsee/open-web-agent-rs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor agent function names and streamline imports

Browse Source 
Unified the naming convention for agent functions across modules to `agent` for consistency. Adjusted relevant imports and cleaned up unused imports in `webhooks.rs` to improve readability and maintainability.
This commit is contained in:
geoffsee
2025-05-27 13:05:05 -04:00
parent 3256f254ad
commit 77d5a51e76
6 changed files with 31 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
src/handlers/webhooks.rs
View File

@@ -1,7 +1,3 @@
use crate::agents;
use crate::agents::news::news_agent;
use crate::agents::scrape::scrape_agent;
use crate::agents::search::search_agent;
use axum::response::Response;
use axum::{
body::Body, extract::Path, extract::Query, http::StatusCode, response::IntoResponse, Json,
@@ -18,7 +14,6 @@ use std::time::Duration;
use tokio::io::{AsyncBufReadExt, BufReader};
use tokio::process::Command;
use tokio::sync::Mutex;
use crate::agents::image_generator::image_generator;
// init sled
lazy_static! {
@@ -31,7 +26,6 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
let db = DB.lock().await;
match db.get(&agent_id) {
Ok(Some(data)) => {
let mut info: StreamInfo = match serde_json::from_slice(&data) {
Ok(info) => info,
Err(e) => {
@@ -40,7 +34,6 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
}
};
// Increment the call_count in the database
info.call_count += 1;
let updated_info_bytes = match serde_json::to_vec(&info) {
@@ -54,7 +47,10 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
match db.insert(&agent_id, updated_info_bytes) {
Ok(_) => {
if let Err(e) = db.flush_async().await {
tracing::error!("Failed to persist updated call_count to the database: {}", e);
tracing::error!(
"Failed to persist updated call_count to the database: {}",
e
);
return StatusCode::INTERNAL_SERVER_ERROR.into_response();
}
}
@@ -82,7 +78,7 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
}
};
if(info.call_count > 1) {
if (info.call_count > 1) {
return StatusCode::OK.into_response();
}
@@ -94,12 +90,14 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
resource,
agent_id
);
let cmd = match resource.as_str() {
"web-search" => search_agent(agent_id.as_str(), &*input).await,
"news-search" => news_agent(agent_id.as_str(), &*input).await,
"image-generator" => image_generator(agent_id.as_str(), &*input).await,
"web-scrape" => scrape_agent(agent_id.as_str(), &*input).await,
"web-search" => crate::agents::search::agent(agent_id.as_str(), &*input).await,
"news-search" => crate::agents::news::agent(agent_id.as_str(), &*input).await,
"image-generator" => {
crate::agents::image_generator::agent(agent_id.as_str(), &*input).await
}
"web-scrape" => crate::agents::scrape::agent(agent_id.as_str(), &*input).await,
_ => {
tracing::error!("Unsupported resource type: {}", resource);
return StatusCode::BAD_REQUEST.into_response();
@@ -131,7 +129,7 @@ pub async fn handle_webhooks(Path(agent_id): Path<String>) -> impl IntoResponse
.header("Connection", "keep-alive")
.header("X-Accel-Buffering", "yes")
.body(Body::from_stream(sse_stream))
.unwrap()
.unwrap();
}
Ok(None) => {
tracing::error!("Stream ID not found: {}", agent_id);
@@ -183,7 +181,6 @@ struct StreamInfo {
call_count: i32,
}
#[derive(Deserialize, Serialize, Debug)]
pub struct WebhookPostRequest {
id: String,
@@ -207,7 +204,7 @@ pub async fn handle_webhooks_post(Json(payload): Json<WebhookPostRequest>) -> im
resource: payload.resource.clone(),
payload: payload.payload,
parent: payload.parent.clone(),
call_count: 0
call_count: 0,
};
let info_bytes = match serde_json::to_vec(&info) {
@@ -232,19 +229,22 @@ pub async fn handle_webhooks_post(Json(payload): Json<WebhookPostRequest>) -> im
match db.get(&stream_id) {
Ok(Some(_)) => {
let stream_url = format!("/webhooks/{}", stream_id);
tracing::info!("Successfully created and verified stream URL: {}", stream_url);
tracing::info!(
"Successfully created and verified stream URL: {}",
stream_url
);
Json(WebhookPostResponse { stream_url }).into_response()
},
}
Ok(None) => {
tracing::error!("Failed to verify stream creation: {}", stream_id);
StatusCode::INTERNAL_SERVER_ERROR.into_response()
},
}
Err(e) => {
tracing::error!("Error verifying stream creation: {}", e);
StatusCode::INTERNAL_SERVER_ERROR.into_response()
}
}
},
}
Err(e) => {
tracing::error!("Failed to flush DB: {}", e);
StatusCode::INTERNAL_SERVER_ERROR.into_response()