fix: switch gemini embed to sync (#7096)

GitOrigin-RevId: 558fe7235766b48c942954e05a0f7e1027768459

CHANGELOG.md

@@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file.
 This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 ## [Unreleased]
 
+### Fixed
+- Switched `pw.xpacks.llm.embedders.GeminiEmbedder` to be sync to resolve compatibility issues with the Google Colab runs.
+- Pinned `surya-ocr` module version for stability.
+
 ## [0.14.1] - 2024-08-05
 
 ### Added

python/pathway/xpacks/llm/embedders.py

@@ -396,7 +396,7 @@ def __init__(
         if api_key is not None:
             self.kwargs["api_key"] = api_key
 
-    async def __wrapped__(self, input: str, **kwargs) -> list[float]:
+    def __wrapped__(self, input: str, **kwargs) -> list[float]:
         import google.generativeai as genai
 
         kwargs = {**self.kwargs, **kwargs}
@@ -406,6 +406,6 @@ async def __wrapped__(self, input: str, **kwargs) -> list[float]:
         if api_key is not None:
             genai.configure(api_key=api_key)
 
-        response = await genai.embed_content_async(model, content=[input], **kwargs)
+        response = genai.embed_content(model, content=[input], **kwargs)
         embedding = response["embedding"][0]
         return embedding