hjxdl 0.1.12__py3-none-any.whl → 0.1.13__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.12'
-__version_tuple__ = version_tuple = (0, 1, 12)
+__version__ = version = '0.1.13'
+__version_tuple__ = version_tuple = (0, 1, 13)

hdl/utils/llm/chat.py

@@ -364,6 +364,7 @@ class OpenAI_M():
         prompt : str,
         stop: list[str] | None = ["USER:", "ASSISTANT:"],
         # history: list = [],
+        model="default_model",
         **kwargs: t.Any,
     ) -> str:
         """Invoke the chatbot with the given prompt and return the response.
@@ -383,6 +384,7 @@ class OpenAI_M():
                 "content": prompt
             }],
             stream=False,
+            model=model,
             # model=kwargs.get("model", "default_model")
             **kwargs
         )
@@ -393,6 +395,7 @@ class OpenAI_M():
         prompt : str,
         stop: list[str] | None = ["USER:", "ASSISTANT:"],
         # history: list = [],
+        model="default_model",
         **kwargs: t.Any,
     ):
         """Generate text completion in a streaming fashion.
@@ -412,6 +415,7 @@ class OpenAI_M():
                 "content": prompt
             }],
             stream=True,
+            model=model,
             # model=kwargs.get("model", "default_model")
             **kwargs
         )

{hjxdl-0.1.12.dist-info → hjxdl-0.1.13.dist-info}/METADATA

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

{hjxdl-0.1.12.dist-info → hjxdl-0.1.13.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=5sZZNySv08wwfzJcSDssGTqUn9wlmDsR6R4XB8J8mFM,70
-hdl/_version.py,sha256=cuLjhzc4k2qL9vS7TDaPLs82wisSY2yFKQtDhC08Kis,413
+hdl/_version.py,sha256=S22EPqqZRb53L2H7sobVA3TUXv9skvkYd-YtLuHuV6M,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=FtpQjp87FIBSIOQBPYKjtscA13RpMRwLk7sKlMAnOQ0,14974
+hdl/utils/llm/chat.py,sha256=Xypu7KlnTeI9oep02tuw04GooAfysZEgY2T_P29xZuU,15086
 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.12.dist-info/METADATA,sha256=oM9y6W0hDN6TupHYR2OECwnI1sI7BLlpWqkbOV9slK0,543
-hjxdl-0.1.12.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
-hjxdl-0.1.12.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.12.dist-info/RECORD,,
+hjxdl-0.1.13.dist-info/METADATA,sha256=UlAuoHbbPfrnFywQuD4azAWgCHUGNpvR8snZAAhdnz4,543
+hjxdl-0.1.13.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
+hjxdl-0.1.13.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.13.dist-info/RECORD,,