Added support for calling summarize (#56)

Author: Deepak-Kesavan

src/unstract/sdk/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.28.0"
+__version__ = "0.29.0"
 
 
 def get_sdk_version():

src/unstract/sdk/embedding.py

@@ -22,7 +22,7 @@ def __init__(
         self,
         tool: BaseTool,
         adapter_instance_id: str,
-        usage_kwargs: dict[Any, Any] = None,
+        usage_kwargs: dict[Any, Any] = {},
     ):
         self._tool = tool
         self._adapter_instance_id = adapter_instance_id

src/unstract/sdk/index.py

@@ -32,7 +32,7 @@ def get_text_from_index(
         embedding_instance_id: str,
         vector_db_instance_id: str,
         doc_id: str,
-        usage_kwargs: dict[Any, Any] = None,
+        usage_kwargs: dict[Any, Any] = {},
     ):
         try:
             embedding = Embedding(
@@ -126,7 +126,7 @@ def index_file(
         reindex: bool = False,
         file_hash: Optional[str] = None,
         output_file_path: Optional[str] = None,
-        usage_kwargs: dict[Any, Any] = None,
+        usage_kwargs: dict[Any, Any] = {},
     ) -> str:
         """Indexes an individual file using the passed arguments.
 

src/unstract/sdk/llm.py

@@ -31,7 +31,7 @@ def __init__(
         self,
         tool: BaseTool,
         adapter_instance_id: str,
-        usage_kwargs: dict[Any, Any] = None,
+        usage_kwargs: dict[Any, Any] = {},
     ):
         """
 

src/unstract/sdk/prompt.py

@@ -37,6 +37,9 @@ def answer_prompt(self, payload: dict[str, Any]) -> dict[str, Any]:
     def single_pass_extraction(self, payload: dict[str, Any]) -> dict[str, Any]:
         return self._post_call("single-pass-extraction", payload)
 
+    def summarize(self, payload: dict[str, Any]) -> dict[str, Any]:
+        return self._post_call("summarize", payload)
+
     def _post_call(self, url_path: str, payload: dict[str, Any]) -> dict[str, Any]:
         """Invokes and communicates to prompt service to fetch response for the
         prompt.