apps/cursor-api-go
1
0
Fork 0
mirror of https://github.com/zeke-chin/cursor-api.git synced 2025-10-05 23:46:51 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

add: 打印原始输入

Browse Source
This commit is contained in:
zeke
2024-11-26 16:36:08 +08:00
parent d6ae2b1d0e
commit 08a3d42f51
2 changed files with 32 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
rs-capi/.dockerignore Normal file
View File

@@ -0,0 +1 @@
target/

35
rs-capi/src/main.rs
View File

@@ -1,3 +1,5 @@
use axum::body::Body;
use axum::extract::Request;
use axum::{ use axum::{
http::{HeaderMap, StatusCode}, http::{HeaderMap, StatusCode},
response::{ response::{
@@ -7,15 +9,14 @@ use axum::{
routing::{get, post}, routing::{get, post},
Json, Router, Json, Router,
}; };
use std::error::Error;
use tower_http::trace::TraceLayer;
use bytes::Bytes; use bytes::Bytes;
use futures::{ use futures::{
channel::mpsc, channel::mpsc,
stream::{Stream, StreamExt}, stream::{Stream, StreamExt},
SinkExt, SinkExt,
}; };
use std::error::Error;
use tower_http::trace::TraceLayer;
// use http::HeaderName as HttpHeaderName; // use http::HeaderName as HttpHeaderName;
use regex::Regex; use regex::Regex;
use serde::Deserializer; use serde::Deserializer;
@@ -259,8 +260,34 @@ async fn main() {
// 处理聊天完成请求 // 处理聊天完成请求
async fn chat_completions( async fn chat_completions(
headers: HeaderMap, headers: HeaderMap,
Json(chat_request): Json<ChatRequest>, request: Request<Body>,
// Json(chat_request): Json<ChatRequest>,
) -> Result<Response, StatusCode> { ) -> Result<Response, StatusCode> {
// 提取并打印原始请求体
const MAX_BODY_SIZE: usize = 20 * 1024 * 1024;
let bytes = match axum::body::to_bytes(request.into_body(), MAX_BODY_SIZE).await {
Ok(bytes) => bytes,
Err(err) => {
tracing::error!("读取请求体失败: {}", err);
return Err(StatusCode::BAD_REQUEST);
}
};
// 打印原始请求体
if let Ok(body_str) = String::from_utf8(bytes.to_vec()) {
tracing::info!("原始请求体: {}", body_str);
}
// 尝试解析 JSON
let chat_request: ChatRequest = match serde_json::from_slice(&bytes) {
Ok(req) => req,
Err(err) => {
tracing::error!("JSON解析失败: {}", err);
return Err(StatusCode::BAD_REQUEST);
}
};
// 验证认证 // 验证认证
let auth_header = headers let auth_header = headers
.get("authorization") .get("authorization")