hjxdl 0.2.68__py3-none-any.whl → 0.2.70__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.68'
-__version_tuple__ = version_tuple = (0, 2, 68)
+__version__ = version = '0.2.70'
+__version_tuple__ = version_tuple = (0, 2, 70)

hdl/utils/desc/func_desc.py

@@ -5,20 +5,14 @@ TOOL_DESC = {
 ## 函数名：get_weather
 描述：用户询问一个城市的天气时，调用此工具获得此城市的天气信息
 参数：
-# city (str): 城市名
+- city (str): 城市名
 返回值 (str)：天气信息
         """,
-        "json": """
-需要返回的json
-{
-    "function_name": "get_weather",
-    "params":
-        {
-            "city": <city_name>
-        }
-}
-
-        """
+        "md": """
+需要返回的markdown:
+- function_name: get_weather
+- city: <city_name>
+"""
     },
 
     "get_datetime_by_cityname": {
@@ -26,19 +20,14 @@ TOOL_DESC = {
 ## 函数名：get_datetime_by_cityname
 描述：用户询问一个城市当前的日期或时间时，调用此工具可以获得此城市当前的日期和时间
 参数：
-# city (str): 城市名
+- city (str): 城市名
 返回值 (str):这个城市当前所在时区的日期和时间
         """,
-        "json": """
-需要给出的json
-{
-    "function_name": "get_datetime_by_cityname",
-    "params":
-        {
-            "city": <city_name>
-        }
-}
-        """
+        "md": """
+需要给出的markdown:
+- function_name: get_datetime_by_cityname
+- city: <city_name>
+"""
     },
 
     "execute_code": {
@@ -46,19 +35,14 @@ TOOL_DESC = {
 ## 函数名：execute_code,
 描述：当用户明确要求执行一段代码时，调用此工具，执行这段代码，返回执行结果。
 参数：
-# code (str): 用户要求执行的代码
+- code (str): 用户要求执行的代码
 返回值 (str): 执行结果
         """,
-        "json": """
-需要返回的json
-{
-    "function_name": "execute_code",
-    "params":
-        {
-            "code": <code>
-        }
-}
-        """
+        "md": """
+需要返回的markdown
+- function_name: execute_code
+- code: <code>
+"""
     },
 
     "calculate": {
@@ -66,19 +50,14 @@ TOOL_DESC = {
 ## 函数名：calculate，
 描述：当需要计算一个直接的在 python 中可交互的数据学算式，调用此工具，计算这个表达式，返回计算结果。
 参数：
-# expression (str): 用户要求的数学表达式
+- expression (str): 用户要求的数学表达式
 返回值 (str): 计算结果
         """,
-        "json": """
-需要返回的json
-{
-    "function_name": "calculate",
-    "params":
-        {
-            "expression": <expression>
-        }
-}
-        """
+        "md": """
+需要返回的markdown
+- function_name: calculate
+- expression: <expression>
+"""
     },
 
     "fetch_baidu_results": {
@@ -86,21 +65,16 @@ TOOL_DESC = {
 ## 函数名：fetch_baidu_results,
 描述：在用户提的问题需要联网搜索才能得到结果或者问题中有联网搜索的需求时，调用此工具，返回查询结果。
 参数：
-# query (str): 用户要求的查询内容
-# max_n_links (int, optional): 搜索条目的最大数目，若用户指定了数目，则使用用户指定的数目，若用户提问中没有指定，你需要在下面的json中"max_n_links"这一项指定为数值3。
+- query (str): 用户要求的查询内容
+- max_n_links (int, optional): 搜索条目的最大数目，若用户指定了数目，则使用用户指定的数目，默认数值为3。
 返回值 (str): 联网搜索到的信息
         """,
-        "json": """
-需要返回的json
-{
-    "function_name": "fetch_baidu_results",
-        "params":
-        {
-            "query": <query from user question>,
-            "max_n_links": <num of max results, 如果用户没有要求，这一项则指定为3>
-        }
-}
-        """
+        "md": """
+需要返回的markdown
+- function_name: fetch_baidu_results
+- query: <query from user question>
+- max_n_links: <num of max results, 如果用户没有要求，这一项则指定为3>
+"""
     },
 
     "wolfram_alpha_calculate": {
@@ -108,19 +82,14 @@ TOOL_DESC = {
 ## 函数名：wolfram_alpha_calculate
 描述：当用户要求计算一个明确用数学语言描述的表达式时，调用此工具，返回计算结果。注意不能描述物理、化学、应用题等非数学表达式。
 参数：
-# query (str): 用户要求的数学表达式，需要转换为英文，比如"integrate sin(x)"
+- query (str): 用户要求的数学表达式，需要转换为英文，比如"integrate sin(x)"
 返回值 (str): 计算结果
         """,
-        "json": """
-需要返回的json
-{
-    "function_name": "wolfram_alpha_calculate",
-    "params":
-        {
-            "query": <expression>
-        }
-}
-        """
+        "md": """
+需要返回的markdown:
+- function_name: wolfram_alpha_calculate
+- query: <expression>
+"""
     },
 
     "count_character_occurrences": {
@@ -128,20 +97,15 @@ TOOL_DESC = {
 ## 函数名：count_character_occurrences
 描述：当用户要求计算一段文本中某个字符出现的次数时，调用此工具，返回计算结果。
 参数：
-# text (str): 输入的单词或句子
-# char (str): 要统计的字符
+- text (str): 输入的单词或句子
+- char (str): 要统计的字符
 返回值 (str): 字符在文本中出现的次数
 """,
-    "json": """
-需要返回的json
-{
-    "function_name": "count_character_occurrences",
-    "params":
-        {
-            "text": <str>,
-            "char": <str>
-        }
-}
+    "md": """
+需要返回的markdown:
+- function_name: count_character_occurrences
+- text: <str>
+- char: <str>
 """
     },
 

hdl/utils/desc/template.py

@@ -1,18 +1,17 @@
 FN_TEMPLATE = """
 
 你是一个可以调用函数来执行任务的人工智能助手，根据用户最后的问题来决定是否调用函数
-你的回答格式是固定的：若要调用函数，需要使用包含两个字段的JSON对象进行响应，并且不应包含其他多余文字，避免出现格式化问题：
+你的回答格式是固定的：若要调用函数，需要使用包含如下信息 markdown 进行响应，并且不应包含其他多余文字，避免出现格式化问题：
 
-"function_name"：需要要调用的函数的名称。
-"params"：函数所需的参数，一般为字符串，注意字符串要在双引号 "" 之间。
+- function_name: 需要要调用的函数的名称。
+- <param1>: 函数所需的参数1，<>中为具体的参数名。
+- <param2>: 函数所需的参数2，后面更多的参数也依照同样的规则。
 
-若不需要调用函数，或没达到如下函数的触发条件时，则返回如下的json，并且不应包含其他多余文字，避免出现格式化问题：
-{
-    "function_name": null
-}
+若不需要调用函数，或没达到如下函数的触发条件时，则返回如下的 markdown 字符串，并且不应包含其他多余文字，避免出现格式化问题：
+- function_name: None
 
 ### 函数：
-以下是可用于与系统交互的函数列表，每个函数以 “##” 作为标记开始，并带有“触发条件”说明，每个参数名会以 “#” 作为标记。
+以下是可用于与系统交互的函数列表，每个函数以 “##” 作为标记开始，并带有“触发条件”说明，每个参数名会以 “-” 作为标记。
 每个函数都有特定的参数和要求说明，确保仔细遵循每个功能的说明。根据最后用户的问题和函数的触发条件判断要执行的任务选择合适的一个函数。请严格遵守每个函数的触发条件，以说明中的JSON格式提供函数调用所需要的参数，其中参数的具体值从用户的提问中获取，并且不能带“<>”符号：
 
 """
@@ -20,9 +19,22 @@ FN_TEMPLATE = """
 COT_TEMPLATE = """
 你是一个专家级AI助手，有能针对一个问题和当前解决到的步骤给出下一步该进行操作的能力。
 以JSON格式回复，该步骤对应的 json 包括'title'、'tool', 'content'和'stop_thinking',
-若已有的信息可以回答用户问题，则'stop_thinking'键的值为 true 并把答案写在'content'键的值中，若还需要下一步操作，'stop_thinking'键的则为 false 并在 'title' 键对应的值中给出该进行的操作；
+若已有的信息不合理或者针对用户的问题提供的信息有错误，则'stop_thinking'键的值为 false 并且继续思考过程。
+若还需要下一步操作，'stop_thinking'键的则为 false 并在 'title' 键对应的值中给出该进行的操作；
+若已有的信息足够回答用户问题，则'stop_thinking'键的值为 true 并把答案写在'content'键的值中。
 若本步骤的操作不需要调用下文中提到的工具或信息足够回答用户问题，而不必包含'tool'键，若需要调用下文中的工具，则必须包含'tool'键，并且其值是工具函数名称。
 你的回答中应只能是 json 格式，且不能包含其他多余文字不能有格式错误。
 
+你是一个专家级 AI 助手，有能力针对一个问题和当前解决到的步骤，给出下一步的操作建议。
+
+以 Markdown 格式回复，该步骤对应的 Markdown 包括以下部分：
+## <title> 用二级标题（##）标记步骤名称，且不包含<>组合符号。
+- tool<可选>: <若需要调用工具，列出工具名称；如果不需要，则忽略该部分>
+- content: <描述详细内容或答案>
+- stop_thinking: <true/false>
+若已有的信息不合理或者针对用户的问题提供的信息有错误，则需要进一步思考，并说明原因。若还需要下一步操作，给出具体建议。
+若已有的信息足够回答用户问题，则直接回答用户问题。
+
+你的回答中应只能是 Markdown 格式，且不能包含其他多余文字或格式错误。
 以下是可用的工具：
 """

hdl/utils/llm/chat.py

@@ -4,6 +4,7 @@ import os
 from concurrent.futures import ProcessPoolExecutor
 import subprocess
 from typing import Generator
+import re
 
 
 from openai import OpenAI
@@ -12,6 +13,53 @@ from ..desc.func_desc import TOOL_DESC
 import json
 # import traceback
 
+def parse_fn_markdown(markdown_text, params_key="params"):
+    lines = markdown_text.strip().split("\n")
+    result = {}
+    params = {}
+
+    for line in lines:
+        # 使用正则提取 key 和 value
+        match = re.match(r"-\s*(\w+):\s*(.+)", line.strip())
+        if match:
+            key, value = match.groups()
+            if key == "function_name":
+                result[key] = value  # 固定的 function_name
+            else:
+                params[key] = value  # 其他的进入 params（或替代键名）
+
+    # 将提取的参数嵌套到指定的键名下
+    result[params_key] = params
+    return result
+
+def parse_cot_markdown(markdown_text):
+    # 提取标题
+    title_match = re.search(r"##\s*(.+)", markdown_text)
+    title = title_match.group(1) if title_match else None
+    title = title.replace("：", "").replace(":", "").replace(" ", "")
+
+    # 提取工具
+    tool_match = re.search(r"tool\s*(.+)", markdown_text)
+    tool = tool_match.group(1) if tool_match else None
+    tool = tool.replace("：", "").replace(":", "").replace(" ", "")
+
+    # 提取内容
+    content_match = re.search(r"content\s*(.+)", markdown_text)
+    content = content_match.group(1) if content_match else None
+    content = content.replace("：", "").replace(":", "").replace(" ", "")
+
+    # 提取停止思考
+    stop_thinking_match = re.search(r"stop_thinking\s*(.+)", markdown_text)
+    stop_thinking = stop_thinking_match.group(1) == "true" or stop_thinking_match.group(1) == "True" if stop_thinking_match else None
+
+
+    # 组装为字典
+    return {
+        "title": title,
+        "tool": tool,
+        "content": content,
+        "stop_thinking": stop_thinking
+    }
 
 def chat_oai_stream(
     base_url="http://127.0.0.1:8000/v1",
@@ -202,7 +250,7 @@ class OpenAI_M():
 
             try:
                 # 将思考结果解析为JSON格式，以便后续处理
-                step_json = json.loads(resp)
+                step_json = parse_cot_markdown(resp)
                 # print(step_json)
                 # 将当前思考步骤添加到步骤列表中
                 steps.append(step_json)
@@ -473,7 +521,8 @@ class OpenAI_M():
             **kwargs
         )
         try:
-            decision_dict = json.loads(decision_dict_str)
+            decision_dict = parse_fn_markdown(decision_dict_str)
+            # decision_dict = json.loads(decision_dict_str)
         except Exception as e:
             print(e)
             return ""

{hjxdl-0.2.68.dist-info → hjxdl-0.2.70.dist-info}/METADATA

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

{hjxdl-0.2.68.dist-info → hjxdl-0.2.70.dist-info}/RECORD

@@ -1,5 +1,5 @@
 hdl/__init__.py,sha256=GffnD0jLJdhkd-vo989v40N90sQbofkayRBwxc6TVhQ,72
-hdl/_version.py,sha256=TD22ilmsCxAfrwjbRrqwKGfjPxJbditO0KzSuHXXUJU,413
+hdl/_version.py,sha256=LRulaxMCVBJJM9jsOgQlfsrR1gcTZooiLo_gZ7p0U70,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
@@ -122,13 +122,13 @@ hdl/utils/database_tools/connect.py,sha256=xCacGucKxlQUXs6AsNddpeECvdqT1180V1ZWq
 hdl/utils/database_tools/datetime.py,sha256=xqE2xNiOpADzX-R8_bM0bioJRF3Ay9Jp1CAG6dy6uVI,1202
 hdl/utils/database_tools/web.py,sha256=PJ7R1chUD-vz-Up0orfFqdrZq3CFOlwiO9gIjFSb-R8,5224
 hdl/utils/desc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/desc/func_desc.py,sha256=DDbcLIW-2Hk_WSRg2NO8PF7S338PmbssGcjvMa_P8Dk,3905
-hdl/utils/desc/template.py,sha256=WcoyRDfLzYCwhV9MGnisjdjAeiFr0z8iZ7dSrol7Je0,2121
+hdl/utils/desc/func_desc.py,sha256=DPcYH1xDSlY5kkxzyE3a1QK0s71P2Hrn52vLd9j-q-I,3362
+hdl/utils/desc/template.py,sha256=3DlJK4-n3_nDrK8hO6XHyNmwmMeKHwxb9dqzOxUC8uM,3137
 hdl/utils/general/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hdl/utils/general/glob.py,sha256=8-RCnt6L297wMIfn34ZAMCsGCZUjHG3MGglGZI1cX0g,491
 hdl/utils/general/runners.py,sha256=EFTk9PeVnqc51yU4b-HxsLwJyMA7dFMZBsdAa-VZ-sQ,2880
 hdl/utils/llm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-hdl/utils/llm/chat.py,sha256=f4hxeZG9stwO9pmlCW6ki-R3V-NOCJF7dgn0mTHCeCo,20126
+hdl/utils/llm/chat.py,sha256=4waDVnCFesL2tO0SC7D1padknwFywd8J-v3KmmatmJ8,21902
 hdl/utils/llm/chatgr.py,sha256=A9SCWAvQulgO2x0ArIqcmgd_iVWVbSi1Oj15Diwb3Ok,3743
 hdl/utils/llm/embs.py,sha256=Tf0FOYrOFZp7qQpEPiSCXzlgyHH0X9HVTUtsup74a9E,7174
 hdl/utils/llm/extract.py,sha256=2sK_WJzmYIc8iuWaM9DA6Nw3_6q1O4lJ5pKpcZo-bBA,6512
@@ -139,7 +139,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.2.68.dist-info/METADATA,sha256=xa-GC1TfYXJAhm-MPNwACg6tvOYfGxMHw71alslSNZk,836
-hjxdl-0.2.68.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-hjxdl-0.2.68.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
-hjxdl-0.2.68.dist-info/RECORD,,
+hjxdl-0.2.70.dist-info/METADATA,sha256=CE8I8hb5F7ETz8ba9kVF2ZmnoV8KG3Zogd-a-PpMB_E,836
+hjxdl-0.2.70.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+hjxdl-0.2.70.dist-info/top_level.txt,sha256=-kxwTM5JPhylp06z3zAVO3w6_h7wtBfBo2zgM6YZoTk,4
+hjxdl-0.2.70.dist-info/RECORD,,