Skip to content

Commit 1b1f1b3

Browse files
gcf-owl-bot[bot]parthea
gcf-owl-bot[bot]
and
parthea
authored
feat: support conversational / multi-turn search (#11613)
Co-authored-by: Owl Bot <gcf-owl-bot[bot]@users.noreply.github.com> Co-authored-by: Anthonios Partheniou <partheniou@google.com>
1 parent 85fb0fe commit 1b1f1b3

File tree

136 files changed

+14664
-144
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

136 files changed

+14664
-144
lines changed

‎packages/google-cloud-discoveryengine/docs/discoveryengine_v1/conversational_search_service.rst

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,10 @@
1+
ConversationalSearchService
2+
---------------------------------------------
3+
4+
.. automodule:: google.cloud.discoveryengine_v1.services.conversational_search_service
5+
:members:
6+
:inherited-members:
7+
8+
.. automodule:: google.cloud.discoveryengine_v1.services.conversational_search_service.pagers
9+
:members:
10+
:inherited-members:

‎packages/google-cloud-discoveryengine/docs/discoveryengine_v1/services.rst

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ Services for Google Cloud Discoveryengine v1 API
44
:maxdepth: 2
55

66
completion_service
7+
conversational_search_service
78
document_service
89
schema_service
910
search_service

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine/__init__.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@
6262
)
6363
from google.cloud.discoveryengine_v1beta.types.common import (
6464
CustomAttribute,
65+
DoubleList,
6566
Interval,
6667
UserInfo,
6768
)
@@ -162,6 +163,7 @@
162163
"UserEventServiceClient",
163164
"UserEventServiceAsyncClient",
164165
"CustomAttribute",
166+
"DoubleList",
165167
"Interval",
166168
"UserInfo",
167169
"CompleteQueryRequest",

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine/gapic_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515
#
16-
__version__ = "0.10.0" # {x-release-please-version}
16+
__version__ = "0.0.0" # {x-release-please-version}

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/__init__.py

