update embedding

Author: iamNarcisse

jigsawstack/__init__.py

@@ -13,6 +13,7 @@
 from .summary import Summary, AsyncSummary
 from .geo import Geo, AsyncGeo
 from .prompt_engine import PromptEngine, AsyncPromptEngine
+from .embedding import Embedding, AsyncEmbedding
 from .exceptions import JigsawStackError
 
 
@@ -110,6 +111,11 @@ def __init__(
             api_url=api_url,
             disable_request_logging=disable_request_logging,
         )
+        self.embedding = Embedding(
+            api_key=api_key,
+            api_url=api_url,
+            disable_request_logging=disable_request_logging,
+        ).execute
 
 
 class AsyncJigsawStack:
@@ -215,6 +221,11 @@ def __init__(
             api_url=api_url,
             disable_request_logging=disable_request_logging,
         )
+        self.embedding = AsyncEmbedding(
+            api_key=api_key,
+            api_url=api_url,
+            disable_request_logging=disable_request_logging,
+        ).execute
 
 
 # Create a global instance of the Web class

jigsawstack/embedding_api.py jigsawstack/embedding.pyjigsawstack/embedding_api.py renamed to jigsawstack/embedding.py

@@ -7,31 +7,18 @@
 
 
 class EmbeddingParams(TypedDict):
-    text: Union[str, List[str]]
-    """
-    The text to summarize.
-    """
-
-    type: NotRequired[Literal["text", "points"]]
-
-    """
-   The summary result type. Supported values are: text, points
-    """
+    text: NotRequired[str]
+    file_content: NotRequired[Any]
+    type: Literal["text", "text-other", "image", "audio", "pdf"]
     url: NotRequired[str]
     file_store_key: NotRequired[str]
-    max_points: NotRequired[int]
-    max_characters: NotRequired[int]
+    token_overflow_mode: NotRequired[Literal["truncate", "chunk", "error"]] = "chunk"
 
 
 class EmbeddingResponse(TypedDict):
     success: bool
-    """
-    Indicates whether the translation was successful.
-    """
-    summary: str
-    """
-    The summarized text.
-    """
+    embeddings: List[List[float]]
+    chunks: List[str]
 
 
 class Embedding(ClientConfig):
@@ -51,6 +38,16 @@ def __init__(
             disable_request_logging=disable_request_logging,
         )
 
+    def execute(self, params: EmbeddingParams) -> EmbeddingResponse:
+        path = "/embedding"
+        resp = Request(
+            config=self.config,
+            path=path,
+            params=cast(Dict[Any, Any], params),
+            verb="post",
+        ).perform_with_content()
+        return resp
+
 
 class AsyncEmbedding(ClientConfig):
 
@@ -70,7 +67,7 @@ def __init__(
         )
 
     async def execute(self, params: EmbeddingParams) -> EmbeddingResponse:
-        path = "/ai/embedding"
+        path = "/embedding"
         resp = await AsyncRequest(
             config=self.config,
             path=path,

setup.py

@@ -6,7 +6,7 @@
 
 setup(
     name="jigsawstack",
-    version="0.1.24",
+    version="0.1.25",
     description="JigsawStack Python SDK",
     long_description=open("README.md", encoding="utf8").read(),
     long_description_content_type="text/markdown",