hjxdl 0.0.4__tar.gz → 0.0.6__tar.gz

{hjxdl-0.0.4/hjxdl.egg-info → hjxdl-0.0.6}/PKG-INFO

@@ -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 → hjxdl-0.0.6}/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)

{hjxdl-0.0.4 → hjxdl-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 → hjxdl-0.0.6/hjxdl.egg-info}/PKG-INFO

@@ -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 → hjxdl-0.0.6}/hjxdl.egg-info/SOURCES.txt

@@ -4,6 +4,7 @@ MANIFEST.in
 README.md
 __init__.py
 pyproject.toml
+requirements.txt
 setup.py
 update_main.sh
 version.txt
@@ -109,4 +110,5 @@ hdl/utils/schedulers/norm_lr.py
 hjxdl.egg-info/PKG-INFO
 hjxdl.egg-info/SOURCES.txt
 hjxdl.egg-info/dependency_links.txt
+hjxdl.egg-info/requires.txt
 hjxdl.egg-info/top_level.txt

hjxdl-0.0.6/hjxdl.egg-info/requires.txt

@@ -0,0 +1 @@
+jupyfuncs

hjxdl-0.0.6/requirements.txt

@@ -0,0 +1 @@
+jupyfuncs

{hjxdl-0.0.4 → hjxdl-0.0.6}/setup.py

@@ -2,6 +2,16 @@ from setuptools import setup, find_packages
 import setuptools_scm
 
 
+def read_requirements():
+    """读取 requirements.txt 文件并返回依赖列表"""
+    with open('requirements.txt', 'r', encoding='utf-8') as file:
+        return [
+            line.strip()
+            for line in file
+            if line.strip() and not line.startswith('#')
+        ]
+
+
 def custom_version_scheme(version):
     """自定义版本号方案，确保没有 .dev 后缀"""
     if version.exact:
@@ -11,10 +21,12 @@ def custom_version_scheme(version):
     else:
         return version.format_with("0.0.0")
 
+
 def custom_local_scheme(version):
     """自定义本地版本方案，确保没有本地版本后缀"""
     return ""
 
+
 setup(
     name="hjxdl",
     use_scm_version={
@@ -35,5 +47,6 @@ setup(
         "Operating System :: OS Independent",
     ],
     python_requires='>=3.6',
-    setup_requires=['setuptools_scm']
+    setup_requires=['setuptools_scm'],
+    install_requires=read_requirements()
 )

hjxdl-0.0.6/version.txt

@@ -0,0 +1 @@
+0.0.6

hjxdl-0.0.4/version.txt

@@ -1 +0,0 @@
-0.0.4