update integration tests output

Author: GuanyiLi-Craig

tests_integration/agents/run_agents.py

@@ -7,34 +7,49 @@
 from pathlib import Path
 
 
-sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
+try:
+    sys.stdout.reconfigure(encoding="utf-8")
+except AttributeError:
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", write_through=True)
 
 
-def run_scripts(pass_local: bool = True) -> int:
+
+def run_scripts(pass_local: bool = True, collect: bool = False):
     """Run all example scripts in this directory.
 
     Args:
         pass_local: If True, skip tests with 'ollama' or 'local' in their name.
+        collect: If True, return per-script results without printing.
 
     Returns:
-        Exit code (0 for success, 1 for failure).
+        List of per-script results if collect is True, otherwise exit code (0 for success, 1 for failure).
     """
     python_executable = sys.executable
     current_directory = Path(__file__).parent
 
     # Find all example files
     example_files = sorted(current_directory.glob("*_example.py"))
 
-    passed_scripts = []
-    failed_scripts = {}
+    results = []
 
     for file in example_files:
         filename = file.name
         if pass_local and ("ollama" in filename or "_local" in filename):
-            print(f"Skipping {filename} (local test)")
+            message = f"Skipping {filename} (local test)"
+            if not collect:
+                print(message)
+            results.append(
+                {
+                    "name": filename,
+                    "status": "skipped",
+                    "output": "",
+                    "error": message,
+                }
+            )
             continue
 
-        print(f"Running {filename}...")
+        if not collect:
+            print(f"Running {filename}...")
         try:
             result = subprocess.run(
                 [python_executable, str(file)],
@@ -43,23 +58,44 @@ def run_scripts(pass_local: bool = True) -> int:
                 check=True,
                 cwd=current_directory,
             )
-            print(f"Output of {filename}:\n{result.stdout}")
-            passed_scripts.append(filename)
+            if not collect:
+                print(f"Output of {filename}:\n{result.stdout}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "passed",
+                    "output": result.stdout,
+                    "error": "",
+                }
+            )
         except subprocess.CalledProcessError as e:
-            print(f"Error running {filename}:\n{e.stderr}")
-            failed_scripts[filename] = e.stderr
+            if not collect:
+                print(f"Error running {filename}:\n{e.stderr}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "failed",
+                    "output": e.stdout,
+                    "error": e.stderr,
+                }
+            )
+
+    if collect:
+        return results
+
+    passed_scripts = [r for r in results if r["status"] == "passed"]
+    failed_scripts = [r for r in results if r["status"] == "failed"]
 
-    # Summary
     print("\n" + "=" * 50)
     print("Summary:")
     print(f"Passed: {len(passed_scripts)}")
     for script in passed_scripts:
-        print(f"  ✓ {script}")
+        print(f"  ✓ {script['name']}")
 
     if failed_scripts:
         print(f"\nFailed: {len(failed_scripts)}")
         for script in failed_scripts:
-            print(f"  ✗ {script}")
+            print(f"  ✗ {script['name']}")
         return 1
 
     return 0

tests_integration/embedding_assistant/run_embedding_assistant.py

@@ -7,34 +7,49 @@
 from pathlib import Path
 
 
-sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
+try:
+    sys.stdout.reconfigure(encoding="utf-8")
+except AttributeError:
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", write_through=True)
 
 
-def run_scripts(pass_local: bool = True) -> int:
+
+def run_scripts(pass_local: bool = True, collect: bool = False):
     """Run all example scripts in this directory.
 
     Args:
         pass_local: If True, skip tests with 'ollama' or 'local' in their name.
+        collect: If True, return per-script results without printing.
 
     Returns:
-        Exit code (0 for success, 1 for failure).
+        List of per-script results if collect is True, otherwise exit code (0 for success, 1 for failure).
     """
     python_executable = sys.executable
     current_directory = Path(__file__).parent
 
     # Find all example files
     example_files = sorted(current_directory.glob("*_example.py"))
 
-    passed_scripts = []
-    failed_scripts = {}
+    results = []
 
     for file in example_files:
         filename = file.name
         if pass_local and ("ollama" in filename or "_local" in filename):
-            print(f"Skipping {filename} (local test)")
+            message = f"Skipping {filename} (local test)"
+            if not collect:
+                print(message)
+            results.append(
+                {
+                    "name": filename,
+                    "status": "skipped",
+                    "output": "",
+                    "error": message,
+                }
+            )
             continue
 
-        print(f"Running {filename}...")
+        if not collect:
+            print(f"Running {filename}...")
         try:
             result = subprocess.run(
                 [python_executable, str(file)],
@@ -43,23 +58,44 @@ def run_scripts(pass_local: bool = True) -> int:
                 check=True,
                 cwd=current_directory,
             )
