o
    &i_                     @   s&   d dl mZ d dlZdZdddZdS )    )OpenAINz#sk-e108ff73c53144379d0b8a4aefbaf525  c                 C   s   t tdd}|jjjdddddt| dg|d}|jd	 jj	 }|
d
r>|dd  }|dr9|d d }|	 }n|
drY|dd  }|drT|d d }|	 }n|}zt|}t| }|rn|d	 W S |W S  tjy|   | Y S w )Nzhttps://api.deepseek.com)api_keybase_urlzdeepseek-chatsystemz*You are an AI doctor who knows everything.)rolecontentuser)Zmodelmessages
max_tokensr   z	```python	   z```z```json   )r   AIML_API_KEYZchatZcompletionscreatestrchoicesmessager   strip
startswithendswithjsonloadslistvaluesJSONDecodeError)r   r   clientresponser   Zcleaned_strdatar    r   L/Users/harjaapdhillon/Desktop/newidea/medvista_backend/Services/ai_worker.pychatgpt_call   sD   







r!   )r   )Zopenair   r   r   r!   r   r   r   r    <module>   s    