hjxdl 0.2.98__py3-none-any.whl → 0.3.1__py3-none-any.whl

hdl/_version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.2.98'
-__version_tuple__ = version_tuple = (0, 2, 98)
+__version__ = version = '0.3.1'
+__version_tuple__ = version_tuple = (0, 3, 1)

hdl/utils/llm/chat.py

@@ -450,12 +450,12 @@ class OpenAI_M():
             str: The agent response based on the prompt.
         '''
         """
-        decision_dict = self.get_decision(prompt, **kwargs)
-        decision_dict = parse_fn_markdown(decision_dict)
+        decision_dict_str = self.get_decision(prompt, **kwargs)
+        decision_dict = parse_fn_markdown(decision_dict_str)
         if decision_dict.get("function_name", None) is None:
             return self.chat(prompt=prompt, stream=stream, **kwargs)
         else:
-            tool_result = str(self.get_tool_result(decision_dict))
+            tool_result = str(self.get_tool_result(decision_dict_str))
             prompt_final = "根据上下文回答最后的用户问题：\n上下文信息：\n"
             prompt_final += tool_result
             # prompt_final += f"\n用户的问题：\n{prompt}"
@@ -502,7 +502,7 @@ class OpenAI_M():
 
     def get_tool_result(
         self,
-        decision_dict: dict,
+        decision_dict_str: str,
     ):
         """Get the result of a tool based on the decision made.
 
@@ -513,7 +513,7 @@ class OpenAI_M():
         Returns:
             str: The result of the tool.
         """
-
+        decision_dict = parse_fn_markdown(decision_dict_str)
         func_name = decision_dict.get("function_name", None)
         if func_name is None:
             return ""

hdl/utils/llm/vis.py

@@ -257,7 +257,7 @@ def draw_and_plot_boxes_from_json(
 
     buf.close()  # Close the buffer after use
 
-    return pil_image
+    return pil_image, save_path
 
 class ImgHandler:
     """

{hjxdl-0.2.98.dist-info → hjxdl-0.3.1.dist-info}/METADATA

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

{hjxdl-0.2.98.dist-info → hjxdl-0.3.1.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=rXnGuKlRz3b8ob_wArCKVOnw9u7jH84c8oNEr3jDE7I,413
+hdl/_version.py,sha256=HzPz9rq3s1AiZXregKlqKaJJ2wGMtvH_a3V9la9CnpM,411
 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,18 +128,18 @@ 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=s0L9d_RFQzQoFdPwmPVcvQYMQhpcS-ZpflhBlhh3Qu4,26188
+hdl/utils/llm/chat.py,sha256=0MXmw9FP10uR_NRJrSUhcrK88zE9oIdgGRbZp5l4N74,26263
 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
 hdl/utils/llm/llama_chat.py,sha256=watcHGOaz-bv3x-yDucYlGk5f8FiqfFhwWogrl334fk,4387
-hdl/utils/llm/vis.py,sha256=TM365x-W5pZt4VFFmpICt3TuWkENYR0uWoRVxu9xW4s,26044
+hdl/utils/llm/vis.py,sha256=SSP6tOwKLq0hWcpM3twI9TitqzBmKjlcGrnXEWYlCzM,26055
 hdl/utils/llm/visrag.py,sha256=0i-VrxqgiV-J7R3VPshu9oc7-rKjFJOldYik3HDXj6M,10176
 hdl/utils/schedulers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 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.2.98.dist-info/METADATA,sha256=Qrkdn1LusABXCpfUnTMiS8K-kVyjfEsI1XncnTme4nA,862
-hjxdl-0.2.98.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-hjxdl-0.2.98.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.98.dist-info/RECORD,,
+hjxdl-0.3.1.dist-info/METADATA,sha256=k2dskJLIIEhVi2HMpjQfEFSoyVlNhxGiY3C6rgPy78k,861
+hjxdl-0.3.1.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+hjxdl-0.3.1.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.3.1.dist-info/RECORD,,