fix: update and print for eval (#536)

Author: badmonster0

examples/code_embedding/main.py

@@ -66,6 +66,10 @@ def search(pool: ConnectionPool, query: str, top_k: int = 5):
             ]
 
 def _main():
+    # Make sure the flow is built and up-to-date.
+    stats = code_embedding_flow.update()
+    print("Updated index: ", stats)
+
     # Initialize the database connection pool.
     pool = ConnectionPool(os.getenv("COCOINDEX_DATABASE_URL"))
     # Run queries in a loop to demonstrate the query capabilities.

python/cocoindex/cli.py

@@ -354,7 +354,7 @@ def evaluate(app_flow_specifier: str, output_dir: str | None, cache: bool = True
 
     fl = _flow_by_name(flow_ref)
     if output_dir is None:
-        output_dir = f"eval_{setting.get_app_namespace(trailing_delimiter='_')}{flow_name}_{datetime.datetime.now().strftime('%y%m%d_%H%M%S')}"
+        output_dir = f"eval_{setting.get_app_namespace(trailing_delimiter='_')}{fl.name}_{datetime.datetime.now().strftime('%y%m%d_%H%M%S')}"
     options = flow.EvaluateAndDumpOptions(output_dir=output_dir, use_cache=cache)
     fl.evaluate_and_dump(options)
 

python/cocoindex/flow.py

@@ -538,8 +538,8 @@ async def update_async(self) -> _engine.IndexUpdateInfo:
         Update the index defined by the flow.
         Once the function returns, the index is fresh up to the moment when the function is called.
         """
-        updater = await FlowLiveUpdater.create_async(self, FlowLiveUpdaterOptions(live_mode=False))
-        await updater.wait_async()
+        async with FlowLiveUpdater(self, FlowLiveUpdaterOptions(live_mode=False)) as updater:
+            await updater.wait_async()
         return updater.update_stats()
 
     def evaluate_and_dump(self, options: EvaluateAndDumpOptions):