hjxdl 0.1.82__py3-none-any.whl → 0.1.84__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.82'
-__version_tuple__ = version_tuple = (0, 1, 82)
+__version__ = version = '0.1.84'
+__version_tuple__ = version_tuple = (0, 1, 84)

hdl/utils/llm/chat.py

@@ -186,22 +186,23 @@ class OpenAI_M():
                 })
         else:
             content = prompt
+
         response = self.client.chat.completions.create(
             messages=[{
                 "role": "user",
                 "content": content
             }],
-            stream=False,
+            stream=stream,
             model=model,
             **kwargs
         )
-        if stream:
+        if not stream:
+            return response.choices[0].message.content
+        else:
             for chunk in response:
                 content = chunk.choices[0].delta.content
                 if content:
                     yield content
-        else:
-            return response.choices[0].message.content
 
     def invoke(
         self,

{hjxdl-0.1.82.dist-info → hjxdl-0.1.84.dist-info}/METADATA

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

{hjxdl-0.1.82.dist-info → hjxdl-0.1.84.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=uD0kkklP47KEgakV5d1QibkwP9W5l8Jh1sG9ohg63cg,413
+hdl/_version.py,sha256=d0L59EfPQwDJVDuCyDuNn4UyYbrKwPsGT6cg6MNTnD0,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
@@ -127,7 +127,7 @@ hdl/utils/desc/template.py,sha256=a0UAkkKctt_EHY9UECsIIAwVkGPcM1Hr01HSkRMeIuw,12
 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=uZQuiQKL0i5cRSiJq0CPQOPS2FUmDE90mZzvk9M8FK4,11742
+hdl/utils/llm/chat.py,sha256=XU_tEwRGRDlg7KPdLetE3K6dbD3aPcmWGU7Vd56g0XQ,11748
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
 hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
@@ -136,7 +136,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.1.82.dist-info/METADATA,sha256=fajRkx2t1HdlBLjWP43OH9OlXsYTM4V7tOZEhfZ49zA,903
-hjxdl-0.1.82.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
-hjxdl-0.1.82.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.82.dist-info/RECORD,,
+hjxdl-0.1.84.dist-info/METADATA,sha256=NmWiAiAY4fIeJrJKSEs_FsD7wI76d8QafZW_N4zNWhI,903
+hjxdl-0.1.84.dist-info/WHEEL,sha256=GV9aMThwP_4oNCtvEC2ec3qUYutgWeAzklro_0m4WJQ,91
+hjxdl-0.1.84.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.84.dist-info/RECORD,,