
    g6j
                        d dl mZ d dlmZmZ d dlmZ d dlmZm	Z	 d dl
mZ  eddg      Zej                  d	      dd
       Zej                  d      d        Zej                  d      dd       Zy)    )annotations)	APIRouterstatus)WorkspaceAgentClass)	api_errorapi_response)WorkspaceChatRequestz/workspace-agentWorkspaceAgent)prefixtagsz/chatc                |   t               j                  | j                        }|j                  d      dk(  rKt	        |j                  dt
        j                        |j                  dd      |j                  d            S t        t
        j                  |j                  dd      |j                  d            S )	Nr   errorhttp_statusmessageErrordata)status_coder   r   OK)	r   trigger_chatinputgetr   r   HTTP_502_BAD_GATEWAYr   HTTP_200_OK)bodyresults     K/var/www/pie360backend.cl/public_html/app/backend/routes/workspace_agent.pytrigger_workspace_agent_chatr      s     "//

;Fzz(w&

=&2M2MNJJy'2F#
 	

 &&

9d+ZZ     z/agentsc                 j    t               j                         } t        | j                  dg             S Nr   )r   )r   list_agentsr   r   )r   s    r   list_workspace_agentsr"      s)     "..0FVZZ344r   z/agents/{agent_id}/filesc                j    t               j                  |       }t        |j                  d            S r    )r   
list_filesr   r   )agent_idr   s     r   list_workspace_agent_filesr&   %   s)     "--h7FVZZ/00r   N)r   r	   )r%   str)
__future__r   fastapir   r   )app.backend.classes.workspace_agent_classr   app.backend.core.responsesr   r   #app.backend.schemas.workspace_agentr	   workspace_agentpostr   r   r"   r&    r   r   <module>r0      s    " % I > D
	 g  Y5  5
 /01 11r   