o
    l"i                     @   s@   d dl mZ d dlmZ dededefddZdefdd	Zd
S )    )HTTPException)database_demo
session_idrolecontentc              
      sR   zd}t j|| ||ddI dH  W dS  ty( } ztdt|dd}~ww )u   
    로봇의 대화(user 또는 ai)를 'chat_history' 테이블에 저장합니다.
    (기존 save_chat_message 함수와 유사)
    z
            INSERT INTO chat_history (session_id, role, content, created_at)
            VALUES (:session_id, :role, :content, NOW())
            )r   r   r   )valuesN  status_codedetail)r   execute	Exceptionr   str)r   r   r   querye r   ./home/air/sanwanet/gpt-api-backup/db_module.pysave_robot_chat_message	   s   r   c              
      sH   zd}t |d| iI dH W S  ty# } ztdt|dd}~ww )u   
    특정 'session_id'의 모든 대화 기록을 시간순으로 가져옵니다.
    (기존 get_chat_history 함수와 유사)
    z
            SELECT role, content
            FROM chat_history
            WHERE session_id = :session_id
            ORDER BY created_at ASC
            r   Nr   r	   )r   	fetch_allr   r   r   )r   r   r   r   r   r   get_robot_chat_history   s   r   N)fastapir   dbr   r   r   r   r   r   r   r   <module>   s   