o
    2gh                     @   s   d dl Z d dlmZmZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ edd e	e d	d
Ze Zejedgdgdgd G dd deZeddefddZeddd ZeddefddZeddd ZdS )    N)FastAPIHTTPException)CORSMiddleware)	BaseModel)AsyncOpenAI)load_dotenvT)overrideOPENAI_API_KEY)api_key*)allow_originsallow_methodsallow_headersc                   @   s   e Zd ZU ee ed< dS )ChatReqmessagesN)__name__
__module____qualname__listdict__annotations__ r   r    /home/air/goalskill/back/main.pyr      s   
 r   z/goalskill/back/gpt-chatreqc              
      sl   zt jjjd| jddI d H }d|jd jjiW S  ty5 } zdd l	}|
  tdt|dd }~ww )Nzgpt-3.5-turbogffffff?)modelr   temperaturereplyr   i  )status_codedetail)openai_clientchatcompletionscreater   choicesmessagecontent	Exception	traceback	print_excr   str)r   reser'   r   r   r   gpt_chat   s   r,   z/goalskill/back/conversationsc                      s   ddddddgS )N   u   자기소개 테스트)idtitle   u   회사추천 테스트r   r   r   r   r   list_conversations,   s   r1   z/goalskill/back/save_transcriptpayloadc                    
   ddiS )Nstatussavedr   )r2   r   r   r   save_transcript5   s   r6   z/goalskill/back/healthc                      r3   )Nr4   okr   r   r   r   r   health<   s   r8   )osfastapir   r   fastapi.middleware.corsr   pydanticr   openair   dotenvr   getenvr   appadd_middlewarer   postr,   getr1   r   r6   r8   r   r   r   r   <module>   s0    

