Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion .librarian/state.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -1960,7 +1960,7 @@ libraries:
tag_format: '{id}-v{version}'
- id: google-cloud-geminidataanalytics
version: 0.5.0
last_generated_commit: 3322511885371d2b2253f209ccc3aa60d4100cfd
last_generated_commit: cf0434f4bd20618db60ddd16a1e7db2c0dfb9158
apis:
- path: google/cloud/geminidataanalytics/v1beta
service_config: geminidataanalytics_v1beta.yaml
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -35,6 +35,7 @@
ChartOptions,
Context,
ConversationOptions,
DatasourceOptions,
ExampleQuery,
GlossaryTerm,
)
Expand Down Expand Up @@ -75,6 +76,7 @@
ChartQuery,
ChartResult,
ChatRequest,
ClientManagedResourceContext,
ConversationReference,
DataAgentContext,
DataMessage,
Expand Down Expand Up @@ -119,6 +121,7 @@
"ChartOptions",
"Context",
"ConversationOptions",
"DatasourceOptions",
"ExampleQuery",
"GlossaryTerm",
"Conversation",
Expand Down Expand Up @@ -149,6 +152,7 @@
"ChartQuery",
"ChartResult",
"ChatRequest",
"ClientManagedResourceContext",
"ConversationReference",
"DataAgentContext",
"DataMessage",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@
ChartOptions,
Context,
ConversationOptions,
DatasourceOptions,
ExampleQuery,
GlossaryTerm,
)
Expand Down Expand Up @@ -66,6 +67,7 @@
ChartQuery,
ChartResult,
ChatRequest,
ClientManagedResourceContext,
ConversationReference,
DataAgentContext,
DataMessage,
Expand Down Expand Up @@ -117,6 +119,7 @@
"ChartQuery",
"ChartResult",
"ChatRequest",
"ClientManagedResourceContext",
"Context",
"Conversation",
"ConversationOptions",
Expand All @@ -135,6 +138,7 @@
"DataQuery",
"DataResult",
"Datasource",
"DatasourceOptions",
"DatasourceReferences",
"DeleteConversationRequest",
"DeleteDataAgentRequest",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
ChartOptions,
Context,
ConversationOptions,
DatasourceOptions,
ExampleQuery,
GlossaryTerm,
)
Expand Down Expand Up @@ -53,6 +54,7 @@
ChartQuery,
ChartResult,
ChatRequest,
ClientManagedResourceContext,
ConversationReference,
DataAgentContext,
DataMessage,
Expand Down Expand Up @@ -93,6 +95,7 @@
"ChartOptions",
"Context",
"ConversationOptions",
"DatasourceOptions",
"ExampleQuery",
"GlossaryTerm",
"Conversation",
Expand Down Expand Up @@ -123,6 +126,7 @@
"ChartQuery",
"ChartResult",
"ChatRequest",
"ClientManagedResourceContext",
"ConversationReference",
"DataAgentContext",
"DataMessage",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,9 +17,10 @@

from typing import MutableMapping, MutableSequence

from google.protobuf import wrappers_pb2 # type: ignore
import proto # type: ignore

from google.cloud.geminidataanalytics_v1alpha.types import datasource
from google.cloud.geminidataanalytics_v1alpha.types import datasource as gcg_datasource

