chore: Track the entire agentplatform folder in noxfile LINT_PATHS and apply standard formatting.

PiperOrigin-RevId: 924566890

Author: vertex-sdk-bot

agentplatform/agent_engines/_agent_engines.py

@@ -1342,7 +1342,9 @@ def _update_deployment_spec_with_env_vars_dict_or_raise(
     for key, value in env_vars.items():
         if isinstance(value, Dict):
             try:
-                secret_ref = _agent_engines_utils._to_proto(value, aip_types.SecretRef())
+                secret_ref = _agent_engines_utils._to_proto(
+                    value, aip_types.SecretRef()
+                )
             except Exception as e:
                 raise ValueError(f"Failed to convert to secret ref: {value}") from e
             deployment_spec.secret_env.append(

agentplatform/agent_engines/templates/ag2.py

@@ -95,7 +95,9 @@ def _default_instrumentor_builder(project_id: str):
     cloud_trace_v2 = _agent_engines_utils._import_cloud_trace_v2_or_warn()
     openinference_autogen = _agent_engines_utils._import_openinference_autogen_or_warn()
     opentelemetry = _agent_engines_utils._import_opentelemetry_or_warn()
-    opentelemetry_sdk_trace = _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    opentelemetry_sdk_trace = (
+        _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    )
     if all(
         (
             cloud_trace_exporter,
@@ -396,7 +398,9 @@ def set_up(self):
         tools = self._tmpl_attrs.get("tools")
         ag2_tool_objects = self._tmpl_attrs.get("ag2_tool_objects")
         if tools and not ag2_tool_objects:
-            from agentplatform._genai import _agent_engines_utils
+            from agentplatform._genai import (
+                _agent_engines_utils,
+            )
 
             autogen_tools = _agent_engines_utils._import_autogen_tools_or_warn()
             if autogen_tools:

agentplatform/agent_engines/templates/langchain.py

@@ -196,9 +196,13 @@ def _default_instrumentor_builder(project_id: str):
 
     cloud_trace_exporter = _agent_engines_utils._import_cloud_trace_exporter_or_warn()
     cloud_trace_v2 = _agent_engines_utils._import_cloud_trace_v2_or_warn()
-    openinference_langchain = _agent_engines_utils._import_openinference_langchain_or_warn()
+    openinference_langchain = (
+        _agent_engines_utils._import_openinference_langchain_or_warn()
+    )
     opentelemetry = _agent_engines_utils._import_opentelemetry_or_warn()
-    opentelemetry_sdk_trace = _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    opentelemetry_sdk_trace = (
+        _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    )
     if all(
         (
             cloud_trace_exporter,

agentplatform/agent_engines/templates/langgraph.py

@@ -172,9 +172,13 @@ def _default_instrumentor_builder(project_id: str):
 
     cloud_trace_exporter = _agent_engines_utils._import_cloud_trace_exporter_or_warn()
     cloud_trace_v2 = _agent_engines_utils._import_cloud_trace_v2_or_warn()
-    openinference_langchain = _agent_engines_utils._import_openinference_langchain_or_warn()
+    openinference_langchain = (
+        _agent_engines_utils._import_openinference_langchain_or_warn()
+    )
     opentelemetry = _agent_engines_utils._import_opentelemetry_or_warn()
-    opentelemetry_sdk_trace = _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    opentelemetry_sdk_trace = (
+        _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+    )
     if all(
         (
             cloud_trace_exporter,

agentplatform/agent_engines/templates/llama_index.py

@@ -384,15 +384,21 @@ def set_up(self):
         that can not be serialized.
         """
         if self._enable_tracing:
-            from agentplatform._genai.agent_engines import _agent_engines_utils
+            from agentplatform._genai.agent_engines import (
+                _agent_engines_utils,
+            )
 
-            cloud_trace_exporter = _agent_engines_utils._import_cloud_trace_exporter_or_warn()
+            cloud_trace_exporter = (
+                _agent_engines_utils._import_cloud_trace_exporter_or_warn()
+            )
             cloud_trace_v2 = _agent_engines_utils._import_cloud_trace_v2_or_warn()
             openinference_llama_index = (
                 _agent_engines_utils._import_openinference_llama_index_or_warn()
             )
             opentelemetry = _agent_engines_utils._import_opentelemetry_or_warn()
-            opentelemetry_sdk_trace = _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+            opentelemetry_sdk_trace = (
+                _agent_engines_utils._import_opentelemetry_sdk_trace_or_warn()
+            )
             if all(
                 (
                     cloud_trace_exporter,
@@ -441,7 +447,9 @@ def set_up(self):
                 # Avoids AttributeError:
                 # 'ProxyTracerProvider' and 'NoOpTracerProvider' objects has no
                 # attribute 'add_span_processor'.
-                if _agent_engines_utils.is_noop_or_proxy_tracer_provider(tracer_provider):
+                if _agent_engines_utils.is_noop_or_proxy_tracer_provider(
+                    tracer_provider
+                ):
                     tracer_provider = opentelemetry_sdk_trace.TracerProvider()
                     opentelemetry.trace.set_tracer_provider(tracer_provider)
                 # Avoids OpenTelemetry client already exists error.
@@ -538,7 +546,9 @@ def query(
         Returns:
             The output of querying the Agent with the given input and config.
         """
-        from agentplatform._genai.agent_engines import _agent_engines_utils
+        from agentplatform._genai.agent_engines import (
+            _agent_engines_utils,
+        )
 
         if isinstance(input, str):
             input = {"input": input}

agentplatform/preview/__init__.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""The agentplatform.preview module."""
+"""The agentplatform.preview module."""

agentplatform/preview/rag/rag_data.py

@@ -55,7 +55,6 @@
     _gapic_utils,
 )
 from agentplatform.preview.rag.utils.resources import (
-    EmbeddingModelConfig,
     JiraSource,
     LayoutParserConfig,
     LlmParserConfig,
@@ -66,7 +65,6 @@
     RagEngineConfig,
     RagFile,
     RagManagedDb,
-    RagManagedVertexVectorSearch,
     RagMetadata,
     RagVectorDbConfig,
     SharePointSources,

agentplatform/preview/rag/rag_retrieval.py

@@ -16,7 +16,6 @@
 
 import re
 from typing import List, Optional
-import warnings
 
 from google.cloud import aiplatform_v1beta1
 from google.cloud.aiplatform import initializer

agentplatform/preview/rag/rag_store.py

@@ -16,7 +16,6 @@
 
 import re
 from typing import List, Optional, Union
-import warnings
 
 from google.cloud import aiplatform_v1beta1
 from google.cloud.aiplatform import initializer