
    yj\                         U d Z ddlmZ ddlZddlmZ ddlmZ ddl	m
Z
mZ ddlmZ dZeed	<   d
Zedz  ed<   dZeee         z  dz  ed<   dZeee         z  dz  ed<   ej        j        ZddZddZdS )zuadd messageembedding table

Revision ID: 917195d9b5e9
Revises: d429de0e5338
Create Date: 2024-01-01 12:00:00.000000

    )SequenceN)op)Vector)index_existstable_exists)settings917195d9b5e9revisiond429de0e5338down_revisionbranch_labels
depends_onreturnc                     t          j        dt          j        dt          j                    t          j                    d          t          j        dt          j                    d          t          j        dt          d          d          t          j        dt          j                    d          t          j        d	t          j                    d          t          j        d
t          j                    d          t          j        dt          j                    d          t          j        dt          j        d          dt          j	        
                                          t          j        dgt           dg          t          j        d	gt           dg          t          j        d
d	gt           dt           dg          t          j        dd	gt           dt           dg          t                     t          j        dddgt                     t          j        ddd	gt                     t          j        ddd
gt                     t          j        dddgt                     t          j        dddgt                     t          j        dt           d           d S )Nmessage_embeddingsidF)nullablecontent	embeddingi   
message_idworkspace_namesession_nameT	peer_name
created_at)timezone)r   server_defaultz.messages.public_idz.workspaces.namez.sessions.namez.sessions.workspace_namez.peers.namez.peers.workspace_nameschema!idx_message_embeddings_message_id%idx_message_embeddings_workspace_name#idx_message_embeddings_session_name idx_message_embeddings_peer_name!idx_message_embeddings_created_atzG
        CREATE INDEX idx_message_embeddings_embedding_hnsw
        ON zu.message_embeddings
        USING hnsw (embedding vector_cosine_ops)
        WITH (m = 16, ef_construction = 64)
    )r   create_tablesaColumn
BigIntegerIdentityTextr   DateTimefuncnowForeignKeyConstraintr   create_indexexecute     c/DATA/AppData/hermes/projects/honcho/migrations/versions/917195d9b5e9_add_messageembedding_table.pyupgrader3      s   O
	$GGG
	)RWYY777
	+vd||e<<<
	,		E:::
	"BGII>>>
	."'))d;;;
	+rwyy5999
	K&&&7;;==		
 	
 	
 	F1O1O1O0PQQ
!1 25P5P5P4QRR
-.&&&6(K(K(KL	
 	
 	*+###%E%E%EF	
 	
 5   < O+		    O/		    O-		    O*		    O+		    J    	 	 	 	 	r1   c                     t          j        t          j                              } t	          d|           sd S g d}|D ]/}t          d||           rt          j        |dt                     0t          j        dt                     d S )Nr   )%idx_message_embeddings_embedding_hnswr   r    r!   r"   r#   )
table_namer   r   )	r%   inspectr   get_bindr   r   
drop_indexr   
drop_table)	inspectorindexes_to_drop
index_names      r2   	downgrader>   a   s    
2;==))I,i88   O & V V
,j)DD 	VM*1EfUUUU M&v666666r1   )r   N)__doc__collections.abcr   
sqlalchemyr%   alembicr   pgvector.sqlalchemyr   migrations.utilsr   r   
src.configr   r
   str__annotations__r   r   r   DBSCHEMAr   r3   r>   r0   r1   r2   <module>rJ      s%     % $ $ $ $ $           & & & & & & 7 7 7 7 7 7 7 7       #   *sTz * * *,0sXc]"T) 0 0 0)-
C(3-$& - - -		D	 D	 D	 D	N7 7 7 7 7 7r1   