!idZddlZddlZddlmZ ddlmZddlmZm Z m Z ddl m Z dej fd Zdeefd Zd edefd Zy#e $rZejdedZ[wwxYw) a Web search service using Google Gemini via Vertex AI with search grounding. Uses the Google GenAI SDK (Vertex AI) to perform a grounded search query and returns a summarised result that the voice agent can speak aloud, plus structured source links for the Android app to display. N)logger)genai)GenerateContentConfig GoogleSearchTool)service_accountz-google-genai package is required for search: returnctjd}tjd}tjd}|r6tjj t j |dg}n0|r#tjj|dg}n tdtjd||d S) zBCreate a Vertex AI GenAI client using service account credentials.GOOGLE_CREDENTIALS_PATHGOOGLE_CREDENTIALS_JSONGOOGLE_PROJECT_IDz.https://www.googleapis.com/auth/cloud-platform)scopesz>GOOGLE_CREDENTIALS_PATH or GOOGLE_CREDENTIALS_JSON must be setTz asia-south1)vertexai credentialsprojectlocation) osgetenvr Credentialsfrom_service_account_infojsonloadsfrom_service_account_file ValueErrorrClient) creds_path creds_json project_idcredss M/Users/thippareddysaicharanreddy/Desktop/agent_all/services/search_service.py_get_vertex_clientr!s45J45J./J++EE JJz "DEF  ++EE DEF  YZZ <<  c g}t} t|dd}|s|S|D]}t|dd}|st|dd}|s#|D]i}t|dd}|st|dd} t|dd} | s0| |vs5|j| |j| xs| | t|d dd k |S#t$r#} t j d | Yd} ~ |Sd} ~ wwxYw) zuExtract source links from Gemini grounding metadata. Returns a list of dicts with keys: title, url, domain. candidatesNgrounding_metadatagrounding_chunksweburititledomain)r*urlr+z.[Search] Could not extract grounding sources: )setgetattraddappend Exceptionrwarning) responsesources seen_urlsr$ candidatemetadatachunkschunkr'r(r*es r _extract_sourcesr;2s GI MX|T: N#Iy*>EHX'94@FeUD1c5"-Wb13i/MM#&NN!&#"")#x"<$ $: N MGsKLL NMs)B7AB76B7;9B77 C#CC#queryc <K tjd|t}|jjj dd|t ttgdd{}|jr|jjnd }t|}tjd t|d t|d d ||dS7k#t$r0}tjd|dt|dcYd}~Sd}~wwxYww)aPerform a web search using Gemini via Vertex AI with Google Search grounding. Args: query: The search query string. Returns: dict with keys: - success (bool) - summary (str): Summarised search results ready for TTS. - sources (list[dict]): Source links with title, url, domain. - error (str, optional): Error message on failure. z[Search] Querying: zgemini-2.5-flashz2Search the web and provide a concise summary for: ) google_searchg333333?)tools temperature)modelcontentsconfigNzNo results found.z[Search] Got response (z chars, z sources)T)successsummaryr4z[Search] Error: F)rDerror)rinfor!aiomodelsgenerate_contentrrrtextstripr;lenr1rFstr)r<clientr3rEr4r:s r web_searchrP_s3 )%34#% **;;$I%Q(,.9:<  ,4==(--%%'>Q"8, -c'l^8CL>QZ[\GHH 3 's+, 3q6223sHDA.C 2C3A*C DC D)%DDDDD)__doc__rrlogururgooglergoogle.genai.typesrrr google.oauth2r ImportErrorexcrFrr!listdictr;rNrPr"r r[s  LL- ELL6*$t**Z%3C%3D%3[ FLL@FG  sA A*A%%A*