mirror of
https://github.com/zyycn/codex-proxy-rs.git
synced 2026-11-06 16:06:08 +00:00
254 lines
8.0 KiB
Rust
254 lines
8.0 KiB
Rust
//! Conversation identity 构建器
|
||
//!
|
||
//! 从客户端提供的值派生账号作用域的 conversation 和 window ID。
|
||
//! 用于会话亲和性和提示缓存。
|
||
|
||
use serde_json::Value;
|
||
use sha2::{Digest, Sha256};
|
||
use uuid::Uuid;
|
||
|
||
use crate::codex::gateway::transport::types::CodexResponsesRequest;
|
||
|
||
const LEADING_SYSTEM_REMINDER_OPEN: &str = "<system-reminder>";
|
||
const LEADING_SYSTEM_REMINDER_CLOSE: &str = "</system-reminder>";
|
||
|
||
/// 从请求上下文派生的 conversation identity
|
||
#[derive(Debug, Clone, PartialEq, Eq)]
|
||
pub struct ConversationIdentity {
|
||
/// 账号作用域的 conversation ID（用作 session_id header）
|
||
pub conversation_id: Option<String>,
|
||
/// 账号作用域的 window ID
|
||
pub window_id: Option<String>,
|
||
}
|
||
|
||
/// 从 prompt_cache_key 和可选的 window_id 构建 conversation identity
|
||
///
|
||
/// 匹配 Node.js 实现：buildConversationIdentity() + buildAccountScopedIdentity()
|
||
pub fn build_conversation_identity(
|
||
prompt_cache_key: Option<&str>,
|
||
client_window_id: Option<&str>,
|
||
account_scope: &str,
|
||
) -> ConversationIdentity {
|
||
let conversation_id = prompt_cache_key
|
||
.filter(|s| !s.trim().is_empty())
|
||
.map(|key| build_account_scoped_identity("conversation", account_scope, key));
|
||
|
||
let window_id = if let Some(client_win) = client_window_id.filter(|s| !s.trim().is_empty()) {
|
||
Some(build_account_scoped_identity(
|
||
"window",
|
||
account_scope,
|
||
client_win,
|
||
))
|
||
} else {
|
||
conversation_id
|
||
.as_ref()
|
||
.map(|conv_id| format!("{}:0", conv_id))
|
||
};
|
||
|
||
ConversationIdentity {
|
||
conversation_id,
|
||
window_id,
|
||
}
|
||
}
|
||
|
||
/// 确保请求拥有上游可复用的 prompt cache key。
|
||
///
|
||
/// 优先保留客户端显式值；否则用模型、instructions 和首条 user 文本派生稳定 key。
|
||
/// 空请求没有稳定锚点时使用随机 UUID，避免所有空请求落到同一条缓存链。
|
||
pub fn ensure_prompt_cache_key(request: &mut CodexResponsesRequest) {
|
||
if let Some(existing) = request
|
||
.prompt_cache_key
|
||
.as_deref()
|
||
.map(str::trim)
|
||
.filter(|value| !value.is_empty())
|
||
{
|
||
request.prompt_cache_key = Some(existing.to_string());
|
||
return;
|
||
}
|
||
|
||
request.prompt_cache_key =
|
||
Some(derive_stable_conversation_key(request).unwrap_or_else(|| Uuid::new_v4().to_string()));
|
||
}
|
||
|
||
/// 按原版 `stable-conversation-key.ts` 的规则派生稳定 conversation key。
|
||
pub fn derive_stable_conversation_key(request: &CodexResponsesRequest) -> Option<String> {
|
||
let instructions = request.instructions.chars().take(2000).collect::<String>();
|
||
let first_user_text = first_user_text(&request.input);
|
||
let normalized_first_user_text = normalize_conversation_anchor_text(&first_user_text);
|
||
let first_user_text = if normalized_first_user_text.is_empty() {
|
||
first_user_text
|
||
} else {
|
||
normalized_first_user_text
|
||
};
|
||
if instructions.is_empty() && first_user_text.is_empty() {
|
||
return None;
|
||
}
|
||
|
||
let mut hasher = Sha256::new();
|
||
hasher.update(request.model.as_bytes());
|
||
hasher.update(b"\0");
|
||
hasher.update(instructions.as_bytes());
|
||
hasher.update(b"\0");
|
||
hasher.update(first_user_text.as_bytes());
|
||
let hex = hex::encode(hasher.finalize());
|
||
|
||
Some(format!(
|
||
"{}-{}-{}-{}-{}",
|
||
&hex[0..8],
|
||
&hex[8..12],
|
||
&hex[12..16],
|
||
&hex[16..20],
|
||
&hex[20..32]
|
||
))
|
||
}
|
||
|
||
fn first_user_text(input: &[Value]) -> String {
|
||
for item in input {
|
||
if item.get("role").and_then(Value::as_str) != Some("user") {
|
||
continue;
|
||
}
|
||
let Some(content) = item.get("content") else {
|
||
return String::new();
|
||
};
|
||
if let Some(text) = content.as_str() {
|
||
return text.to_string();
|
||
}
|
||
if let Some(parts) = content.as_array() {
|
||
return parts
|
||
.iter()
|
||
.filter(|part| part.get("type").and_then(Value::as_str) == Some("input_text"))
|
||
.filter_map(|part| part.get("text").and_then(Value::as_str))
|
||
.collect::<String>();
|
||
}
|
||
return String::new();
|
||
}
|
||
|
||
String::new()
|
||
}
|
||
|
||
fn normalize_conversation_anchor_text(text: &str) -> String {
|
||
let mut rest = text.trim_start();
|
||
loop {
|
||
let lower = rest.to_ascii_lowercase();
|
||
if !lower.starts_with(LEADING_SYSTEM_REMINDER_OPEN) {
|
||
break;
|
||
}
|
||
let Some(close_start) = lower.find(LEADING_SYSTEM_REMINDER_CLOSE) else {
|
||
break;
|
||
};
|
||
let close_end = close_start + LEADING_SYSTEM_REMINDER_CLOSE.len();
|
||
rest = rest[close_end..].trim_start();
|
||
}
|
||
rest.to_string()
|
||
}
|
||
|
||
/// 构建账号作用域的身份哈希
|
||
///
|
||
/// 格式：`{prefix}_{sha256(kind + "\0" + account_scope + "\0" + client_value)[..32]}`
|
||
/// 其中 prefix 为 "cp"（conversation）或 "cw"（window）
|
||
fn build_account_scoped_identity(kind: &str, account_scope: &str, client_value: &str) -> String {
|
||
let mut hasher = Sha256::new();
|
||
hasher.update(kind.as_bytes());
|
||
hasher.update(b"\0");
|
||
hasher.update(account_scope.as_bytes());
|
||
hasher.update(b"\0");
|
||
hasher.update(client_value.as_bytes());
|
||
|
||
let digest = hasher.finalize();
|
||
let hex = hex::encode(digest);
|
||
let truncated = &hex[..32];
|
||
|
||
let prefix = match kind {
|
||
"conversation" => "cp",
|
||
"window" => "cw",
|
||
_ => "cx",
|
||
};
|
||
|
||
format!("{}_{}", prefix, truncated)
|
||
}
|
||
|
||
#[cfg(test)]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn test_build_conversation_identity_full() {
|
||
let identity = build_conversation_identity(
|
||
Some("client_conv_123"),
|
||
Some("client_win_456"),
|
||
"entry_789",
|
||
);
|
||
|
||
assert!(identity.conversation_id.is_some());
|
||
assert!(identity.window_id.is_some());
|
||
|
||
let conv_id = identity.conversation_id.unwrap();
|
||
assert!(conv_id.starts_with("cp_"));
|
||
assert_eq!(conv_id.len(), 3 + 32);
|
||
|
||
let win_id = identity.window_id.unwrap();
|
||
assert!(win_id.starts_with("cw_"));
|
||
}
|
||
|
||
#[test]
|
||
fn test_build_conversation_identity_fallback_window() {
|
||
let identity = build_conversation_identity(Some("client_conv_123"), None, "entry_789");
|
||
|
||
assert!(identity.conversation_id.is_some());
|
||
assert!(identity.window_id.is_some());
|
||
|
||
let conv_id = identity.conversation_id.as_ref().unwrap();
|
||
let win_id = identity.window_id.as_ref().unwrap();
|
||
|
||
assert_eq!(win_id, &format!("{}:0", conv_id));
|
||
}
|
||
|
||
#[test]
|
||
fn test_build_conversation_identity_empty() {
|
||
let identity = build_conversation_identity(None, None, "entry_789");
|
||
|
||
assert!(identity.conversation_id.is_none());
|
||
assert!(identity.window_id.is_none());
|
||
}
|
||
|
||
#[test]
|
||
fn test_build_account_scoped_identity_deterministic() {
|
||
let id1 = build_account_scoped_identity("conversation", "account_1", "client_key");
|
||
let id2 = build_account_scoped_identity("conversation", "account_1", "client_key");
|
||
|
||
assert_eq!(id1, id2);
|
||
assert!(id1.starts_with("cp_"));
|
||
}
|
||
|
||
#[test]
|
||
fn test_build_account_scoped_identity_different_accounts() {
|
||
let id1 = build_account_scoped_identity("conversation", "account_1", "client_key");
|
||
let id2 = build_account_scoped_identity("conversation", "account_2", "client_key");
|
||
|
||
assert_ne!(id1, id2);
|
||
}
|
||
|
||
#[test]
|
||
fn test_build_account_scoped_identity_window_prefix() {
|
||
let id = build_account_scoped_identity("window", "account_1", "win_123");
|
||
assert!(id.starts_with("cw_"));
|
||
}
|
||
|
||
#[test]
|
||
fn derive_stable_conversation_key_uses_first_user_text_after_system_reminder() {
|
||
let request = CodexResponsesRequest::new_http_sse(
|
||
"gpt-5.5",
|
||
"be brief",
|
||
vec![serde_json::json!({
|
||
"role": "user",
|
||
"content": "<system-reminder>ignore</system-reminder>\nhello"
|
||
})],
|
||
);
|
||
|
||
let key = derive_stable_conversation_key(&request).unwrap();
|
||
|
||
assert_eq!(key.len(), 36);
|
||
assert_eq!(key.chars().filter(|ch| *ch == '-').count(), 4);
|
||
}
|
||
}
|