hjxdl 0.3.9__py3-none-any.whl → 0.3.11__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
hdl/_version.py CHANGED
@@ -12,5 +12,5 @@ __version__: str
12
12
  __version_tuple__: VERSION_TUPLE
13
13
  version_tuple: VERSION_TUPLE
14
14
 
15
- __version__ = version = '0.3.9'
16
- __version_tuple__ = version_tuple = (0, 3, 9)
15
+ __version__ = version = '0.3.11'
16
+ __version_tuple__ = version_tuple = (0, 3, 11)
hdl/utils/llm/chat.py CHANGED
@@ -159,8 +159,8 @@ class OpenAI_M:
159
159
  self.client_conf = client_conf
160
160
 
161
161
  # self.clients = {}
162
- for client_id, conf in self.client_conf.items():
163
- conf[client_id]["client"] = OpenAI(
162
+ for _, conf in self.client_conf.items():
163
+ conf["client"] = OpenAI(
164
164
  base_url=conf["host"],
165
165
  api_key=conf.get("api_key", "dummy_key"),
166
166
  *args,
@@ -281,14 +281,14 @@ class OpenAI_M:
281
281
 
282
282
  def get_resp(
283
283
  self,
284
- client_id,
285
284
  prompt: str,
285
+ client_id: str = None,
286
286
  sys_info: str = None,
287
287
  assis_info: str = None,
288
288
  images: list = None,
289
289
  image_keys: tuple = ("image_url", "url"),
290
290
  stop: list[str] | None = ["USER:", "ASSISTANT:"],
291
- model="default_model",
291
+ model: str=None,
292
292
  stream: bool = True,
293
293
  **kwargs: t.Any,
294
294
  ):
@@ -307,6 +307,8 @@ class OpenAI_M:
307
307
  Returns:
308
308
  _type_: The response object from the model.
309
309
  """
310
+ if not model:
311
+ model = self.client_conf[client_id]["model"]
310
312
 
311
313
  # Initialize the content list with at least the user's text input
312
314
  content = [
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: hjxdl
3
- Version: 0.3.9
3
+ Version: 0.3.11
4
4
  Summary: A collection of functions for Jupyter notebooks
5
5
  Home-page: https://github.com/huluxiaohuowa/hdl
6
6
  Author: Jianxing Hu
@@ -1,5 +1,5 @@
1
1
  hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
2
- hdl/_version.py,sha256=nV2HEiFwTdaOZoFEyVxxG_D8Oq_nlSmX2vHL4jK4h6w,411
2
+ hdl/_version.py,sha256=cgZZwxO2Czf2RAwPc5Ugf8lBeLEkSyVqWm7FWjhlv2A,413
3
3
  hdl/args/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
4
4
  hdl/args/loss_args.py,sha256=s7YzSdd7IjD24rZvvOrxLLFqMZQb9YylxKeyelSdrTk,70
5
5
  hdl/controllers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -128,7 +128,7 @@ hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU
128
128
  hdl/utils/general/glob.py,sha256=Zuf7WHU0UdUPOs9UrhxmrCiMC8GrHxQU6n3mTThv6yc,1120
129
129
  hdl/utils/general/runners.py,sha256=x7QBolp3MrqNV6L4rB6Ueybr26bqkRFZTuXhY0SwyLk,3061
130
130
  hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
131
- hdl/utils/llm/chat.py,sha256=u59sIk6e-4-uM-XTR-EjtdKM-rtxlGzYdvNINH4fDAw,25784
131
+ hdl/utils/llm/chat.py,sha256=9xAq8WBav8r0rayp66cPTZhAzEmMgTbtvDK7vP7rMcc,25850
132
132
  hdl/utils/llm/chatgr.py,sha256=5F5PJHe8vz3iCfi4TT54DCLRi1UeJshECdVtgvvvao0,3696
133
133
  hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
134
134
  hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
@@ -139,7 +139,7 @@ hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hS
139
139
  hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
140
140
  hdl/utils/weather/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
141
141
  hdl/utils/weather/weather.py,sha256=k11o6wM15kF8b9NMlEfrg68ak-SfSYLN3nOOflFUv-I,4381
142
- hjxdl-0.3.9.dist-info/METADATA,sha256=Jfl4N6_RI8chY8-LsWR28HByq4gFzuW00k0ANjYCl-0,861
143
- hjxdl-0.3.9.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
144
- hjxdl-0.3.9.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
145
- hjxdl-0.3.9.dist-info/RECORD,,
142
+ hjxdl-0.3.11.dist-info/METADATA,sha256=VXwzXBVPPZ25jzELbe1isXd_ZD83euLboJgyLOuxjNo,862
143
+ hjxdl-0.3.11.dist-info/WHEEL,sha256=A3WOREP4zgxI0fKrHUG8DC8013e3dK3n7a6HDbcEIwE,91
144
+ hjxdl-0.3.11.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
145
+ hjxdl-0.3.11.dist-info/RECORD,,
File without changes