hjxdl 0.0.4__py3-none-any.whl → 0.0.6__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.0.4'
-__version_tuple__ = version_tuple = (0, 0, 4)
+__version__ = version = '0.0.6'
+__version_tuple__ = version_tuple = (0, 0, 6)

hdl/utils/llm/chat.py

@@ -16,14 +16,15 @@ class GGUF_M():
         server_ip: str = "127.0.0.1",
         server_port: int = 8000,
     ):
-        """Initialize the model from a local path.
-        Here it does not manage chat histories.
+        """Initialize the model with the provided model path and optional parameters.
         
         Args:
             model_path (str): The path to the model.
             device (str, optional): The device to use for model initialization. Defaults to 'gpu'.
             generation_kwargs (dict, optional): Additional keyword arguments for model generation. Defaults to {}.
-        """
+            server_ip (str, optional): The IP address of the server. Defaults to "127.0.0.1".
+            server_port (int, optional): The port of the server. Defaults to 8000.
+        """ 
         # 从本地初始化模型
         super().__init__()
         self.generation_kwargs = generation_kwargs
@@ -130,6 +131,15 @@ class OpenAI_M():
         server_ip: str = "172.28.1.2",
         server_port: int = 8000,
     ):
+        """Initialize the class with the specified parameters.
+        
+        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.
+        """
         self.model_path = model_path
         self.server_ip = server_ip
         self.server_port = server_port
@@ -142,6 +152,16 @@ class OpenAI_M():
         # history: list = [],
         **kwargs: t.Any,
     ) -> str:
+        """Invoke the chatbot with the given prompt and return the response.
+        
+        Args:
+            prompt (str): The prompt to provide to the chatbot.
+            stop (list[str], optional): List of strings that indicate the end of the conversation. Defaults to ["USER:", "ASSISTANT:"].
+            **kwargs: Additional keyword arguments to pass to the chatbot.
+        
+        Returns:
+            str: The response generated by the chatbot.
+        """
         resp = chat_oai_invoke(
             base_url=self.base_url,
             model=self.model_path,
@@ -156,6 +176,16 @@ class OpenAI_M():
         # history: list = [],
         **kwargs: t.Any,
     ) -> str:
+        """Generate a response by streaming conversation with the OpenAI chat model.
+        
+        Args:
+            prompt (str): The prompt to start the conversation.
+            stop (list[str], optional): List of strings that indicate when the conversation should stop. Defaults to ["USER:", "ASSISTANT:"].
+            **kwargs: Additional keyword arguments to pass to the chat model.
+        
+        Returns:
+            str: The response generated by the chat model.
+        """
         resp = chat_oai_stream(
             base_url=self.base_url,
             model=self.model_path,

{hjxdl-0.0.4.dist-info → hjxdl-0.0.6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hjxdl
-Version: 0.0.4
+Version: 0.0.6
 Summary: A collection of functions for Jupyter notebooks
 Home-page: https://github.com/huluxiaohuowa/hdl
 Author: Jianxing Hu
@@ -10,6 +10,7 @@ Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Requires-Python: >=3.6
 Description-Content-Type: text/markdown
+Requires-Dist: jupyfuncs
 
 # DL framework by Jianxing
 

{hjxdl-0.0.4.dist-info → hjxdl-0.0.6.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=3bp7HUNfj6gqhNMj_EsyToZqjpqggsv9ohT9Sghk2mA,23
-hdl/_version.py,sha256=yBVOKdXLEcTVc7YV7ZPqRXhRDRt-pKrfXxcgHkgPY5g,411
+hdl/_version.py,sha256=c6ZQWSJeXXzGZ3WoZWjkA-MiNkBFXMIRV9kZPo4MQ_M,411
 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,10 +84,10 @@ 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=SoODp3Sf6FNYdajoI39swvPJm64fSd6bFYFdta4xD28,5344
+hdl/utils/llm/chat.py,sha256=UFMv_0KQoGLCim0YGQqneA_HUKzQA9R6K69_rTeBmKM,7022
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
-hjxdl-0.0.4.dist-info/METADATA,sha256=13WEfg3X9Fi5v3OBvr5Mu04OTODqm3E8Q46VIHHiQmU,525
-hjxdl-0.0.4.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
-hjxdl-0.0.4.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.0.4.dist-info/RECORD,,
+hjxdl-0.0.6.dist-info/METADATA,sha256=X3_IpOR1UIG4a07CtXTAZjqJTF-uzO2dQw4Z-HvcrQA,550
+hjxdl-0.0.6.dist-info/WHEEL,sha256=y4mX-SOX4fYIkonsAGA5N0Oy-8_gI4FXw5HNI1xqvWg,91
+hjxdl-0.0.6.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.0.6.dist-info/RECORD,,