a3f61fcc0f
- DB: segments 用户输入文本（Alembic 0002） - Chat: 阶段检测/阶段提示/回复限制，编排与访谈/画像 prompts 调整 - Memoir: 忠实度检查 agent，叙事与分类等链路更新 - Core: agent 日志、Alembic 启动、LangChain/日志/配置等 - Story: time_hints；Memory 检索与相关测试 - Expo: 助手头像、会话页与消息拆分、实时会话与文案/i18n - Docs/scripts/tests: 迁移脚本、LLM JSON/记忆检索文档、新增单测
145 lines
4.9 KiB
Python
145 lines
4.9 KiB
Python
"""Durable conversation turn persistence + Redis cache sync (feature layer)."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import uuid
|
|
from datetime import datetime, timedelta, timezone
|
|
from typing import Any
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.core import redis as redis_core
|
|
from app.core.logging import get_logger
|
|
from app.features.conversation import repo
|
|
from app.features.conversation.models import ConversationMessage
|
|
from app.features.conversation.session_history import (
|
|
conversation_messages_to_redis_history,
|
|
)
|
|
|
|
logger = get_logger(__name__)
|
|
|
|
|
|
def _utc_now() -> datetime:
|
|
return datetime.now(timezone.utc)
|
|
|
|
|
|
class ConversationHistoryStore:
|
|
def __init__(self, db: AsyncSession):
|
|
self._db = db
|
|
|
|
async def load_canonical_history(
|
|
self, conversation_id: str
|
|
) -> list[dict[str, Any]]:
|
|
rows = await repo.get_conversation_messages(conversation_id, self._db)
|
|
return conversation_messages_to_redis_history(rows)
|
|
|
|
async def _touch_conversation(
|
|
self, conversation_id: str, *, occurred_at: datetime
|
|
) -> None:
|
|
conversation = await repo.get_conversation(conversation_id, self._db)
|
|
if conversation is None:
|
|
return
|
|
current = getattr(conversation, "last_message_at", None)
|
|
if current is None or current < occurred_at:
|
|
conversation.last_message_at = occurred_at
|
|
|
|
async def _sync_redis_from_db(self, conversation_id: str) -> None:
|
|
hist = await self.load_canonical_history(conversation_id)
|
|
await redis_core.redis_service.set_conversation_history(conversation_id, hist)
|
|
|
|
async def _sync_redis_best_effort(self, conversation_id: str) -> None:
|
|
try:
|
|
await self._sync_redis_from_db(conversation_id)
|
|
except Exception as exc:
|
|
logger.warning("conversation history cache sync skipped: {}", exc)
|
|
|
|
async def record_ai_only_turn(
|
|
self, conversation_id: str, responses: list[str]
|
|
) -> None:
|
|
if not responses:
|
|
return
|
|
combined = "\n\n".join(responses)
|
|
created_at = _utc_now()
|
|
msg = ConversationMessage(
|
|
id=str(uuid.uuid4()),
|
|
conversation_id=conversation_id,
|
|
role="ai",
|
|
content=combined,
|
|
message_type="text",
|
|
created_at=created_at,
|
|
)
|
|
repo.add_conversation_message(msg, self._db)
|
|
await self._touch_conversation(conversation_id, occurred_at=created_at)
|
|
await self._db.commit()
|
|
await self._sync_redis_best_effort(conversation_id)
|
|
|
|
async def record_human_ai_turn(
|
|
self,
|
|
conversation_id: str,
|
|
user_message: str,
|
|
responses: list[str],
|
|
*,
|
|
user_message_timestamp: datetime | None,
|
|
is_from_voice: bool,
|
|
voice_session_id: str | None,
|
|
audio_duration_seconds: int | None,
|
|
tts_audio_urls: list[str] | None,
|
|
segment_id: str | None,
|
|
) -> None:
|
|
if not responses:
|
|
return
|
|
human_ts = user_message_timestamp or _utc_now()
|
|
if human_ts.tzinfo is None:
|
|
human_ts = human_ts.replace(tzinfo=timezone.utc)
|
|
ai_ts = human_ts + timedelta(microseconds=1)
|
|
human_type = "audio" if is_from_voice else "text"
|
|
human = ConversationMessage(
|
|
id=str(uuid.uuid4()),
|
|
conversation_id=conversation_id,
|
|
role="human",
|
|
content=user_message,
|
|
message_type=human_type,
|
|
voice_session_id=voice_session_id,
|
|
duration_seconds=audio_duration_seconds
|
|
if audio_duration_seconds is not None and audio_duration_seconds > 0
|
|
else None,
|
|
segment_id=segment_id,
|
|
created_at=human_ts,
|
|
)
|
|
combined = "\n\n".join(responses)
|
|
ai = ConversationMessage(
|
|
id=str(uuid.uuid4()),
|
|
conversation_id=conversation_id,
|
|
role="ai",
|
|
content=combined,
|
|
message_type="text",
|
|
tts_audio_urls=tts_audio_urls if tts_audio_urls else None,
|
|
segment_id=segment_id,
|
|
created_at=ai_ts,
|
|
)
|
|
repo.add_conversation_message(human, self._db)
|
|
repo.add_conversation_message(ai, self._db)
|
|
await self._touch_conversation(conversation_id, occurred_at=ai_ts)
|
|
await self._db.commit()
|
|
await self._sync_redis_best_effort(conversation_id)
|
|
|
|
async def attach_ai_tts_audio_urls(
|
|
self,
|
|
conversation_id: str,
|
|
*,
|
|
tts_audio_urls: list[str],
|
|
segment_id: str | None = None,
|
|
) -> None:
|
|
if not tts_audio_urls:
|
|
return
|
|
row = await repo.set_latest_ai_message_tts_audio_urls(
|
|
conversation_id,
|
|
self._db,
|
|
tts_audio_urls=tts_audio_urls,
|
|
segment_id=segment_id,
|
|
)
|
|
if row is None:
|
|
return
|
|
await self._db.commit()
|
|
await self._sync_redis_best_effort(conversation_id)
|