*e~ddlmZddlZedZej de_GddZGddeZy) ) dotenv_valuesNz.envOPENAI_API_KEYceZdZdZdZy)OpenAiChatCompletionClientcd|_yN)response)selfs C/home/dictionary/smart-dictionary-consec/app/gogyup/openai_tools.py__init__z#OpenAiChatCompletionClient.__init__ s  cdd|dg}tjjd|dddd|_y)Nuser)rolecontentz gpt-3.5-turboir)modelmessages max_tokenstop_pfrequency_penaltypresence_penalty)openaiChatCompletioncreater )r promptrs r sendzOpenAiChatCompletionClient.send s>#78--44! 5  r N)__name__ __module__ __qualname__r rr r rrs   r rc&eZdZfdZddZxZS)OpenAiTranslatorc"t|yr)superr )r __class__s r r zOpenAiTranslator.__init__s r c|j}|jds|dz }dj|||}|j|y)N).?!r(aYou will be given an English sentence, and a selected word that appears in that sentence. Translate the sentence from English to {language}.If the selected word is part of an idiomatic expression, then mark the idiomatic expression instead. Write a definition in {language} of the selected word or words. Return a JSON object with the following keys: 'translation', 'definition'. English sentence: "{context}" selected word: "{word}")wordcontextlanguage)stripendswithformatr)r r+r,r-rs r translatezOpenAiTranslator.translatesO--/0 sNG( )/D'T\(]  &r )Spanish)rrr r r1 __classcell__)r&s@r r#r#s  r r#)dotenvrrconfiggetapi_keyrr#r!r r r8s= v ,-   1r