unique_toolkit 0.5.33__py3-none-any.whl → 0.5.36__py3-none-any.whl

unique_toolkit/app/schemas.py

@@ -37,6 +37,7 @@ class EventAdditionalParameters(BaseModel):
     model_config = model_config
 
     translate_to_language: Optional[str] = None
+    content_id_to_translate: Optional[str] = None
 
 
 class EventPayload(BaseModel):

unique_toolkit/content/service.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import re
 import tempfile
 from pathlib import Path
 from typing import Optional, Union, cast
@@ -242,6 +243,7 @@ class ContentService(BaseService):
         mime_type: str,
         scope_id: Optional[str] = None,
         chat_id: Optional[str] = None,
+        skip_ingestion: Optional[bool] = False,
     ):
         """
         Uploads content to the knowledge base.
@@ -264,6 +266,7 @@ class ContentService(BaseService):
                 mime_type=mime_type,
                 scope_id=scope_id,
                 chat_id=chat_id,
+                skip_ingestion=skip_ingestion,
             )
         except Exception as e:
             self.logger.error(f"Error while uploading content: {e}")
@@ -276,6 +279,7 @@ class ContentService(BaseService):
         mime_type: str,
         scope_id: Optional[str] = None,
         chat_id: Optional[str] = None,
+        skip_ingestion: Optional[bool] = False,
     ):
         if not chat_id and not scope_id:
             raise ValueError("chat_id or scope_id must be provided")
@@ -318,16 +322,21 @@ class ContentService(BaseService):
             self.logger.error(error_msg)
             raise ValueError(error_msg)
 
+        input_dict = {
+            "key": content_name,
+            "title": content_name,
+            "mimeType": mime_type,
+            "byteSize": byte_size,
+        }
+
+        if skip_ingestion:
+            input_dict["ingestionConfig"] = {"uniqueIngestionMode": "SKIP_INGESTION"}
+
         if chat_id:
             unique_sdk.Content.upsert(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                input={
-                    "key": content_name,
-                    "title": content_name,
-                    "mimeType": mime_type,
-                    "byteSize": byte_size,
-                },
+                input=input_dict,
                 fileUrl=read_url,
                 chatId=chat_id,
             )  # type: ignore
@@ -335,18 +344,95 @@ class ContentService(BaseService):
             unique_sdk.Content.upsert(
                 user_id=self.event.user_id,
                 company_id=self.event.company_id,
-                input={
-                    "key": content_name,
-                    "title": content_name,
-                    "mimeType": mime_type,
-                    "byteSize": byte_size,
-                },
+                input=input_dict,
                 fileUrl=read_url,
                 scopeId=scope_id,
             )  # type: ignore
 
         return Content(**created_content)
 
+    def request_content_by_id(
+        self, content_id: str, chat_id: str | None
+    ) -> requests.Response:
+        """
+        Sends a request to download content from a chat.
+
+        Args:
+            content_id (str): The ID of the content to download.
+            chat_id (str): The ID of the chat from which to download the content. Defaults to None to download from knowledge base.
+
+        Returns:
+            requests.Response: The response object containing the downloaded content.
+
+        """
+        url = f"{unique_sdk.api_base}/content/{content_id}/file"
+        if chat_id:
+            url = f"{url}?chatId={chat_id}"
+
+        # Download the file and save it to the random directory
+        headers = {
+            "x-api-version": unique_sdk.api_version,
+            "x-app-id": unique_sdk.app_id,
+            "x-user-id": self.event.user_id,
+            "x-company-id": self.event.company_id,
+            "Authorization": "Bearer %s" % (unique_sdk.api_key,),
+        }
+
+        return requests.get(url, headers=headers)
+
+    def download_content_to_file_by_id(
+        self,
+        content_id: str,
+        chat_id: Optional[str] = None,
+        filename: str | None = None,
+        tmp_dir_path: Optional[Union[str, Path]] = "/tmp",
+    ):
+        """
+        Downloads content from a chat and saves it to a file.
+
+        Args:
+            content_id (str): The ID of the content to download.
+            chat_id (Optional[str]): The ID of the chat to download from. Defaults to None and the file is downloaded from the knowledge base.
+            filename (str | None): The name of the file to save the content as. If not provided, the original filename will be used. Defaults to None.
+            tmp_dir_path (Optional[Union[str, Path]]): The path to the temporary directory where the content will be saved. Defaults to "/tmp".
+
+        Returns:
+            Path: The path to the downloaded file.
+
+        Raises:
+            Exception: If the download fails or the filename cannot be determined.
+        """
+
+        response = self.request_content_by_id(content_id, chat_id)
+        random_dir = tempfile.mkdtemp(dir=tmp_dir_path)
+
+        if response.status_code == 200:
+            if filename:
+                content_path = Path(random_dir) / filename
+            else:
+                pattern = r'filename="([^"]+)"'
+                match = re.search(
+                    pattern, response.headers.get("Content-Disposition", "")
+                )
+                if match:
+                    content_path = Path(random_dir) / match.group(1)
+                else:
+                    error_msg = (
+                        "Error downloading file: Filename could not be determined"
+                    )
+                    self.logger.error(error_msg)
+                    raise Exception(error_msg)
+
+            with open(content_path, "wb") as file:
+                file.write(response.content)
+        else:
+            error_msg = f"Error downloading file: Status code {response.status_code}"
+            self.logger.error(error_msg)
+            raise Exception(error_msg)
+
+        return content_path
+
+    # TODO: Discuss if we should deprecate this method due to unclear use by content_name
     def download_content(
         self,
         content_id: str,
@@ -370,26 +456,11 @@ class ContentService(BaseService):
             Exception: If the download fails.
         """
 
