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
This commit is contained in:
@@ -195,9 +195,7 @@ class BaseRAGExample(ABC):
|
|||||||
if not query:
|
if not query:
|
||||||
continue
|
continue
|
||||||
|
|
||||||
response = await chat.ask(
|
response = chat.ask(query, top_k=args.top_k, complexity=args.search_complexity)
|
||||||
query, top_k=args.top_k, complexity=args.search_complexity
|
|
||||||
)
|
|
||||||
print(f"\nAssistant: {response}\n")
|
print(f"\nAssistant: {response}\n")
|
||||||
|
|
||||||
except KeyboardInterrupt:
|
except KeyboardInterrupt:
|
||||||
@@ -215,7 +213,7 @@ class BaseRAGExample(ABC):
|
|||||||
)
|
)
|
||||||
|
|
||||||
print(f"\n[Query] {query}")
|
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")
|
print(f"\n[Response] {response}\n")
|
||||||
|
|
||||||
async def run(self):
|
async def run(self):
|
||||||
|
|||||||
@@ -26,17 +26,11 @@ class DocumentRAG(BaseRAGExample):
|
|||||||
def _add_specific_arguments(self, parser):
|
def _add_specific_arguments(self, parser):
|
||||||
"""Add document-specific arguments."""
|
"""Add document-specific arguments."""
|
||||||
doc_group = parser.add_argument_group("Document Parameters")
|
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(
|
doc_group.add_argument(
|
||||||
"--data-dir",
|
"--data-dir",
|
||||||
type=str,
|
type=str,
|
||||||
default=str(default_data_dir),
|
default="examples/data",
|
||||||
help=f"Directory containing documents to index (default: {default_data_dir})",
|
help="Directory containing documents to index (default: examples/data)",
|
||||||
)
|
)
|
||||||
doc_group.add_argument(
|
doc_group.add_argument(
|
||||||
"--file-types",
|
"--file-types",
|
||||||
|
|||||||
Reference in New Issue
Block a user