rh6 &UddlmZddlZddlmZddlZddlmZmZmZm Z m Z ddl m Z ddl mcmZddlmZddlmZmZddlmZdd lmZdd lmZdd lmZdd lmZdd lmZdZej>j@Z ee!e"e fZ#ie jHe jHde jHde jHde jHe jJe jJde jJde jJde jJe jLe jLde jLde jLde jLe jNe jNde jNde jNde jNZ(de)d<d'dZ*eejVe,e"ejZfejZffZ.eej^ee.e e"ejZffZ0d(dZ1ee"ejdejdejfejffZ4Gdde Z5ee4e5ejlfZ7d)d Z8 d*d!Z9edddddd" d+d#Z:edddddddd$ d,d%Z;edddddddd$ d,d&Z) r!r SemanticRetrieverConfigrJdictr-r.r/r"rr5)r6r>rIs r%_make_semantic_retriever_configrNs&&889 !& )D D! FD !1&2BCx"6l3345%h (  gw F7OS )'226'?Cw  ) )& 11r')modelinline_passagessemantic_retriever answer_stylesafety_settings temperaturec ftj|}tj|}|rt j |}||t d|d|d| t|}n$|t||d}ntd|d|d|r t|}tj|||||||S)a constructs a protos.GenerateAnswerRequest object by organizing the input parameters for the API call to generate a grounded answer from the model. Args: model: Name of the model used to generate the grounded response. contents: Content of the current conversation with the model. For single-turn query, this is a single question to answer. For multi-turn queries, this is a repeated field that contains conversation history and the last `Content` in the list containing the question. inline_passages: Grounding passages (a list of `Content`-like objects or `(id, content)` pairs, or a `protos.GroundingPassages`) to send inline with the request. Exclusive with `semantic_retriever`, one must be set, but not both. semantic_retriever: A Corpus, Document, or `protos.SemanticRetrieverConfig` to use for grounding. Exclusive with `inline_passages`, one must be set, but not both. answer_style: Style for grounded answers. safety_settings: Safety settings for generated output. temperature: The temperature for randomness in the output. Returns: Call for protos.GenerateAnswerRequest(). zInvalid configuration: Please set either 'inline_passages' or 'semantic_retriever_config', but not both. Received for inline_passages: z, and for semantic_retriever: .zInvalid configuration: Either 'inline_passages' or 'semantic_retriever_config' must be provided, but currently both are 'None'. Received for inline_passages: rOcontentsrPrQrSrTrR) r make_model_namer to_contentsrnormalize_safety_settings ValueErrorr9rNr-r&r GenerateAnswerRequest)rOrYrPrQrRrSrTs r%_make_generate_answer_requestr_s<  ' ' .E((2H&@@Q"'9'E--<,==[\n[oop r    $2?C  '<=OQYZ\Q]^--<,==[\n[oop r  &|4  ' ''-'! r')rOrPrQrRrSrTclientrequest_optionsc p|i}| t}t|||||||} |j| fi|} | S)aCalls the GenerateAnswer API and returns a `types.Answer` containing the response. You can pass a literal list of text chunks: >>> from google.generativeai import answer >>> answer.generate_answer( ... content=question, ... inline_passages=splitter.split(document) ... ) Or pass a reference to a retreiver Document or Corpus: >>> from google.generativeai import answer >>> from google.generativeai import retriever >>> my_corpus = retriever.get_corpus('my_corpus') >>> genai.generate_answer( ... content=question, ... semantic_retriever=my_corpus ... ) Args: model: Which model to call, as a string or a `types.Model`. contents: The question to be answered by the model, grounded in the provided source. inline_passages: Grounding passages (a list of `Content`-like objects or (id, content) pairs, or a `protos.GroundingPassages`) to send inline with the request. Exclusive with `semantic_retriever`, one must be set, but not both. semantic_retriever: A Corpus, Document, or `protos.SemanticRetrieverConfig` to use for grounding. Exclusive with `inline_passages`, one must be set, but not both. answer_style: Style in which the grounded answer should be returned. safety_settings: Safety settings for generated output. Defaults to None. temperature: Controls the randomness of the output. client: If you're not relying on a default client, you pass a `glm.GenerativeServiceClient` instead. request_options: Options for the request. Returns: A `types.Answer` containing the model's text answer response. rX)r r_generate_answer rOrYrPrQrRrSrTr`rarequestresponses r%rcrcsZf ~.0+'-'!G&v%%gAAH Or'c K|i}| t}t|||||||} |j| fi|d{} | S7w)a^ Calls the API and returns a `types.Answer` containing the answer. Args: model: Which model to call, as a string or a `types.Model`. contents: The question to be answered by the model, grounded in the provided source. inline_passages: Grounding passages (a list of `Content`-like objects or (id, content) pairs, or a `protos.GroundingPassages`) to send inline with the request. Exclusive with `semantic_retriever`, one must be set, but not both. semantic_retriever: A Corpus, Document, or `protos.SemanticRetrieverConfig` to use for grounding. Exclusive with `inline_passages`, one must be set, but not both. answer_style: Style in which the grounded answer should be returned. safety_settings: Safety settings for generated output. Defaults to None. temperature: Controls the randomness of the output. client: If you're not relying on a default client, you pass a `glm.GenerativeServiceClient` instead. Returns: A `types.Answer` containing the model's text answer response. NrX)r r_rcrds r%generate_answer_asyncrh9sg@ ~46+'-'!G,V++GGGGH OHs9AAA)r$AnswerStyleOptionsreturn AnswerStyle)r6GroundingPassagesOptionsrjzprotos.GroundingPassages)rjz str | None)r6SemanticRetrieverConfigOptionsr>r=rjzprotos.SemanticRetrieverConfig)rOmodel_types.AnyModelNameOptionsrYr=rPGroundingPassagesOptions | NonerQ%SemanticRetrieverConfigOptions | NonerRAnswerStyle | NonerS(safety_types.SafetySettingOptions | NonerT float | Nonerjzprotos.GenerateAnswerRequest)rOrnrYr=rProrQrprRrqrSrrrTrsr`z"glm.GenerativeServiceClient | Noneraz&helper_types.RequestOptionsType | None)= __future__r dataclassescollections.abcr itertoolstypingrrrrtyping_extensionsr google.ai.generativelanguageaigenerativelanguageglmgoogle.generativeair google.generativeai.clientr r google.generativeai.typesr rrrr)google.generativeai.types.retriever_typesrDEFAULT_ANSWER_MODELr^rkintr"riANSWER_STYLE_UNSPECIFIED ABSTRACTIVE EXTRACTIVEVERBOSErrDr&r2r4 ContentTypeGroundingPassageOptionsr,rlr9rGrHr<r;rLrmrJrNr_rcrhrEr'r%rs#$::'**&22235D#**66 3[019((+*N*N9{++9 D D9;77 9 [44 9 { 9 7 79;**9K229{9{559+((9,,9{9K//9 {""!95( sM,E,E'E!F HaHaa !  $% C** *+- 7F  0H0H&//Y -)-"' ""$"2 *2 %2$2:.B7;@D'+@D $? *?)?5 ? > ? % ?>??"?H.B7;@D'+@D $15>BE *E)E5 E > E % E>EE /E<ET.B7;@D'+@D $15>B2 *2)25 2 > 2 % 2>22 /2<2r'