o *e@sJddlmZddlZedZede_GdddZGdddeZdS) ) dotenv_valuesNz.envZOPENAI_API_KEYc@seZdZddZddZdS)OpenAiChatCompletionClientcCs d|_dSN)responseselfrC/home/dictionary/smart-dictionary-consec/app/gogyup/openai_tools.py__init__ s z#OpenAiChatCompletionClient.__init__cCs*d|dg}tjjd|ddddd|_dS)Nuser)Zrolecontentz gpt-3.5-turboir)modelmessagesZ max_tokensZtop_pZfrequency_penaltyZpresence_penalty)openaiZChatCompletioncreater)rpromptrrrr send s  zOpenAiChatCompletionClient.sendN)__name__ __module__ __qualname__r rrrrr rs rcs&eZdZfddZdddZZS)OpenAiTranslatorcstdSr)superr r __class__rr r szOpenAiTranslator.__init__SpanishcCs8|}|ds |d7}dj|||d}||dS)N).?!raYou 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)rrr r!rrrr translates zOpenAiTranslator.translate)r)rrrr r% __classcell__rrrr rs r)dotenvrrconfiggetZapi_keyrrrrrr s