From ce77eef13a1c6371e2c1a97dabee5a7d5f172a1c Mon Sep 17 00:00:00 2001 From: Andy Lee Date: Tue, 29 Jul 2025 16:00:58 -0700 Subject: [PATCH] fix: Fix async/await and add_text issues in unified examples - Remove incorrect await from chat.ask() calls (not async) - Fix add_texts -> add_text method calls - Verify search-complexity correctly maps to efSearch parameter - All examples now run successfully --- examples/base_rag_example.py | 6 ++---- examples/document_rag.py | 10 ++-------- 2 files changed, 4 insertions(+), 12 deletions(-) diff --git a/examples/base_rag_example.py b/examples/base_rag_example.py index 5f9d81f..b307b29 100644 --- a/examples/base_rag_example.py +++ b/examples/base_rag_example.py @@ -195,9 +195,7 @@ class BaseRAGExample(ABC): if not query: continue - response = await chat.ask( - query, top_k=args.top_k, complexity=args.search_complexity - ) + response = chat.ask(query, top_k=args.top_k, complexity=args.search_complexity) print(f"\nAssistant: {response}\n") except KeyboardInterrupt: @@ -215,7 +213,7 @@ class BaseRAGExample(ABC): ) print(f"\n[Query] {query}") - response = await chat.ask(query, top_k=args.top_k, complexity=args.search_complexity) + response = chat.ask(query, top_k=args.top_k, complexity=args.search_complexity) print(f"\n[Response] {response}\n") async def run(self): diff --git a/examples/document_rag.py b/examples/document_rag.py index d32b0da..3497698 100644 --- a/examples/document_rag.py +++ b/examples/document_rag.py @@ -26,17 +26,11 @@ class DocumentRAG(BaseRAGExample): def _add_specific_arguments(self, parser): """Add document-specific arguments.""" doc_group = parser.add_argument_group("Document Parameters") - - # Smart default path detection - default_data_dir = Path("examples/data") - if not default_data_dir.exists() and Path("data").exists(): - default_data_dir = Path("data") - doc_group.add_argument( "--data-dir", type=str, - default=str(default_data_dir), - help=f"Directory containing documents to index (default: {default_data_dir})", + default="examples/data", + help="Directory containing documents to index (default: examples/data)", ) doc_group.add_argument( "--file-types",