fix slides app args (#7889)

GitOrigin-RevId: da1276e94a8b18f0ff36a0ab2daff71d8d1e05f6
pathwaycom · Dec 19, 2024 · 8238697 · 8238697
1 parent ab8d3ce
commit 8238697
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 5 deletions.
diff --git a/examples/pipelines/slides_ai_search/Dockerfile b/examples/pipelines/slides_ai_search/Dockerfile
@@ -1,4 +1,4 @@
-ARG PATHWAY_SRC_IMAGE=pathwaycom/pathway:0.15.3
+ARG PATHWAY_SRC_IMAGE=pathwaycom/pathway:latest
 
 FROM ${PATHWAY_SRC_IMAGE}
 

diff --git a/examples/pipelines/slides_ai_search/app.py b/examples/pipelines/slides_ai_search/app.py
@@ -12,7 +12,7 @@
 import pathway as pw
 from dotenv import load_dotenv
 from pathway.xpacks import llm
-from pathway_slides_ai_search import CustomDeckRetriever, add_slide_id, get_model
+from pathway_slides_ai_search import DeckRetrieverWithFileSave, add_slide_id, get_model
 from pydantic import BaseModel, ConfigDict, FilePath, InstanceOf
 
 
@@ -53,8 +53,7 @@ def run(self) -> None:
             doc_post_processors=[add_slide_id],
         )
 
-        app = CustomDeckRetriever(
-            llm=self.llm,
+        app = DeckRetrieverWithFileSave(
             indexer=doc_store,
             search_topk=self.search_topk,
         )

diff --git a/examples/pipelines/slides_ai_search/pathway_slides_ai_search/__init__.py b/examples/pipelines/slides_ai_search/pathway_slides_ai_search/__init__.py
@@ -88,7 +88,7 @@ def add_slide_id(text: str, metadata: dict) -> tuple[str, dict]:
     return (text, metadata)
 
 
-class CustomDeckRetriever(DeckRetriever):
+class DeckRetrieverWithFileSave(DeckRetriever):
     def dump_img_callback(self, key, row, time, is_addition):
         # save images parsed by the Pathway
         metadata = row["data"]