o iW@s2ddlmZddlmZddlZGdddZdS)N)settingsc@s,eZdZddZddZddZddZd S) GeminiClientcCsDttddp td}|stdtj|dtd|_d|_ dS)NGOOGLE_API_KEYz=GOOGLE_API_KEY not found in settings or environment variables)api_keyzgemini-2.5-flashzmodels/text-embedding-004) getattrrosgetenv ValueErrorgenai configureGenerativeModelmodelembedding_model)selfrr3/var/www/html/crm_dreinet/ai_agent/gemini_client.py__init__s    zGeminiClient.__init__c Cs>z |j|}|jWSty}ztd|d}~ww)z&Generates content using Gemini 1.5 ProzError generating content: N)r generate_contenttext Exceptionprint)rpromptresponseerrrrs zGeminiClient.generate_contentc CFztj|j|dd}|dWSty"}ztd|d}~ww)z'Generates embeddings for the given textretrieval_documentr content task_type embeddingzError embedding text: Nr embed_contentrrrrrresultrrrr embed_text zGeminiClient.embed_textc Cr)z Generates embeddings for a queryretrieval_queryrrzError embedding query: Nr r"rrr embed_query%r%zGeminiClient.embed_queryN)__name__ __module__ __qualname__rrr$r'rrrrrs   r)google.generativeai generativeair django.confrrrrrrrs