__protobuf__ = proto.module(
package="google.cloud.geminidataanalytics.v1alpha",
Expand All @@ -28,6 +29,7 @@
"ExampleQuery",
"GlossaryTerm",
"ConversationOptions",
"DatasourceOptions",
"ChartOptions",
"AnalysisOptions",
},
Expand Down Expand Up @@ -163,10 +165,10 @@ class SchemaPaths(proto.Message):
proto.STRING,
number=1,
)
datasource_references: datasource.DatasourceReferences = proto.Field(
datasource_references: gcg_datasource.DatasourceReferences = proto.Field(
proto.MESSAGE,
number=7,
message=datasource.DatasourceReferences,
message=gcg_datasource.DatasourceReferences,
)
options: "ConversationOptions" = proto.Field(
proto.MESSAGE,
Expand Down Expand Up @@ -265,6 +267,8 @@ class ConversationOptions(proto.Message):
Optional. Options for chart generation.
analysis (google.cloud.geminidataanalytics_v1alpha.types.AnalysisOptions):
Optional. Options for analysis.
datasource (google.cloud.geminidataanalytics_v1alpha.types.DatasourceOptions):
Optional. Options for datasources.
"""

chart: "ChartOptions" = proto.Field(
Expand All @@ -277,6 +281,31 @@ class ConversationOptions(proto.Message):
number=2,
message="AnalysisOptions",
)
datasource: "DatasourceOptions" = proto.Field(
proto.MESSAGE,
number=3,
message="DatasourceOptions",
)


class DatasourceOptions(proto.Message):
r"""Options for datasources configurations.

Attributes:
big_query_max_billed_bytes (google.protobuf.wrappers_pb2.Int64Value):
Optional. This option applies to datasources
that require BigQuery queries only. Limits the
bytes billed for each BQ query job. Queries that
will have bytes billed beyond this limit will
fail (without incurring a charge). If
unspecified, no limit will be applied.
"""

big_query_max_billed_bytes: wrappers_pb2.Int64Value = proto.Field(
proto.MESSAGE,
number=1,
message=wrappers_pb2.Int64Value,
)


class ChartOptions(proto.Message):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,7 @@
"ChatRequest",
"DataAgentContext",
"ConversationReference",
"ClientManagedResourceContext",
"Message",
"UserMessage",
"SystemMessage",
Expand Down Expand Up @@ -184,6 +185,14 @@ class ChatRequest(proto.Message):
this to chat with an Agent statelessly, without
managed conversation persistence.

This field is a member of `oneof`_ ``context_provider``.
client_managed_resource_context (google.cloud.geminidataanalytics_v1alpha.types.ClientManagedResourceContext):
Optional. Context with client managed
resources. Some clients may not use GDA managed
resources including conversations and agents,
instead they create and manage their own
conversations and agents resources.

This field is a member of `oneof`_ ``context_provider``.
project (str):
Optional. The Google Cloud project to be used
Expand Down Expand Up @@ -213,6 +222,12 @@ class ChatRequest(proto.Message):
oneof="context_provider",
message="DataAgentContext",
)
client_managed_resource_context: "ClientManagedResourceContext" = proto.Field(
proto.MESSAGE,
number=105,
oneof="context_provider",
message="ClientManagedResourceContext",
)
project: str = proto.Field(
proto.STRING,
number=1,
Expand Down Expand Up @@ -305,6 +320,39 @@ class ConversationReference(proto.Message):
)


class ClientManagedResourceContext(proto.Message):
r"""Context with client managed resources.
Some clients may not use GDA managed resources including
conversations and agents, instead they create and manage their
own conversations and agents resources.

Attributes:
inline_context (google.cloud.geminidataanalytics_v1alpha.types.Context):
Required. Context for the chat request. Use
this to chat without GDA API managed
conversation and agent persistence by passing
all context inline.
conversation_id (str):
Optional. The client managed conversation id.
agent_id (str):
Optional. The client managed agent id.
"""

inline_context: context.Context = proto.Field(
proto.MESSAGE,
number=1,
message=context.Context,
)
conversation_id: str = proto.Field(
proto.STRING,
number=2,
)
agent_id: str = proto.Field(
proto.STRING,
number=3,
)


class Message(proto.Message):
r"""A message from an interaction between the user and the
system.
Expand Down Expand Up @@ -502,10 +550,17 @@ class TextType(proto.Enum):
THOUGHT (2):
The text is a thinking plan generated by the
thinking tool.
PROGRESS (3):
The text is an informational message about the agent's
progress, such as a tool being invoked. This is distinct
from the agent's internal thought process (``THOUGHT``) and
the final answer to the user (``FINAL_RESPONSE``). These
messages provide insight into the agent's actions.
"""
TEXT_TYPE_UNSPECIFIED = 0
FINAL_RESPONSE = 1
THOUGHT = 2
PROGRESS = 3

parts: MutableSequence[str] = proto.RepeatedField(
proto.STRING,
Expand Down Expand Up @@ -1112,7 +1167,21 @@ class ChartResult(proto.Message):


class ErrorMessage(proto.Message):
r"""An error message.
r"""An error message from a tool call. This message is used to represent
an error that occurred while an agent was trying to use a tool. It's
important to note that not all errors are terminal. Many are
recoverable, and the agent may use the information from this error
message to self-correct and retry the tool call or try a different
approach.

For example, if a data query fails, the agent might receive an
``ErrorMessage``, analyze it, and then generate a corrected query.

Clients should be cautious about interpreting this message as a
definitive failure. It can be part of the agent's normal, iterative
process of completing a task. Surfacing these errors directly to
end-users without context (e.g., as a "hard failure") may be
misleading.

Attributes:
text (str):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@

from typing import MutableMapping, MutableSequence

from google.protobuf import struct_pb2 # type: ignore
import proto # type: ignore

from google.cloud.geminidataanalytics_v1alpha.types import (
Expand Down Expand Up @@ -318,6 +319,14 @@ class Datasource(proto.Message):
This field is a member of `oneof`_ ``reference``.
schema (google.cloud.geminidataanalytics_v1alpha.types.Schema):
Optional. The schema of the datasource.
struct_schema (google.protobuf.struct_pb2.Struct):
Optional. A struct representation of the schema. This is
populated for datasources with schemas that cannot be fully
represented by the strongly-typed ``schema`` field.

For Looker datasources, this maps to the LookmlModelExplore
type:
https://cloud.google.com/looker/docs/reference/looker-api/latest/types/LookmlModelExplore
"""

bigquery_table_reference: "BigQueryTableReference" = proto.Field(
Expand All @@ -342,6 +351,11 @@ class Datasource(proto.Message):
number=7,
message="Schema",
)
struct_schema: struct_pb2.Struct = proto.Field(
proto.MESSAGE,
number=10,
message=struct_pb2.Struct,
)


class Schema(proto.Message):
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -31,11 +31,14 @@
ChartOptions,
Context,
ConversationOptions,
DatasourceOptions,
ExampleQuery,
GlossaryTerm,
)
from .types.conversation import (
Conversation,
CreateConversationRequest,
DeleteConversationRequest,
GetConversationRequest,
ListConversationsRequest,
ListConversationsResponse,
Expand Down Expand Up @@ -64,12 +67,14 @@
ChartQuery,
ChartResult,
ChatRequest,
ClientManagedResourceContext,
ConversationReference,
DataAgentContext,
DataMessage,
DataQuery,
DataResult,
ErrorMessage,
ExampleQueries,
ListMessagesRequest,
ListMessagesResponse,
LookerQuery,
Expand Down Expand Up @@ -114,6 +119,7 @@
"ChartQuery",
"ChartResult",
"ChatRequest",
"ClientManagedResourceContext",
"Context",
"Conversation",
"ConversationOptions",
Expand All @@ -132,13 +138,17 @@
"DataQuery",
"DataResult",
"Datasource",
"DatasourceOptions",
"DatasourceReferences",
"DeleteConversationRequest",
"DeleteDataAgentRequest",
"ErrorMessage",
"ExampleQueries",
"ExampleQuery",
"Field",
"GetConversationRequest",
"GetDataAgentRequest",
"GlossaryTerm",
"ListAccessibleDataAgentsRequest",
"ListAccessibleDataAgentsResponse",
"ListConversationsRequest",
Expand Down
Loading
Loading