Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat: Add rag_files_count to RagCorpus to count number of associated files #4863

Closed
wants to merge 18 commits into from
Closed
Show file tree
Hide file tree
Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
e6028fb
feat: Remove autorater config related visibility v1beta1
gcf-owl-bot[bot] Jan 27, 2025
7c2a8b5
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 27, 2025
1478c17
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 27, 2025
e5f7af9
Merge branch 'owl-bot-copy' of https://github.com/googleapis/python-a…
gcf-owl-bot[bot] Jan 27, 2025
76387e7
feat: Model Registry Checkpoint API
gcf-owl-bot[bot] Jan 28, 2025
b688844
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 28, 2025
c2c1aeb
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 28, 2025
d2fc961
Merge branch 'owl-bot-copy' of https://github.com/googleapis/python-a…
gcf-owl-bot[bot] Jan 28, 2025
a811fd1
feat: Model Registry Checkpoint API
gcf-owl-bot[bot] Jan 28, 2025
5f334cb
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 28, 2025
647a1da
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 28, 2025
c8d6481
Merge branch 'owl-bot-copy' of https://github.com/googleapis/python-a…
gcf-owl-bot[bot] Jan 28, 2025
fe66103
feat: add optimized config in v1 API
gcf-owl-bot[bot] Jan 28, 2025
f5cb8a3
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 28, 2025
5e97c9a
feat: add support for field generate_omitted_as_internal in selective…
gcf-owl-bot[bot] Jan 30, 2025
dd0ec7a
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Jan 30, 2025
e0c487c
feat: Add rag_files_count to RagCorpus to count number of associated …
gcf-owl-bot[bot] Feb 3, 2025
fd9ac35
🦉 Updates from OwlBot post-processor
gcf-owl-bot[bot] Feb 3, 2025
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
The table of contents is too big for display.
Diff view
Diff view
  •  
  •  
  •  
Prev Previous commit
Next Next commit
🦉 Updates from OwlBot post-processor
  • Loading branch information
