hjxdl 0.1.9__py3-none-any.whl → 0.1.11__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.1.9'
-__version_tuple__ = version_tuple = (0, 1, 9)
+__version__ = version = '0.1.11'
+__version_tuple__ = version_tuple = (0, 1, 11)

hdl/utils/llm/chat.py

@@ -327,19 +327,30 @@ class OpenAI_M():
         *args,
         **kwargs
     ):
-        """Initialize the class with the specified parameters.
+        """Initialize the OpenAI client.
         
         Args:
-            model_path (str, optional): Path to the model file. Defaults to None.
-            device (str, optional): Device to run the model on. Defaults to 'gpu'.
-            generation_kwargs (dict, optional): Additional keyword arguments for model generation. Defaults to {}.
-            server_ip (str, optional): IP address of the server. Defaults to "172.28.1.2".
-            server_port (int, optional): Port number of the server. Defaults to 8000.
+            model_path (str): Path to the model file. Defaults to "default_model".
+            device (str): Device to use for model inference. Defaults to 'gpu'.
+            generation_kwargs (dict): Additional keyword arguments for model generation. Defaults to {}.
+            server_ip (str): IP address of the server. Defaults to "172.28.1.2".
+            server_port (int): Port number of the server. Defaults to 8000.
+            api_key (str): API key for authentication. Defaults to "dummy_key".
+            *args: Variable length argument list.
+            **kwargs: Arbitrary keyword arguments.
+        
+        Attributes:
+            model_path (str): Path to the model file.
+            server_ip (str): IP address of the server.
+            server_port (int): Port number of the server.
+            base_url (str): Base URL for API requests.
+            api_key (str): API key for authentication.
+            client (OpenAI): OpenAI client for making API requests.
         """
         self.model_path = model_path
         self.server_ip = server_ip
         self.server_port = server_port
-        self.base_url = "http://{self.server_ip}:{str(self.server_port)}/v1"
+        self.base_url = f"http://{self.server_ip}:{str(self.server_port)}/v1"
         self.api_key = api_key
         self.client = OpenAI(
             base_url=self.base_url,

{hjxdl-0.1.9.dist-info → hjxdl-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hjxdl
-Version: 0.1.9
+Version: 0.1.11
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu

{hjxdl-0.1.9.dist-info → hjxdl-0.1.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=5sZZNySv08wwfzJcSDssGTqUn9wlmDsR6R4XB8J8mFM,70
-hdl/_version.py,sha256=NWmu2cvzOcqY9v-ee-qFLmtXRczssdN-cFGZ9qMNSmY,411
+hdl/_version.py,sha256=HreDwlLXV189L3kiBj3huM_kqWD1usijlC8LN1YXcCM,413
 hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
 hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -84,12 +84,12 @@ hdl/utils/database_tools/connect.py,sha256=KUnVG-8raifEJ_N0b3c8LkTTIfn9NIyw8LX6q
 hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/general/glob.py,sha256=8-RCnt6L297wMIfn34ZAMCsGCZUjHG3MGglGZI1cX0g,491
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=67Mx1Q077BOscQbHiDxJpjBQalJKO8SQMLcRch5_Xj8,14352
+hdl/utils/llm/chat.py,sha256=WElHLIi9kEU2v_42moqSp9b4UYCwB5xow1rTsi1HMRc,14854
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
-hjxdl-0.1.9.dist-info/METADATA,sha256=O6Utim9uYtAH6dTsbrPQrgkdzCOHjwsYaGUHn6_noIM,542
-hjxdl-0.1.9.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-hjxdl-0.1.9.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.9.dist-info/RECORD,,
+hjxdl-0.1.11.dist-info/METADATA,sha256=0xzd32HG6sfj4SDK9zdrwb-fD5leA6JqunVVTB4wpc0,543
+hjxdl-0.1.11.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+hjxdl-0.1.11.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.11.dist-info/RECORD,,