
    q\i~                        d dl mZmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZmZ d dlmZmZ d dlmZ  ed	      Z G d
 de      Z G d de      Z G d de      Zej-                  d       ee       ee      fdededefd       Zy)    )	APIRouterDependsstatus)JSONResponse)	BaseModel)Optional)get_query_routerget_task_manager)ModeQueryRouter)TaskManagerz/api/v1)prefixc                   :    e Zd ZU dZee   ed<   eed<   i Zeed<   y)ChatRequestNconversation_idmessagecontext)	__name__
__module____qualname__r   r   str__annotations__r   dict     8/home/jang/Projects/ai-researcher/src/api/routes/chat.pyr   r      s    %)OXc])LGTr   r   c                   ,    e Zd ZU eed<   eed<   eed<   y)ChatResponse
message_idcontenttrace_idN)r   r   r   r   r   r   r   r   r   r   r      s    OMMr   r   c                   ,    e Zd ZU eed<   eed<   eed<   y)ChatAsyncResponsetask_idr   estimated_secondsN)r   r   r   r   r   intr   r   r   r#   r#      s    LKr   r#   z/chatrequestquery_routertask_managerc                   K   |j                  | j                  | j                        \  }}|t        j                  k(  rt        ddddd      S |j                  | j                  || j                  j                  d      | j                         d {   }t        |j                  |j                  d	
      }t        t        j                  |j                               S 7 Ow)Nmsg_sync_123textu   这是一个同步响应)typer,   	trace_123)r   r    r!   user_id)querycomplexity_scorer/   r      )r$   r   r%   )status_coder    )assess_complexityr   r   r   SYNCr   create_taskgetr   r#   idr   r   HTTP_202_ACCEPTED
model_dump)r'   r(   r)   modescoretaskasync_responses          r   chatr?      s      00'//RKD%tyy%#-GH 
 	
 "--//"OO''	2#33	 . 
 
 +GG;;

 00"--/
 	

s   BC'C%AC'N)fastapir   r   r   fastapi.responsesr   pydanticr   typingr   src.api.dependenciesr	   r
   src.core.query_routerr   r   src.core.task_managerr   routerr   r   r#   postr?   r   r   r   <module>rI      s    . . *   C 3 -	)	$) 9 	  W !((8 9 '(8 9


 
 
r   