o &i@s&ddlZddlmZGdddZdS)N)OpenAIc@sReZdZdedededefddZddZd ed efd d Zd ed efd dZ dS) LLMServicebase_urlapi_keymodeltimeout_secondscCs"||_||_||_d|_||_dS)N)rrrclientr)selfrrrrr E/home/aiteam/pcaa-dev/ai-call-quality-rag/app/services/llm_service.py__init__s  zLLMService.__init__cCs&|jdurt|j|j|jd|_|jS)N)rrtimeout)rrrrr)r r r r _get_client s zLLMService._get_clientpromptreturncCsL|jjj|jddidddd|dgdd}|jd jjp d }t |S) Ntype json_objectsystemz"You output strict valid JSON only.rolecontentuserg?)rresponse_formatmessages temperaturerz{}) rchat completionscreaterchoicesmessagerjsonloads)r rresprr r r generate_jsons  zLLMService.generate_jsoncCs<|jjj|jdddd|dgdd}|jdjjpdS) Nrz)You are a call quality advisor assistant.rrg?)rrrr)rrrrrrrr)r rr"r r r answer#s zLLMService.answerN) __name__ __module__ __qualname__strintr rdictr#r%r r r r rs   r)r openairrr r r r s