unique_toolkit 0.5.14__py3-none-any.whl → 0.5.15__py3-none-any.whl

unique_toolkit/content/service.py

@@ -2,7 +2,7 @@ import logging
 import os
 import tempfile
 from pathlib import Path
-from typing import Optional, cast
+from typing import Optional, Union, cast
 
 import requests
 import unique_sdk
@@ -344,6 +344,7 @@ class ContentService(BaseService):
         content_id: str,
         content_name: str,
         chat_id: Optional[str] = None,
+        dir_path: Optional[Union[str, Path]] = "/tmp",
     ) -> Path:
         """
         Downloads content to temporary directory
@@ -352,6 +353,7 @@ class ContentService(BaseService):
             content_id (str): The id of the uploaded content.
             content_name (str): The name of the uploaded content.
             chat_id (Optional[str]): The chat_id, defaults to None.
+            dir_path (Optional[Union[str, Path]]): The directory path to download the content to, defaults to "/tmp". If not provided, the content will be downloaded to a random directory inside /tmp. Be aware that this directory won't be cleaned up automatically.
 
         Returns:
             content_path: The path to the downloaded content in the temporary directory.
@@ -365,7 +367,7 @@ class ContentService(BaseService):
             url = f"{url}?chatId={chat_id}"
 
         # Create a random directory inside /tmp
-        random_dir = tempfile.mkdtemp(dir="/tmp")
+        random_dir = tempfile.mkdtemp(dir=dir_path)
 
         # Create the full file path
         content_path = Path(random_dir) / content_name

{unique_toolkit-0.5.14.dist-info → unique_toolkit-0.5.15.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 0.5.14
+Version: 0.5.15
 Summary: 
 License: MIT
 Author: Martin Fadler
@@ -101,6 +101,9 @@ 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.15] - 2024-08-27
+- Possibility to specify root directory in `ContentService.download_content`
+
 ## [0.5.14] - 2024-08-26
 - Add AZURE_GPT_4o_MINI_2024_0718 to language model infos
 

{unique_toolkit-0.5.14.dist-info → unique_toolkit-0.5.15.dist-info}/RECORD

@@ -15,7 +15,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=UlC5nBIaFkq9TD31LR6ioG9JRZ1ScmtABi0l06HZR70,2231
-unique_toolkit/content/service.py,sha256=4CHqZDE3C-VQpG6RAlenarvBmBgI28Ec6T9HZf9TB84,13565
+unique_toolkit/content/service.py,sha256=iBS1ntg5xdL-i5lfDlN2MGYDTIdg3u-uxQH9h_ss3KM,13894
 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
@@ -26,7 +26,7 @@ unique_toolkit/language_model/infos.py,sha256=JkugUAFFlrhTHXeM3A_R5QLkNSR9Ro85xW
 unique_toolkit/language_model/schemas.py,sha256=h5zjZNk7O-wLKtRuiNtMCIbp5hEVXrAOviKonQcjFuI,4594
 unique_toolkit/language_model/service.py,sha256=JjsOOcGDcR7db3yF3_oDXclEGfxqmwWpL5jor7Q42cU,10470
 unique_toolkit/language_model/utils.py,sha256=WBPj1XKkDgxy_-T8HCZvsfkkSzj_1w4UZzNmyvdbBLY,1081
-unique_toolkit-0.5.14.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
-unique_toolkit-0.5.14.dist-info/METADATA,sha256=p_jPjGwKWiNOLzbVcibqdWCrXA1TA813LnIdLfkkmVE,10124
-unique_toolkit-0.5.14.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-0.5.14.dist-info/RECORD,,
+unique_toolkit-0.5.15.dist-info/LICENSE,sha256=bIeCWCYuoUU_MzNdg48-ubJSVm7qxakaRbzTiJ5uxrs,1065
+unique_toolkit-0.5.15.dist-info/METADATA,sha256=YWF1EolvIQgwdP9-oV04EE-o6hWzA5HP69bl3_Ui400,10227
+unique_toolkit-0.5.15.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-0.5.15.dist-info/RECORD,,