Lines changed: 38 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,15 +22,36 @@
2222
CompletionServiceAsyncClient,
2323
CompletionServiceClient,
2424
)
25+
from .services.conversational_search_service import (
26+
ConversationalSearchServiceAsyncClient,
27+
ConversationalSearchServiceClient,
28+
)
2529
from .services.document_service import DocumentServiceAsyncClient, DocumentServiceClient
2630
from .services.schema_service import SchemaServiceAsyncClient, SchemaServiceClient
2731
from .services.search_service import SearchServiceAsyncClient, SearchServiceClient
2832
from .services.user_event_service import (
2933
UserEventServiceAsyncClient,
3034
UserEventServiceClient,
3135
)
32-
from .types.common import CustomAttribute, UserInfo
36+
from .types.common import CustomAttribute, Interval, UserInfo
3337
from .types.completion_service import CompleteQueryRequest, CompleteQueryResponse
38+
from .types.conversation import (
39+
Conversation,
40+
ConversationContext,
41+
ConversationMessage,
42+
Reply,
43+
TextInput,
44+
)
45+
from .types.conversational_search_service import (
46+
ConverseConversationRequest,
47+
ConverseConversationResponse,
48+
CreateConversationRequest,
49+
DeleteConversationRequest,
50+
GetConversationRequest,
51+
ListConversationsRequest,
52+
ListConversationsResponse,
53+
UpdateConversationRequest,
54+
)
3455
from .types.document import Document
3556
from .types.document_service import (
3657
CreateDocumentRequest,
@@ -83,6 +104,7 @@
83104

84105
__all__ = (
85106
"CompletionServiceAsyncClient",
107+
"ConversationalSearchServiceAsyncClient",
86108
"DocumentServiceAsyncClient",
87109
"SchemaServiceAsyncClient",
88110
"SearchServiceAsyncClient",
@@ -93,17 +115,26 @@
93115
"CompleteQueryResponse",
94116
"CompletionInfo",
95117
"CompletionServiceClient",
118+
"Conversation",
119+
"ConversationContext",
120+
"ConversationMessage",
121+
"ConversationalSearchServiceClient",
122+
"ConverseConversationRequest",
123+
"ConverseConversationResponse",
124+
"CreateConversationRequest",
96125
"CreateDocumentRequest",
97126
"CreateSchemaMetadata",
98127
"CreateSchemaRequest",
99128
"CustomAttribute",
129+
"DeleteConversationRequest",
100130
"DeleteDocumentRequest",
101131
"DeleteSchemaMetadata",
102132
"DeleteSchemaRequest",
103133
"Document",
104134
"DocumentInfo",
105135
"DocumentServiceClient",
106136
"GcsSource",
137+
"GetConversationRequest",
107138
"GetDocumentRequest",
108139
"GetSchemaRequest",
109140
"ImportDocumentsMetadata",
@@ -113,6 +144,9 @@
113144
"ImportUserEventsMetadata",
114145
"ImportUserEventsRequest",
115146
"ImportUserEventsResponse",
147+
"Interval",
148+
"ListConversationsRequest",
149+
"ListConversationsResponse",
116150
"ListDocumentsRequest",
117151
"ListDocumentsResponse",
118152
"ListSchemasRequest",
@@ -123,13 +157,16 @@
123157
"PurgeDocumentsMetadata",
124158
"PurgeDocumentsRequest",
125159
"PurgeDocumentsResponse",
160+
"Reply",
126161
"Schema",
127162
"SchemaServiceClient",
128163
"SearchInfo",
129164
"SearchRequest",
130165
"SearchResponse",
131166
"SearchServiceClient",
167+
"TextInput",
132168
"TransactionInfo",
169+
"UpdateConversationRequest",
133170
"UpdateDocumentRequest",
134171
"UpdateSchemaMetadata",
135172
"UpdateSchemaRequest",

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/gapic_metadata.json

Lines changed: 109 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,115 @@
3939
}
4040
}
4141
},
42+
"ConversationalSearchService": {
43+
"clients": {
44+
"grpc": {
45+
"libraryClient": "ConversationalSearchServiceClient",
46+
"rpcs": {
47+
"ConverseConversation": {
48+
"methods": [
49+
"converse_conversation"
50+
]
51+
},
52+
"CreateConversation": {
53+
"methods": [
54+
"create_conversation"
55+
]
56+
},
57+
"DeleteConversation": {
58+
"methods": [
59+
"delete_conversation"
60+
]
61+
},
62+
"GetConversation": {
63+
"methods": [
64+
"get_conversation"
65+
]
66+
},
67+
"ListConversations": {
68+
"methods": [
69+
"list_conversations"
70+
]
71+
},
72+
"UpdateConversation": {
73+
"methods": [
74+
"update_conversation"
75+
]
76+
}
77+
}
78+
},
79+
"grpc-async": {
80+
"libraryClient": "ConversationalSearchServiceAsyncClient",
81+
"rpcs": {
82+
"ConverseConversation": {
83+
"methods": [
84+
"converse_conversation"
85+
]
86+
},
87+
"CreateConversation": {
88+
"methods": [
89+
"create_conversation"
90+
]
91+
},
92+
"DeleteConversation": {
93+
"methods": [
94+
"delete_conversation"
95+
]
96+
},
97+
"GetConversation": {
98+
"methods": [
99+
"get_conversation"
100+
]
101+
},
102+
"ListConversations": {
103+
"methods": [
104+
"list_conversations"
105+
]
106+
},
107+
"UpdateConversation": {
108+
"methods": [
109+
"update_conversation"
110+
]
111+
}
112+
}
113+
},
114+
"rest": {
115+
"libraryClient": "ConversationalSearchServiceClient",
116+
"rpcs": {
117+
"ConverseConversation": {
118+
"methods": [
119+
"converse_conversation"
120+
]
121+
},
122+
"CreateConversation": {
123+
"methods": [
124+
"create_conversation"
125+
]
126+
},
127+
"DeleteConversation": {
128+
"methods": [
129+
"delete_conversation"
130+
]
131+
},
132+
"GetConversation": {
133+
"methods": [
134+
"get_conversation"
135+
]
136+
},
137+
"ListConversations": {
138+
"methods": [
139+
"list_conversations"
140+
]
141+
},
142+
"UpdateConversation": {
143+
"methods": [
144+
"update_conversation"
145+
]
146+
}
147+
}
148+
}
149+
}
150+
},
42151
"DocumentService": {
43152
"clients": {
44153
"grpc": {

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/gapic_version.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,4 +13,4 @@
1313
# See the License for the specific language governing permissions and
1414
# limitations under the License.
1515
#
16-
__version__ = "0.10.0" # {x-release-please-version}
16+
__version__ = "0.0.0" # {x-release-please-version}

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/services/completion_service/async_client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@
4242
except AttributeError: # pragma: NO COVER
4343
OptionalRetry = Union[retries.Retry, object] # type: ignore
4444

45+
from google.cloud.location import locations_pb2 # type: ignore
4546
from google.longrunning import operations_pb2
4647

4748
from google.cloud.discoveryengine_v1.types import completion_service

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/services/completion_service/client.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@
4646
except AttributeError: # pragma: NO COVER
4747
OptionalRetry = Union[retries.Retry, object] # type: ignore
4848

49+
from google.cloud.location import locations_pb2 # type: ignore
4950
from google.longrunning import operations_pb2
5051

5152
from google.cloud.discoveryengine_v1.types import completion_service

‎packages/google-cloud-discoveryengine/google/cloud/discoveryengine_v1/services/completion_service/transports/base.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@
2222
from google.api_core import retry as retries
2323
import google.auth # type: ignore
2424
from google.auth import credentials as ga_credentials # type: ignore
25+
from google.cloud.location import locations_pb2 # type: ignore
2526
from google.longrunning import operations_pb2
2627
from google.oauth2 import service_account # type: ignore
2728

0 commit comments

Comments
 (0)