3fi)NddlZddlmZddlmZddlmZmZmZGddeZ y)N)List)BaseChatMessageHistory) BaseMessagemessage_to_dictmessages_from_dictcxeZdZdZ ddededededededd fd Zdd Zd edd fd Z e de efdZ ddZ y )XataChatMessageHistoryz/Chat message history stored in a Xata database. session_iddb_urlapi_key branch_name table_name create_tablereturnNc ddlm}|||||_||_||_|r|j yy#t$r tdwxYw)zInitialize with Xata client.r) XataClientzPCould not import xata python package. Please install it with `pip install xata`.)r r r N) xata.clientr ImportError_client _table_name _session_id_create_table_if_not_exists)selfr r r r rrrs m/var/www/auto_recruiter/arenv/lib/python3.12/site-packages/langchain_community/chat_message_histories/xata.py__init__zXataChatMessageHistory.__init__sh  . "F   &%   , , .  =  s :Ac |jjj|j}|jdkry|jdk7rt d|jd||jjj |j}|jdkDrt d|jd||jjj|jdddd d dd d dd d d d ddd ddd gi}|jdkDrt d|jd|y)N+iz(Error checking if table exists in Xata:  zError creating table in Xata: columns sessionIdstring)nametyper#rolecontenttextr"additionalKwargsjsonpayloadz$Error setting table schema in Xata: )rtable get_schemar status_code Exceptioncreate set_schema)rrs rrz2XataChatMessageHistory._create_table_if_not_exists)sJ LL   + +D,<,< = ==C   ==C :1==/1#N  LL   ' '(8(8 9 ==3 r]s) >zS3zSr2