From 33b351a1e608662cd7a8e62e6bcc8833f1a89306 Mon Sep 17 00:00:00 2001
From: Miguel Grinberg <miguel.grinberg@gmail.com>
Date: Mon, 19 Aug 2024 11:31:29 +0100
Subject: [PATCH] feedback

---
 examples/async/semantic_text.py | 8 +++-----
 examples/semantic_text.py       | 8 +++-----
 2 files changed, 6 insertions(+), 10 deletions(-)

diff --git a/examples/async/semantic_text.py b/examples/async/semantic_text.py
index 00965f30..cf63b003 100644
--- a/examples/async/semantic_text.py
+++ b/examples/async/semantic_text.py
@@ -108,11 +108,9 @@ async def create() -> None:
 
 
 async def search(query: str) -> dsl.AsyncSearch[WorkplaceDoc]:
-    return WorkplaceDoc.search()[:5].query(
-        "semantic",
-        field=WorkplaceDoc.content,
-        query=query,
-    )
+    search = WorkplaceDoc.search()
+    search = search[:5]
+    return search.query(dsl.query.Semantic(field=WorkplaceDoc.content, query=query))
 
 
 def parse_args() -> argparse.Namespace:
diff --git a/examples/semantic_text.py b/examples/semantic_text.py
index 7f935b80..7461e18d 100644
--- a/examples/semantic_text.py
+++ b/examples/semantic_text.py
@@ -107,11 +107,9 @@ def create() -> None:
 
 
 def search(query: str) -> dsl.Search[WorkplaceDoc]:
-    return WorkplaceDoc.search()[:5].query(
-        "semantic",
-        field=WorkplaceDoc.content,
-        query=query,
-    )
+    search = WorkplaceDoc.search()
+    search = search[:5]
+    return search.query(dsl.query.Semantic(field=WorkplaceDoc.content, query=query))
 
 
 def parse_args() -> argparse.Namespace: