hjxdl 0.1.11__py3-none-any.whl → 0.1.12__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.1.11'
16
- __version_tuple__ = version_tuple = (0, 1, 11)
15
+ __version__ = version = '0.1.12'
16
+ __version_tuple__ = version_tuple = (0, 1, 12)
hdl/utils/llm/chat.py CHANGED
@@ -347,7 +347,7 @@ class OpenAI_M():
347
347
  api_key (str): API key for authentication.
348
348
  client (OpenAI): OpenAI client for making API requests.
349
349
  """
350
- self.model_path = model_path
350
+ # self.model_path = model_path
351
351
  self.server_ip = server_ip
352
352
  self.server_port = server_port
353
353
  self.base_url = f"http://{self.server_ip}:{str(self.server_port)}/v1"
@@ -377,12 +377,13 @@ class OpenAI_M():
377
377
  str: The response generated by the chatbot.
378
378
  """
379
379
  response = self.client.chat.completions.create(
380
- model=self.model_path,
380
+ # model=self.model_path,
381
381
  messages=[{
382
382
  "role": "user",
383
383
  "content": prompt
384
384
  }],
385
385
  stream=False,
386
+ # model=kwargs.get("model", "default_model")
386
387
  **kwargs
387
388
  )
388
389
  return response.choices[0].message.content
@@ -405,12 +406,13 @@ class OpenAI_M():
405
406
  str: The generated text completion in a streaming fashion.
406
407
  """
407
408
  response = self.client.chat.completions.create(
408
- model=self.model_path,
409
+ # model=self.model_path,
409
410
  messages=[{
410
411
  "role": "user",
411
412
  "content": prompt
412
413
  }],
413
414
  stream=True,
415
+ # model=kwargs.get("model", "default_model")
414
416
  **kwargs
415
417
  )
416
418
 
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: hjxdl
3
- Version: 0.1.11
3
+ Version: 0.1.12
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=5sZZNySv08wwfzJcSDssGTqUn9wlmDsR6R4XB8J8mFM,70
2
- hdl/_version.py,sha256=HreDwlLXV189L3kiBj3huM_kqWD1usijlC8LN1YXcCM,413
2
+ hdl/_version.py,sha256=cuLjhzc4k2qL9vS7TDaPLs82wisSY2yFKQtDhC08Kis,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
@@ -84,12 +84,12 @@ hdl/utils/database_tools/connect.py,sha256=KUnVG-8raifEJ_N0b3c8LkTTIfn9NIyw8LX6q
84
84
  hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
85
85
  hdl/utils/general/glob.py,sha256=8-RCnt6L297wMIfn34ZAMCsGCZUjHG3MGglGZI1cX0g,491
86
86
  hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
87
- hdl/utils/llm/chat.py,sha256=WElHLIi9kEU2v_42moqSp9b4UYCwB5xow1rTsi1HMRc,14854
87
+ hdl/utils/llm/chat.py,sha256=FtpQjp87FIBSIOQBPYKjtscA13RpMRwLk7sKlMAnOQ0,14974
88
88
  hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
89
89
  hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
90
90
  hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
91
91
  hdl/utils/schedulers/norm_lr.py,sha256=bDwCmdEK-WkgxQMFBiMuchv8Mm7C0-GZJ6usm-PQk14,4461
92
- hjxdl-0.1.11.dist-info/METADATA,sha256=0xzd32HG6sfj4SDK9zdrwb-fD5leA6JqunVVTB4wpc0,543
93
- hjxdl-0.1.11.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
94
- hjxdl-0.1.11.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
95
- hjxdl-0.1.11.dist-info/RECORD,,
92
+ hjxdl-0.1.12.dist-info/METADATA,sha256=oM9y6W0hDN6TupHYR2OECwnI1sI7BLlpWqkbOV9slK0,543
93
+ hjxdl-0.1.12.dist-info/WHEEL,sha256=R0nc6qTxuoLk7ShA2_Y-UWkN8ZdfDBG2B6Eqpz2WXbs,91
94
+ hjxdl-0.1.12.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
95
+ hjxdl-0.1.12.dist-info/RECORD,,
File without changes