o
    ֙i
                     @   s   d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	m
Z
mZ d dlmZ G dd dedd	ZG d
d deeZG dd deZG dd deZdS )    )SQLModelFieldIndex)	BaseModel)Enum)AnyOptionalDict)Vectorc                   @   sp   e Zd ZU dZedddZee ed< edddZ	e
ed< eed	d
Zeed< edddddiddidfZdS )	Documents	documentsNT)defaultprimary_keyidF)nullableuniquechunki   )sa_type	embeddingdocuments_embedding_indexivfflatlistsd   vector_cosine_ops)postgresql_usingpostgresql_withpostgresql_ops)__name__
__module____qualname____tablename__r   r   r   int__annotations__r   strr
   r   r   r   __table_args__ r%   r%   ./var/www/html/karishye-ai-python/app/models.pyr      s   
 r   T)tablec                   @   s   e Zd ZdZdZdZdS )IntentLabelgeneral_queryshoppingstate_updateN)r   r   r   GENERAL_QUERYSHOPPINGSTATE_UPDATEr%   r%   r%   r&   r(      s    r(   c                   @   s   e Zd ZU eed< eed< dS )UserMessageuser_idcontentN)r   r   r   r#   r"   r%   r%   r%   r&   r/      s   
 r/   c                   @   s.   e Zd ZU eed< eed< dZee ed< dS )AgentResponseintentresponseNupdated_state)	r   r   r   r(   r"   r#   r5   r   r	   r%   r%   r%   r&   r2   !   s   
 r2   N)sqlmodelr   r   r   pydanticr   enumr   typingr   r   r	   pgvector.sqlalchemyr
   r   r#   r(   r/   r2   r%   r%   r%   r&   <module>   s    