hjxdl 0.3.9__py3-none-any.whl → 0.3.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.3.9'
-__version_tuple__ = version_tuple = (0, 3, 9)
+__version__ = version = '0.3.11'
+__version_tuple__ = version_tuple = (0, 3, 11)

hdl/utils/llm/chat.py

@@ -159,8 +159,8 @@ class OpenAI_M:
             self.client_conf = client_conf
 
         # self.clients = {}
-        for client_id, conf in self.client_conf.items():
-            conf[client_id]["client"] = OpenAI(
+        for _, conf in self.client_conf.items():
+            conf["client"] = OpenAI(
                 base_url=conf["host"],
                 api_key=conf.get("api_key", "dummy_key"),
                 *args,
@@ -281,14 +281,14 @@ class OpenAI_M:
 
     def get_resp(
         self,
-        client_id,
         prompt: str,
+        client_id: str = None,
         sys_info: str = None,
         assis_info: str = None,
         images: list = None,
         image_keys: tuple = ("image_url", "url"),
         stop: list[str] | None = ["USER:", "ASSISTANT:"],
-        model="default_model",
+        model: str=None,
         stream: bool = True,
         **kwargs: t.Any,
     ):
@@ -307,6 +307,8 @@ class OpenAI_M:
         Returns:
             _type_: The response object from the model.
         """
+        if not model:
+            model = self.client_conf[client_id]["model"]
 
         # Initialize the content list with at least the user's text input
         content = [

{hjxdl-0.3.9.dist-info → hjxdl-0.3.11.dist-info}/METADATA

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

{hjxdl-0.3.9.dist-info → hjxdl-0.3.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=nV2HEiFwTdaOZoFEyVxxG_D8Oq_nlSmX2vHL4jK4h6w,411
+hdl/_version.py,sha256=cgZZwxO2Czf2RAwPc5Ugf8lBeLEkSyVqWm7FWjhlv2A,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
@@ -128,7 +128,7 @@ hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
 hdl/utils/general/glob.py,sha256=Zuf7WHU0UdUPOs9UrhxmrCiMC8GrHxQU6n3mTThv6yc,1120
 hdl/utils/general/runners.py,sha256=x7QBolp3MrqNV6L4rB6Ueybr26bqkRFZTuXhY0SwyLk,3061
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=u59sIk6e-4-uM-XTR-EjtdKM-rtxlGzYdvNINH4fDAw,25784
+hdl/utils/llm/chat.py,sha256=9xAq8WBav8r0rayp66cPTZhAzEmMgTbtvDK7vP7rMcc,25850
 hdl/utils/llm/chatgr.py,sha256=5F5PJHe8vz3iCfi4TT54DCLRi1UeJshECdVtgvvvao0,3696
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
@@ -139,7 +139,7 @@ hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
 hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
 hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
-hjxdl-0.3.9.dist-info/METADATA,sha256=Jfl4N6_RI8chY8-LsWR28HByq4gFzuW00k0ANjYCl-0,861
-hjxdl-0.3.9.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
-hjxdl-0.3.9.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.3.9.dist-info/RECORD,,
+hjxdl-0.3.11.dist-info/METADATA,sha256=VXwzXBVPPZ25jzELbe1isXd_ZD83euLboJgyLOuxjNo,862
+hjxdl-0.3.11.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
+hjxdl-0.3.11.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.3.11.dist-info/RECORD,,