hjxdl 0.1.46__py3-none-any.whl → 0.1.48__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.46'
-__version_tuple__ = version_tuple = (0, 1, 46)
+__version__ = version = '0.1.48'
+__version_tuple__ = version_tuple = (0, 1, 48)

hdl/utils/llm/chat.py

@@ -239,7 +239,10 @@ class OpenAI_M():
             else:
                 return self.invoke(prompt_final, **kwargs)
 
-    def get_decision(self, prompt: str, **kwargs: t.Any):
+    def get_decision(
+        self, prompt: str,
+        **kwargs: t.Any,
+    ):
         """Get decision based on the given prompt.
 
         Args:
@@ -249,7 +252,8 @@ class OpenAI_M():
         Returns:
             str: The decision dictionary string.
         """
-        prompt_final = FN_TEMPLATE
+        fn_template = kwargs.get("fn_template", FN_TEMPLATE)
+        prompt_final = fn_template
         for tool in self.tools:
             prompt_final += self.tool_desc.get(tool.__name__, "")
         prompt_final += f"\n用户的问题：\n{prompt}"
@@ -257,7 +261,11 @@ class OpenAI_M():
         print(decision_dict_str)
         return decision_dict_str
 
-    def get_tool_result(self, prompt: str, **kwargs: t.Any):
+    def get_tool_result(
+        self,
+        prompt: str,
+        **kwargs: t.Any
+    ):
         """Get the result of a tool based on the decision made.
 
         Args:
@@ -267,7 +275,10 @@ class OpenAI_M():
         Returns:
             str: The result of the tool.
         """
-        decision_dict_str = self.get_decision(prompt, **kwargs)
+        decision_dict_str = self.get_decision(
+            prompt,
+            **kwargs
+        )
         try:
             decision_dict = json.loads(decision_dict_str)
         except Exception as e:
@@ -287,7 +298,11 @@ class OpenAI_M():
                 print(e)
                 return ""
 
-    async def get_tool_result_async(self, prompt: str, **kwargs: t.Any):
+    async def get_tool_result_async(
+        self,
+        prompt,
+        **kwargs: t.Any
+    ):
         """
         Asynchronous version of the get_tool_result function that can run in parallel using multiprocessing.
 

{hjxdl-0.1.46.dist-info → hjxdl-0.1.48.dist-info}/METADATA

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

{hjxdl-0.1.46.dist-info → hjxdl-0.1.48.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=GIP30BERZTiARroobX5RavhGe2QI6H5rhUq7zTSjgD4,413
+hdl/_version.py,sha256=lZPH5wBKKByJEDRWOxZ8kP20Dhk7U4w8vFxTT3OwcOY,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=Yo4moFbRk0y4XN7GkLvoQD9p-8XgR-mVj9LcYFGv-PE,10749
+hdl/utils/llm/chat.py,sha256=hoB6GugoxFofhK2xECe_dA3NUHxv0Wu8kctg5j6Mdcc,10922
 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.46.dist-info/METADATA,sha256=IwtnV_pVUjub8cBdwv36ZfmdL3NrwleMs3aVGiU2xLY,737
-hjxdl-0.1.46.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
-hjxdl-0.1.46.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.1.46.dist-info/RECORD,,
+hjxdl-0.1.48.dist-info/METADATA,sha256=1q2IJyFBetdVDWBAU1FJedAaUg-JxCqLmVwXodTAsf4,737
+hjxdl-0.1.48.dist-info/WHEEL,sha256=UvcQYKBHoFqaQd6LKyqHw9fxEolWLQnlzP0h_LgJAfI,91
+hjxdl-0.1.48.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.1.48.dist-info/RECORD,,