googleapis / googleapis/google-cloud-python

Using an Agent in gemini enterprise whenever submit a StreamAssistRequest

Open
#16,019 0 comments 0 reactions 0 assignees View on GitHub
api: discoveryengine type: feature request
Dominant language
Python
Stars
5.4k
Forks
1.8k
Avg merge
3d 4h
Merged PRs (30d)
122

Description

### Determine this is the right repository

- [x] 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

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.