-            print(f"Output of {filename}:\n{result.stdout}")
-            passed_scripts.append(filename)
+            if not collect:
+                print(f"Output of {filename}:\n{result.stdout}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "passed",
+                    "output": result.stdout,
+                    "error": "",
+                }
+            )
         except subprocess.CalledProcessError as e:
-            print(f"Error running {filename}:\n{e.stderr}")
-            failed_scripts[filename] = e.stderr
+            if not collect:
+                print(f"Error running {filename}:\n{e.stderr}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "failed",
+                    "output": e.stdout,
+                    "error": e.stderr,
+                }
+            )
+
+    if collect:
+        return results
+
+    passed_scripts = [r for r in results if r["status"] == "passed"]
+    failed_scripts = [r for r in results if r["status"] == "failed"]
 
-    # Summary
     print("\n" + "=" * 50)
     print("Summary:")
     print(f"Passed: {len(passed_scripts)}")
     for script in passed_scripts:
-        print(f"  ✓ {script}")
+        print(f"  ✓ {script['name']}")
 
     if failed_scripts:
         print(f"\nFailed: {len(failed_scripts)}")
         for script in failed_scripts:
-            print(f"  ✗ {script}")
+            print(f"  ✗ {script['name']}")
         return 1
 
     return 0

tests_integration/event_store_postgres/run_event_store_postgres.py

@@ -7,34 +7,49 @@
 from pathlib import Path
 
 
-sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8")
+try:
+    sys.stdout.reconfigure(encoding="utf-8")
+except AttributeError:
+    sys.stdout = io.TextIOWrapper(sys.stdout.buffer, encoding="utf-8", write_through=True)
 
 
-def run_scripts(pass_local: bool = True) -> int:
+
+def run_scripts(pass_local: bool = True, collect: bool = False):
     """Run all example scripts in this directory.
 
     Args:
         pass_local: If True, skip tests with 'ollama' or 'local' in their name.
+        collect: If True, return per-script results without printing.
 
     Returns:
-        Exit code (0 for success, 1 for failure).
+        List of per-script results if collect is True, otherwise exit code (0 for success, 1 for failure).
     """
     python_executable = sys.executable
     current_directory = Path(__file__).parent
 
     # Find all example files
     example_files = sorted(current_directory.glob("*_example.py"))
 
-    passed_scripts = []
-    failed_scripts = {}
+    results = []
 
     for file in example_files:
         filename = file.name
         if pass_local and ("ollama" in filename or "_local" in filename):
-            print(f"Skipping {filename} (local test)")
+            message = f"Skipping {filename} (local test)"
+            if not collect:
+                print(message)
+            results.append(
+                {
+                    "name": filename,
+                    "status": "skipped",
+                    "output": "",
+                    "error": message,
+                }
+            )
             continue
 
-        print(f"Running {filename}...")
+        if not collect:
+            print(f"Running {filename}...")
         try:
             result = subprocess.run(
                 [python_executable, str(file)],
@@ -43,23 +58,44 @@ def run_scripts(pass_local: bool = True) -> int:
                 check=True,
                 cwd=current_directory,
             )
-            print(f"Output of {filename}:\n{result.stdout}")
-            passed_scripts.append(filename)
+            if not collect:
+                print(f"Output of {filename}:\n{result.stdout}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "passed",
+                    "output": result.stdout,
+                    "error": "",
+                }
+            )
         except subprocess.CalledProcessError as e:
-            print(f"Error running {filename}:\n{e.stderr}")
-            failed_scripts[filename] = e.stderr
+            if not collect:
+                print(f"Error running {filename}:\n{e.stderr}")
+            results.append(
+                {
+                    "name": filename,
+                    "status": "failed",
+                    "output": e.stdout,
+                    "error": e.stderr,
+                }
+            )
+
+    if collect:
+        return results
+
+    passed_scripts = [r for r in results if r["status"] == "passed"]
+    failed_scripts = [r for r in results if r["status"] == "failed"]
 
-    # Summary
     print("\n" + "=" * 50)
     print("Summary:")
     print(f"Passed: {len(passed_scripts)}")
     for script in passed_scripts:
-        print(f"  ✓ {script}")
+        print(f"  ✓ {script['name']}")
 
     if failed_scripts:
         print(f"\nFailed: {len(failed_scripts)}")
         for script in failed_scripts:
-            print(f"  ✗ {script}")
+            print(f"  ✗ {script['name']}")
         return 1
 
     return 0