hjxdl 0.1.42__py3-none-any.whl → 0.1.44__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.42'
-__version_tuple__ = version_tuple = (0, 1, 42)
+__version__ = version = '0.1.44'
+__version_tuple__ = version_tuple = (0, 1, 44)

hdl/models/utils.py

@@ -49,10 +49,10 @@ def load_model(
         init_args = checkpoint['init_args']
         assert model_name is not None
         model = MODEL_DICT[model_name](**init_args)
-        model.load_state_dict( 
-            checkpoint['model_state_dict'], 
+        model.load_state_dict(
+            checkpoint['model_state_dict'],
         )
-    
+
     elif isinstance(model, nn.DataParallel):
         state_dict = checkpoint['model_state_dict']
         from collections import OrderedDict
@@ -66,10 +66,10 @@ def load_model(
             new_state_dict[k] = v
         model.load_state_dict(new_state_dict)
     else:
-        model.load_state_dict( 
-            checkpoint['model_state_dict'], 
+        model.load_state_dict(
+            checkpoint['model_state_dict'],
         )
-    
+
     if optimizer is not None:
         optimizer.load_state_dict(checkpoint['optimizer_state_dict'])
     epoch = checkpoint.get('epoch', 0)
@@ -80,4 +80,4 @@ def load_model(
     else:
         model.eval()
 
-    return model, optimizer, epoch, loss
+    return model, optimizer, epoch, loss

hdl/utils/desc/func_desc.py

@@ -1,7 +1,7 @@
 FN_DESC = {
     "get_weather": """
 ## 函数名：get_weather
-描述：在用户询问一个城市的天气时，调用此工具获得此城市的天气信息
+描述：只有在用户询问一个城市的天气时，调用此工具获得此城市的天气信息
 参数：
 # city (str): 城市名
 返回值 (str)：天气信息
@@ -17,7 +17,7 @@ FN_DESC = {
 """,
     "get_datetime_by_cityname": """
 ## 函数名：get_datetime_by_cityname
-描述：在用户询问一个城市当前的日期或时间时，调用此工具可以获得此城市当前的日期和时间
+描述：只有在用户询问一个城市当前的日期或时间时，调用此工具可以获得此城市当前的日期和时间
 参数：
 # city (str): 城市名
 返回值 (str):这个城市当前所在时区的日期和时间
@@ -33,7 +33,7 @@ FN_DESC = {
 """,
     "web_search_text": """
 ## 函数名：web_search_text
-描述：在用户明确要联网查询回答他的问题时，调用此工具可以获得该问题联网搜索的相关内容
+描述：在用户明确要联网查询回答他的问题时，调用此工具可以获得该问题联网搜索的相关内容，若用户的提问中没有提到“联网查询”，则不要调用此工具
 参数：
 # query_text (str): 从用户提的问题中获取，用于在网络中搜索信息
 # max_results (int, optional): 搜索条目的最大数目，若用户指定了数目，则使用用户指定的数目，若用户提问中没有指定，你需要在下面的json中"max_results"这一项指定为数值3。

hdl/utils/desc/template.py

@@ -6,7 +6,7 @@ FN_TEMPLATE = """
 "function_name"：需要要调用的函数的名称。
 "params"：函数所需的参数，一般为字符串，注意字符串要在双引号 "" 之间。
 
-若不需要调用函数，或提供的函数无法解决用户的问题时，则返回如下字典，并且不应包含其他多余文字，避免出现格式化问题：
+若不需要调用函数，或提供的函数无法解决用户的问题时，或没有达到下文中相关函数描述的触发条件时，则返回如下字典，并且不应包含其他多余文字，避免出现格式化问题：
 {
     "function_name": null
 }

hdl/utils/llm/chat.py

@@ -87,6 +87,15 @@ def chat_oai_invoke(
     return response.choices[0].message.content
 
 def run_tool_with_kwargs(tool, func_kwargs):
+    """Run the specified tool with the provided keyword arguments.
+
+    Args:
+        tool (callable): The tool to be executed.
+        func_kwargs (dict): The keyword arguments to be passed to the tool.
+
+    Returns:
+        The result of executing the tool with the provided keyword arguments.
+    """
     return tool(**func_kwargs)
 
 class OpenAI_M():
@@ -231,7 +240,15 @@ class OpenAI_M():
                 return self.invoke(prompt_final, **kwargs)
 
     def get_decision(self, prompt: str, **kwargs: t.Any):
-        # 该方法与之前一致...
+        """Get decision based on the given prompt.
+
+        Args:
+            prompt (str): The prompt for decision making.
+            **kwargs: Additional keyword arguments for decision making.
+
+        Returns:
+            str: The decision dictionary string.
+        """
         prompt_final = FN_TEMPLATE
         for tool in self.tools:
             prompt_final += self.tool_desc.get(tool.__name__, "")
@@ -241,7 +258,15 @@ class OpenAI_M():
         return decision_dict_str
 
     def get_tool_result(self, prompt: str, **kwargs: t.Any):
-        # 同步方法与之前一致...
+        """Get the result of a tool based on the decision made.
+
+        Args:
+            prompt (str): The prompt to make a decision.
+            **kwargs: Additional keyword arguments.
+
+        Returns:
+            str: The result of the tool.
+        """
         decision_dict_str = self.get_decision(prompt, **kwargs)
         try:
             decision_dict = json.loads(decision_dict_str)

{hjxdl-0.1.42.dist-info → hjxdl-0.1.44.dist-info}/METADATA

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

{hjxdl-0.1.42.dist-info → hjxdl-0.1.44.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=bifgu3Apk3qbd-fEq7Ym58g4dYRITzNZrlo7DOythrE,413
+hdl/_version.py,sha256=IIqs5uWJ8FFsWjDm6cFCsir1SFd1NDSs1R-SbdEIUJQ,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
@@ -108,7 +108,7 @@ hdl/models/model_dict.py,sha256=uRh13nch7rab3DajnDR_usp_1R0b37WUgNoiqOobk-s,402
 hdl/models/norm_flows.py,sha256=nROlAaToUOtqi99_BPe6rlPHpUS38YFAtDll1xmGv5U,830
 hdl/models/optim_dict.py,sha256=xvefZwCGebIXhRLSbcoGa3WVUWW2kEEm0Gsgp8Q9SQw,231
 hdl/models/rxn.py,sha256=6MEkzjWEgWgl14EkdprvVtycX1q-xlYnXekM1ROKFL0,1557
-hdl/models/utils.py,sha256=OOdwZ7f5ciaEFNM9zouNpLxj8S8kGKsSIM0msnoJ3bQ,2221
+hdl/models/utils.py,sha256=LgeZhsGw3_VAjgMFB9vGamIFoe9UAHM21xhDIRA8vtE,2210
 hdl/ops/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/ops/utils.py,sha256=GIe95pJYVqOQQ91ELb1tfogVFzGg2VJdQmIFX8aHeyM,1053
 hdl/optims/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -122,12 +122,12 @@ hdl/utils/database_tools/connect.py,sha256=KUnVG-8raifEJ_N0b3c8LkTTIfn9NIyw8LX6q
 hdl/utils/database_tools/datetime.py,sha256=xqE2xNiOpADzX-R8_bM0bioJRF3Ay9Jp1CAG6dy6uVI,1202
 hdl/utils/database_tools/web.py,sha256=vZJYWA02QbyUUMGLDX710cexn7RlfWeHa8YwK3UsDZ0,1934
 hdl/utils/desc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/desc/func_desc.py,sha256=z7KXjTyE-tr0VZ1ZtA7G7phSfuq48lUaTwWb0LRyY6I,1626
-hdl/utils/desc/template.py,sha256=a3NcSihzZMm9Bk76iDVe54_xBDceGmLebS0XMONE3nk,1172
+hdl/utils/desc/func_desc.py,sha256=wjGAfMPmpj_iSNJvi6vpRBDIuQGF318aL44wRdz8TK0,1719
+hdl/utils/desc/template.py,sha256=K45KX6LwHXW-VcPa5jUJRUDmUQMX0-zSi3tXUHlyMG8,1235
 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=a4mIjxNp2tlW3eAc0tkG_eM_Xqg65DAIHjgTsGRRBl0,9990
+hdl/utils/llm/chat.py,sha256=Yo4moFbRk0y4XN7GkLvoQD9p-8XgR-mVj9LcYFGv-PE,10749
 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
@@ -135,7 +135,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.42.dist-info/METADATA,sha256=Xkyg_wZMUlb_B9u38B9daXVPIDA1CQK-_yfJ_PEHEO4,737
-hjxdl-0.1.42.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-hjxdl-0.1.42.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.42.dist-info/RECORD,,
+hjxdl-0.1.44.dist-info/METADATA,sha256=Ax6NueiUU7GZifaDxk2uti5gMBfSoXLTqyrdZw_-xMc,737
+hjxdl-0.1.44.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+hjxdl-0.1.44.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.44.dist-info/RECORD,,