o Zh@sdZddlmZmZmZddlmmZddl m Z ddl m Z ddl mZmZddlmZGd d d eZGd d d eZGd ddeZGddde eefZdS)a+Google GenerativeAI Attributed Question and Answering (AQA) service. The GenAI Semantic AQA API is a managed end to end service that allows developers to create responses grounded on specified passages based on a user query. For more information visit: https://developers.generativeai.google/guide )AnyListOptionalN)RunnableSerializable)RunnableConfig) BaseModel PrivateAttr)_genai_extensionc@s&eZdZUdZeed<eeed<dS)AqaInputzInput to `GenAIAqa.invoke`. Attributes: prompt: The user's inquiry. source_passages: A list of passage that the LLM should use only to answer the user's inquiry. promptsource_passagesN)__name__ __module__ __qualname____doc__str__annotations__rrrW/var/www/html/lang_env/lib/python3.10/site-packages/langchain_google_genai/genai_aqa.pyr s r c@s.eZdZUdZeed<eeed<eed<dS) AqaOutputaDOutput from `GenAIAqa.invoke`. Attributes: answer: The answer to the user's inquiry. attributed_passages: A list of passages that the LLM used to construct the answer. answerable_probability: The probability of the question being answered from the provided passages. answerattributed_passagesanswerable_probabilityN)rrrrrrrfloatrrrrr s   rc seZdZUdZeZejed<eZ e ed<eZ e ej ed<eZeeed<ejjjgdfde de ej d eed ed df fd d Zdede ed ejfddZZS) _AqaModelz(Wrapper for Google's internal AQA model._client _answer_style_safety_settings _temperatureN answer_stylesafety_settings temperaturekwargsreturnc s2tjdi|t|_||_||_||_dS)Nr)super__init__genaixZbuild_generative_servicerrrr)selfr r!r"r# __class__rrr&8s   z_AqaModel.__init__r passagescCstj|||j|j|j|jdS)N)r r+clientr r!r")r'generate_answerrrrr)r(r r+rrrr-Esz_AqaModel.generate_answer)rrrrrrgenaiZGenerativeServiceClientrrintrrZ SafetySettingrrrZGenerateAnswerRequestZ AnswerStyleZ ABSTRACTIVErr&rr'ZGroundedAnswerr- __classcell__rrr)rr0s6  rc sheZdZUdZeZeed<dZe ed<de ddffdd Z dd e d e ede defd d ZZS)GenAIAqaaGoogle's Attributed Question and Answering service. Given a user's query and a list of passages, Google's server will return a response that is grounded to the provided list of passages. It will not base the response on parametric memory. Attributes: answer_style: keyword-only argument. See `google.ai.generativelanguage.AnswerStyle` for details. rr r r#r$Nc s&tjdi|tdi||_dS)aJConstruct a Google Generative AI AQA model. All arguments are optional. Args: answer_style: See `google.ai.generativelanguage.GenerateAnswerRequest.AnswerStyle`. safety_settings: See `google.ai.generativelanguage.SafetySetting`. temperature: 0.0 to 1.0. Nr)r%r&rr)r(r#r)rrr&is zGenAIAqa.__init__inputconfigcKs6|jj|j|jd}t|jdd|jD|jpddS)z:Generates a grounded response using the provided passages.)r r+cSsg|]}|jqSr)text).0Zpassagerrr sz#GenAIAqa.invoke..g)rrr)rr-r r rrrr)r(r2r3r#responserrrinvokewszGenAIAqa.invoke)N)rrrrrrrrr r/rr&r rrrr8r0rrr)rr1Ts  r1)rtypingrrrZgoogle.ai.generativelanguageZaiZgenerativelanguager.Zlangchain_core.runnablesrZlangchain_core.runnables.configrZpydanticrrr r'r rrr1rrrrs    $