Skip to content

Using an Agent in gemini enterprise whenever submit a StreamAssistRequest #16019

@al-droid

Description

@al-droid

Determine this is the right repository

  • I determined this is the correct repository in which to report this feature request.

Summary of the feature request

I would like to easily be able to use an agent for my stream assist request in discovery agent, but seems to be impossible using the actual client libraries

`
from google.api_core.client_options import ClientOptions
from google.cloud import discoveryengine_v1 as discoveryengine

project_id = "myproject"
location = "eu"
engine_id = "myEngine"
search_query = "Ciao come ti chiami?"

def stream_assist_sample(
project_id: str,
location: str,
engine_id: str,
query: str,
):
client_options = (
ClientOptions(api_endpoint=f"{location}-discoveryengine.googleapis.com")
if location != "global"
else None
)

client = discoveryengine.AssistantServiceClient(client_options=client_options)

request = discoveryengine.StreamAssistRequest(
    name=client.assistant_path(
        project=project_id,
        location=location,
        collection="default_collection",
        engine=engine_id,
        assistant="default_assistant",
    ),
    query=discoveryengine.Query(text=query),
)

stream = client.stream_assist(request=request)


for response in stream:
    print(response)

stream_assist_sample(project_id, location, engine_id, search_query)`

Additional context

I'm not able to make a StreamAssistRequest specifying an agent neither from the REST API, it seems to let me choose an AgentsSpec, but it completely ignores it

Metadata

Metadata

Assignees

No one assigned

    Labels

    triage meI really want to be triaged.type: feature request‘Nice-to-have’ improvement, new feature or different behavior or design.

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions