Merge branch 'main' into adk-python-processing

docs/en/samples/pre_post_processing/python/__init__.py

@@ -1,3 +1,18 @@
+# Copyright 2026 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
 # This file makes the 'pre_post_processing/python' directory a Python package.
 
 # You can include any package-level initialization logic here if needed.

docs/en/samples/pre_post_processing/python/agent_test.py

@@ -23,19 +23,11 @@ ORCH_NAME = os.environ.get("ORCH_NAME")
 module_path = f"python.{ORCH_NAME}.agent"
 agent = importlib.import_module(module_path)
 
-
-@pytest.fixture(scope="module")
-def golden_keywords():
-    """Loads expected keywords from the golden.txt file."""
-    golden_file_path = Path(__file__).resolve().parent.parent / "golden.txt"
-    if not golden_file_path.exists():
-        pytest.fail(f"Golden file not found: {golden_file_path}")
-    try:
-        with open(golden_file_path, "r") as f:
-            return [line.strip() for line in f.readlines() if line.strip()]
-    except Exception as e:
-        pytest.fail(f"Could not read golden.txt: {e}")
-
+GOLDEN_KEYWORDS = [
+    "AI:",
+    "Loyalty Points",
+    "POLICY CHECK: Intercepting 'update-hotel'",
+]
 
 # --- Execution Tests ---
 class TestExecution:
@@ -51,9 +43,9 @@ class TestExecution:
         """Test that the script runs and produces no stderr."""
         assert script_output.err == "", f"Script produced stderr: {script_output.err}"
 
-    def test_keywords_in_output(self, script_output, golden_keywords):
+    def test_keywords_in_output(self, script_output):
         """Test that expected keywords are present in the script's output."""
         output = script_output.out
         print(f"\nAgent Output:\n{output}\n")
-        missing_keywords = [kw for kw in golden_keywords if kw not in output]
+        missing_keywords = [kw for kw in GOLDEN_KEYWORDS if kw not in output]
         assert not missing_keywords, f"Missing keywords in output: {missing_keywords}"

docs/en/samples/pre_post_processing/python/langchain/agent.py

@@ -47,7 +47,13 @@ async def enforce_business_rules(request, handler):
             except ValueError:
                 pass  # Ignore invalid date formats
 
-    return await handler(request)
+    # PRE: Code here runs BEFORE the tool execution
+    
+    # EXEC: Execute the tool (or next middleware)
+    result = await handler(request)
+
+    # POST: Code here runs AFTER the tool execution
+    return result
 
 
 # Post processing
@@ -58,8 +64,12 @@ async def enrich_response(request, handler):
     Adds loyalty points information to successful bookings.
     Standardizes output format.
     """
+    # PRE: Code here runs BEFORE the tool execution
+    
+    # EXEC: Execute the tool (or next middleware)
     result = await handler(request)
 
+    # POST: Code here runs AFTER the tool execution
     if isinstance(result, ToolMessage):
         content = str(result.content)
         tool_name = request.tool_call["name"]
@@ -79,6 +89,7 @@ async def main():
             system_prompt=system_prompt,
             model=model,
             tools=tools,
+            # add any pre and post processing methods
             middleware=[enforce_business_rules, enrich_response],
         )
 
@@ -88,7 +99,6 @@ async def main():
         )
 
         print("-" * 50)
-        print("Final Client Response:")
         last_ai_msg = response["messages"][-1].content
         print(f"AI: {last_ai_msg}")