gcf-owl-bot[bot] committed Jan 28, 2025
commit f5cb8a352e68ace938658de3b6c9bdb1fbf6f7a3
18 changes: 9 additions & 9 deletions google/cloud/aiplatform_v1/services/migration_service/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -262,40 +262,40 @@ def parse_dataset_path(path: str) -> Dict[str, str]:
@staticmethod
def dataset_path(
project: str,
location: str,
dataset: str,
) -> str:
"""Returns a fully-qualified dataset string."""
return "projects/{project}/locations/{location}/datasets/{dataset}".format(
return "projects/{project}/datasets/{dataset}".format(
project=project,
location=location,
dataset=dataset,
)

@staticmethod
def parse_dataset_path(path: str) -> Dict[str, str]:
"""Parses a dataset path into its component segments."""
m = re.match(
r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
path,
)
m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
return m.groupdict() if m else {}

@staticmethod
def dataset_path(
project: str,
location: str,
dataset: str,
) -> str:
"""Returns a fully-qualified dataset string."""
return "projects/{project}/datasets/{dataset}".format(
return "projects/{project}/locations/{location}/datasets/{dataset}".format(
project=project,
location=location,
dataset=dataset,
)

@staticmethod
def parse_dataset_path(path: str) -> Dict[str, str]:
"""Parses a dataset path into its component segments."""
m = re.match(r"^projects/(?P<project>.+?)/datasets/(?P<dataset>.+?)$", path)
m = re.match(
r"^projects/(?P<project>.+?)/locations/(?P<location>.+?)/datasets/(?P<dataset>.+?)$",
path,
)
return m.groupdict() if m else {}

@staticmethod
Expand Down
12 changes: 6 additions & 6 deletions google/cloud/aiplatform_v1/types/feature_online_store_service.py
Original file line number Diff line number Diff line change
Expand Up @@ -318,11 +318,11 @@ class StringFilter(proto.Message):
class NumericFilter(proto.Message):
r"""Numeric filter is used to search a subset of the entities by using
boolean rules on numeric columns. For example: Database Point 0:
{name: “a” value_int: 42} {name: “b” value_float: 1.0} Database
Point 1: {name: “a” value_int: 10} {name: “b” value_float: 2.0}
Database Point 2: {name: “a” value_int: -1} {name: “b” value_float:
3.0} Query: {name: “a” value_int: 12 operator: LESS} // Matches
Point 1, 2 {name: “b” value_float: 2.0 operator: EQUAL} // Matches
{name: "a" value_int: 42} {name: "b" value_float: 1.0} Database
Point 1: {name: "a" value_int: 10} {name: "b" value_float: 2.0}
Database Point 2: {name: "a" value_int: -1} {name: "b" value_float:
3.0} Query: {name: "a" value_int: 12 operator: LESS} // Matches
Point 1, 2 {name: "b" value_float: 2.0 operator: EQUAL} // Matches
Point 1

This message has `oneof`_ fields (mutually exclusive fields).
Expand Down Expand Up @@ -356,7 +356,7 @@ class NumericFilter(proto.Message):
"""

class Operator(proto.Enum):
r"""Datapoints for which Operator is true relative to the querys
r"""Datapoints for which Operator is true relative to the query's
Value field will be allowlisted.

Values:
Expand Down
29 changes: 29 additions & 0 deletions google/cloud/aiplatform_v1/types/feature_view.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,7 @@

import proto # type: ignore

from google.cloud.aiplatform_v1.types import machine_resources
from google.protobuf import timestamp_pb2 # type: ignore


Expand Down Expand Up @@ -100,6 +101,9 @@ class FeatureView(proto.Message):
data, so that approximate nearest neighbor
(a.k.a ANN) algorithms search can be performed
during online serving.
optimized_config (google.cloud.aiplatform_v1.types.FeatureView.OptimizedConfig):
Optional. Configuration for FeatureView
created under Optimized FeatureOnlineStore.
service_agent_type (google.cloud.aiplatform_v1.types.FeatureView.ServiceAgentType):
Optional. Service agent type used during data sync. By
default, the Vertex AI Service Agent is used. When using an
Expand Down Expand Up @@ -413,6 +417,26 @@ class VertexRagSource(proto.Message):
number=2,
)

class OptimizedConfig(proto.Message):
r"""Configuration for FeatureViews created in Optimized
FeatureOnlineStore.

Attributes:
automatic_resources (google.cloud.aiplatform_v1.types.AutomaticResources):
Optional. A description of resources that the FeatureView
uses, which to large degree are decided by Vertex AI, and
optionally allows only a modest additional configuration. If
min_replica_count is not set, the default value is 2. If
max_replica_count is not set, the default value is 6. The
max allowed replica count is 1000.
"""

automatic_resources: machine_resources.AutomaticResources = proto.Field(
proto.MESSAGE,
number=7,
message=machine_resources.AutomaticResources,
)

big_query_source: BigQuerySource = proto.Field(
proto.MESSAGE,
number=6,
Expand Down Expand Up @@ -464,6 +488,11 @@ class VertexRagSource(proto.Message):
number=15,
message=IndexConfig,
)
optimized_config: OptimizedConfig = proto.Field(
proto.MESSAGE,
number=16,
message=OptimizedConfig,
)
service_agent_type: ServiceAgentType = proto.Field(
proto.ENUM,
number=14,
Expand Down
13 changes: 0 additions & 13 deletions owl-bot-staging/v1/.coveragerc

This file was deleted.

33 changes: 0 additions & 33 deletions owl-bot-staging/v1/.flake8

This file was deleted.

2 changes: 0 additions & 2 deletions owl-bot-staging/v1/MANIFEST.in

This file was deleted.

49 changes: 0 additions & 49 deletions owl-bot-staging/v1/README.rst

This file was deleted.

3 changes: 0 additions & 3 deletions owl-bot-staging/v1/docs/_static/custom.css

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/dataset_service.rst

This file was deleted.

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/endpoint_service.rst

This file was deleted.

6 changes: 0 additions & 6 deletions owl-bot-staging/v1/docs/aiplatform_v1/evaluation_service.rst

This file was deleted.

This file was deleted.

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/feature_registry_service.rst

This file was deleted.

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/featurestore_service.rst

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/gen_ai_cache_service.rst

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/gen_ai_tuning_service.rst

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/index_endpoint_service.rst

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/index_service.rst

This file was deleted.

10 changes: 0 additions & 10 deletions owl-bot-staging/v1/docs/aiplatform_v1/job_service.rst

This file was deleted.

6 changes: 0 additions & 6 deletions owl-bot-staging/v1/docs/aiplatform_v1/llm_utility_service.rst

This file was deleted.

6 changes: 0 additions & 6 deletions owl-bot-staging/v1/docs/aiplatform_v1/match_service.rst

This file was deleted.

Loading