Increase req body limit (#2965)
force merging this since it has been going back and forth for so long
This commit is contained in:
@@ -1,7 +1,7 @@
|
|||||||
use super::utils::verify_secret_key;
|
use super::utils::verify_secret_key;
|
||||||
use crate::state::AppState;
|
use crate::state::AppState;
|
||||||
use axum::{
|
use axum::{
|
||||||
extract::State,
|
extract::{DefaultBodyLimit, State},
|
||||||
http::{self, HeaderMap, StatusCode},
|
http::{self, HeaderMap, StatusCode},
|
||||||
response::IntoResponse,
|
response::IntoResponse,
|
||||||
routing::post,
|
routing::post,
|
||||||
@@ -400,9 +400,15 @@ async fn submit_tool_result(
|
|||||||
|
|
||||||
pub fn routes(state: Arc<AppState>) -> Router {
|
pub fn routes(state: Arc<AppState>) -> Router {
|
||||||
Router::new()
|
Router::new()
|
||||||
.route("/reply", post(reply_handler))
|
.route(
|
||||||
|
"/reply",
|
||||||
|
post(reply_handler).layer(DefaultBodyLimit::max(50 * 1024 * 1024)),
|
||||||
|
)
|
||||||
.route("/confirm", post(confirm_permission))
|
.route("/confirm", post(confirm_permission))
|
||||||
.route("/tool_result", post(submit_tool_result))
|
.route(
|
||||||
|
"/tool_result",
|
||||||
|
post(submit_tool_result).layer(DefaultBodyLimit::max(10 * 1024 * 1024)),
|
||||||
|
)
|
||||||
.with_state(state)
|
.with_state(state)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -353,6 +353,7 @@ impl From<PromptMessage> for Message {
|
|||||||
pub struct Message {
|
pub struct Message {
|
||||||
pub id: Option<String>,
|
pub id: Option<String>,
|
||||||
pub role: Role,
|
pub role: Role,
|
||||||
|
#[serde(default = "default_created")]
|
||||||
pub created: i64,
|
pub created: i64,
|
||||||
pub content: Vec<MessageContent>,
|
pub content: Vec<MessageContent>,
|
||||||
}
|
}
|
||||||
@@ -390,6 +391,10 @@ pub fn push_message(messages: &mut Vec<Message>, message: Message) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn default_created() -> i64 {
|
||||||
|
0 // old messages do not have timestamps.
|
||||||
|
}
|
||||||
|
|
||||||
impl Message {
|
impl Message {
|
||||||
pub fn new(role: Role, created: i64, content: Vec<MessageContent>) -> Self {
|
pub fn new(role: Role, created: i64, content: Vec<MessageContent>) -> Self {
|
||||||
Message {
|
Message {
|
||||||
|
|||||||
Reference in New Issue
Block a user