wehub-resource-sync
81 lines
2.3 KiB
Python
81 lines
2.3 KiB
Python
import uuid
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from services.chat.chat_memory_store import CHAT_MEMORY_STORE
|
|
from services.chat import sql_chat_history
|
|
|
|
|
|
class ChatConversationStore:
|
|
def __init__(self, sql_session: AsyncSession):
|
|
self._sql = sql_session
|
|
|
|
async def load_history(
|
|
self,
|
|
*,
|
|
presentation_id: uuid.UUID,
|
|
conversation_id: uuid.UUID,
|
|
) -> list[dict[str, str]]:
|
|
messages = await sql_chat_history.load_messages(
|
|
self._sql,
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
)
|
|
if messages:
|
|
return messages
|
|
legacy = await CHAT_MEMORY_STORE.load_history(
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
)
|
|
if legacy:
|
|
await sql_chat_history.replace_messages(
|
|
self._sql,
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
messages=legacy,
|
|
)
|
|
return legacy
|
|
|
|
async def append_turn(
|
|
self,
|
|
*,
|
|
presentation_id: uuid.UUID,
|
|
conversation_id: uuid.UUID,
|
|
user_message: str,
|
|
assistant_message: str,
|
|
tool_calls: list[str] | None = None,
|
|
) -> None:
|
|
await sql_chat_history.append_turn(
|
|
self._sql,
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
user_message=user_message,
|
|
assistant_message=assistant_message,
|
|
tool_calls=tool_calls,
|
|
)
|
|
await CHAT_MEMORY_STORE.store_chat_turn(
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
user_message=user_message,
|
|
assistant_message=assistant_message,
|
|
)
|
|
|
|
async def retrieve_semantic_context(
|
|
self,
|
|
*,
|
|
presentation_id: uuid.UUID,
|
|
conversation_id: uuid.UUID,
|
|
query: str,
|
|
) -> str:
|
|
return await CHAT_MEMORY_STORE.retrieve_context(
|
|
presentation_id=presentation_id,
|
|
conversation_id=conversation_id,
|
|
query=query,
|
|
)
|
|
|
|
async def ensure_conversation_id(
|
|
self,
|
|
conversation_id: uuid.UUID | None,
|
|
) -> uuid.UUID:
|
|
return conversation_id or uuid.uuid4()
|