iframe-proxy | Sunbelt Computer Software

History

100 lines (87 loc) · 4.13 KB

Raw

100

# Licensed under the Apache License, Version 2.0 (the "License");

# you may not use this file except in compliance with the License.

# You may obtain a copy of the License at

# http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software

# distributed under the License is distributed on an "AS IS" BASIS,

# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

# See the License for the specific language governing permissions and

# limitations under the License.

# [START genappbuilder_search]

from google.api_core.client_options import ClientOptions

from google.cloud import discoveryengine_v1 as discoveryengine

# TODO(developer): Uncomment these variables before running the sample.

# project_id = "YOUR_PROJECT_ID"

# location = "YOUR_LOCATION" # Values: "global", "us", "eu"

# engine_id = "YOUR_APP_ID"

# search_query = "YOUR_SEARCH_QUERY"

def search_sample(

project_id: str,

location: str,

engine_id: str,

search_query: str,

) -> discoveryengine.services.search_service.pagers.SearchPager:

# For more information, refer to:

# https://cloud.google.com/generative-ai-app-builder/docs/locations#specify_a_multi-region_for_your_data_store

client_options = (

ClientOptions(api_endpoint=f"{location}-discoveryengine.googleapis.com")

if location != "global"

else None

)

# Create a client

client = discoveryengine.SearchServiceClient(client_options=client_options)

# The full resource name of the search app serving config

serving_config = f"projects/{project_id}/locations/{location}/collections/default_collection/engines/{engine_id}/servingConfigs/default_config"

# Optional - only supported for unstructured data: Configuration options for search.

# Refer to the `ContentSearchSpec` reference for all supported fields:

# https://cloud.google.com/python/docs/reference/discoveryengine/latest/google.cloud.discoveryengine_v1.types.SearchRequest.ContentSearchSpec

content_search_spec = discoveryengine.SearchRequest.ContentSearchSpec(

# For information about snippets, refer to:

# https://cloud.google.com/generative-ai-app-builder/docs/snippets

snippet_spec=discoveryengine.SearchRequest.ContentSearchSpec.SnippetSpec(

return_snippet=True

# For information about search summaries, refer to:

# https://cloud.google.com/generative-ai-app-builder/docs/get-search-summaries

summary_spec=discoveryengine.SearchRequest.ContentSearchSpec.SummarySpec(

summary_result_count=5,

include_citations=True,

ignore_adversarial_query=True,

ignore_non_summary_seeking_query=True,

model_prompt_spec=discoveryengine.SearchRequest.ContentSearchSpec.SummarySpec.ModelPromptSpec(

preamble="YOUR_CUSTOM_PROMPT"

model_spec=discoveryengine.SearchRequest.ContentSearchSpec.SummarySpec.ModelSpec(

version="stable",

)

# Refer to the `SearchRequest` reference for all supported fields:

# https://cloud.google.com/python/docs/reference/discoveryengine/latest/google.cloud.discoveryengine_v1.types.SearchRequest

request = discoveryengine.SearchRequest(

serving_config=serving_config,

query=search_query,

page_size=10,

content_search_spec=content_search_spec,

query_expansion_spec=discoveryengine.SearchRequest.QueryExpansionSpec(

condition=discoveryengine.SearchRequest.QueryExpansionSpec.Condition.AUTO,

spell_correction_spec=discoveryengine.SearchRequest.SpellCorrectionSpec(

mode=discoveryengine.SearchRequest.SpellCorrectionSpec.Mode.AUTO

# Optional: Use fine-tuned model for this request

# custom_fine_tuning_spec=discoveryengine.CustomFineTuningSpec(

# enable_search_adaptor=True

# ),

)

page_result = client.search(request)

# Handle the response

for response in page_result:

print(response)

return page_result

# [END genappbuilder_search]

Sunbelt Computer Software

PL/B Language Development and Support

Navigation Menu

Sunbelt Computer Software

PL/B Language Development and Support

FilesExpand file tree

search_sample.py

Latest commit

History

search_sample.py

File metadata and controls