-        url = f"{unique_sdk.api_base}/content/{content_id}/file"
-        if chat_id:
-            url = f"{url}?chatId={chat_id}"
+        response = self.request_content_by_id(content_id, chat_id)
 
-        # Create a random directory inside /tmp
         random_dir = tempfile.mkdtemp(dir=dir_path)
-
-        # Create the full file path
         content_path = Path(random_dir) / content_name
 
-        # Download the file and save it to the random directory
-        headers = {
-            "x-api-version": unique_sdk.api_version,
-            "x-app-id": unique_sdk.app_id,
-            "x-user-id": self.event.user_id,
-            "x-company-id": self.event.company_id,
-            "Authorization": "Bearer %s" % (unique_sdk.api_key,),
-        }
-
-        response = requests.get(url, headers=headers)
         if response.status_code == 200:
             with open(content_path, "wb") as file:
                 file.write(response.content)

{unique_toolkit-0.5.33.dist-info → unique_toolkit-0.5.36.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.33
+Version: 0.5.36
 Summary: 
 License: Proprietary
 Author: Martin Fadler
@@ -100,6 +100,16 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.5.36] - 2024-11-19
+- Add possibility to return the response from the download file from chat request
+- Add possibility to not specify a filename and use filename from response headers
+
+## [0.5.35] - 2024-11-18
+- Add the possibilty to upload files without triggering ingestion by setting `skip_ingestion` to `True` in `ContentService.upload_content`
+
+## [0.5.34] - 2024-11-15
+- Add `content_id_to_translate` to `EventAdditionalParameters`
+
 ## [0.5.33] - 2024-10-30
 - Force randomizing tool_call_id. This is helpful to better identify the tool_calls.
 

{unique_toolkit-0.5.33.dist-info → unique_toolkit-0.5.36.dist-info}/RECORD

@@ -8,7 +8,7 @@ unique_toolkit/app/init_logging.py,sha256=Sh26SRxOj8i8dzobKhYha2lLrkrMTHfB1V4jR3
 unique_toolkit/app/init_sdk.py,sha256=Nv4Now4pMfM0AgRhbtatLpm_39rKxn0WmRLwmPhRl-8,1285
 unique_toolkit/app/performance/async_tasks.py,sha256=H0l3OAcosLwNHZ8d2pd-Di4wHIXfclEvagi5kfqLFPA,1941
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=tzrmUFKZUdC1P3LxZ7DrElpkMtekUDoClb7jCRzGqNQ,1521
+unique_toolkit/app/schemas.py,sha256=6RY7Ex-B3pOnFILlitHEi9sqJvupbpdxlN9xt33qRsM,1571
 unique_toolkit/app/verification.py,sha256=UZqTHg3PX_QxMjeLH_BVBYoMVqMnMpeMoqvyTBKDqj8,1996
 unique_toolkit/chat/__init__.py,sha256=1prdTVfLOf6NgU-Aa1VIO-XiR6OYuRm51LaVRfKDCqc,267
 unique_toolkit/chat/schemas.py,sha256=IHOnb3pWlRlSPoEUWBTl6LK8YeMdlg2iXi9ghyBeiLw,1495
@@ -17,7 +17,7 @@ unique_toolkit/chat/state.py,sha256=Cjgwv_2vhDFbV69xxsn7SefhaoIAEqLx3ferdVFCnOg,
 unique_toolkit/chat/utils.py,sha256=ihm-wQykBWhB4liR3LnwPVPt_qGW6ETq21Mw4HY0THE,854
 unique_toolkit/content/__init__.py,sha256=MSH2sxjQyKD2Sef92fzE5Dt9SihdzivB6yliSwJfTmQ,890
 unique_toolkit/content/schemas.py,sha256=zks_Pkki2VhxICJJgHZyc-LPmRuj5dLbw3pgcUT7SW8,2362
-unique_toolkit/content/service.py,sha256=ZGYWYTphXpcByXyMqr1VOVUHdmdnsR-XIS_YRX0Wyv4,14211
+unique_toolkit/content/service.py,sha256=e23GX2pzIppHCFPUHng-82pSELT8JKT4TRMCO6M2Lok,16968
 unique_toolkit/content/utils.py,sha256=Lake671plRsqNvO3pN_rmyVcpwbdED_KQpLcCnc4lv4,6902
 unique_toolkit/embedding/__init__.py,sha256=dr8M9jvslQTxPpxgaGwzxY0FildiWf-DidN_cahPAWw,191
 unique_toolkit/embedding/schemas.py,sha256=1GvKCaSk4jixzVQ2PKq8yDqwGEVY_hWclYtoAr6CC2g,96
@@ -40,7 +40,7 @@ unique_toolkit/language_model/infos.py,sha256=kQK6F3r8xTN7oT6b39J7rxW-Y4iPXjx_Fr
 unique_toolkit/language_model/schemas.py,sha256=0pMLM1cCplw1OlM-eLqldZtYMObGr7tjHKXSHQP7hwE,6216
 unique_toolkit/language_model/service.py,sha256=s5X6EStyYumiYqlD9gkW4GANif18d9QZOMysmUSfv8M,13433
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.33.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-0.5.33.dist-info/METADATA,sha256=c063R7-la1op6nRW1W2IIFdQrl21NA_P3JmorSbNV8Y,13190
-unique_toolkit-0.5.33.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.33.dist-info/RECORD,,
+unique_toolkit-0.5.36.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-0.5.36.dist-info/METADATA,sha256=4YTkAbPH4uPA3IxsLyxr4-eI3wvn8LLJTAxhO-UDNTw,13635
+unique_toolkit-0.5.36.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.36.dist-info/RECORD,,