beswarm 0.1.12__py3-none-any.whl → 0.1.13__py3-none-any.whl

beswarm/aient/test/chatgpt.py

@@ -0,0 +1,161 @@
+function_call_list = \
+{
+    "base": {
+        "tools": [],
+        "tool_choice": "auto"
+    },
+    "current_weather": {
+        "name": "get_current_weather",
+        "description": "Get the current weather in a given location",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "location": {
+                    "type": "string",
+                    "description": "The city and state, e.g. San Francisco, CA"
+                },
+                "unit": {
+                    "type": "string",
+                    "enum": [
+                        "celsius",
+                        "fahrenheit"
+                    ]
+                }
+            },
+            "required": [
+                "location"
+            ]
+        }
+    },
+    "SEARCH": {
+        "name": "get_search_results",
+        "description": "Search Google to enhance knowledge.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "The prompt to search."
+                }
+            },
+            "required": [
+                "prompt"
+            ]
+        }
+    },
+    "URL": {
+        "name": "get_url_content",
+        "description": "Get the webpage content of a URL",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "url": {
+                    "type": "string",
+                    "description": "the URL to request"
+                }
+            },
+            "required": [
+                "url"
+            ]
+        }
+    },
+    "DATE": {
+        "name": "get_time",
+        "description": "Get the current time, date, and day of the week",
+    },
+    "VERSION": {
+        "name": "get_version_info",
+        "description": "Get version information",
+    },
+    "TARVEL": {
+        "name": "get_city_tarvel_info",
+        "description": "Get the city's travel plan by city name.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "city": {
+                    "type": "string",
+                    "description": "the city to search"
+                }
+            },
+            "required": [
+                "city"
+            ]
+        }
+    },
+    "IMAGE": {
+        "name": "generate_image",
+        "description": "Generate images based on user descriptions.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "the prompt to generate image"
+                }
+            },
+            "required": [
+                "prompt"
+            ]
+        }
+    },
+    "CODE": {
+        "name": "run_python_script",
+        "description": "Convert the string to a Python script and return the Python execution result. Assign the result to the variable result. The results must be printed to the console using the print function. Directly output the code, without using quotation marks or other symbols to enclose the code.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "the code to run"
+                }
+            },
+            "required": [
+                "prompt"
+            ]
+        }
+    },
+    "ARXIV": {
+        "name": "download_read_arxiv_pdf",
+        "description": "Get the content of the paper corresponding to the arXiv ID",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "prompt": {
+                    "type": "string",
+                    "description": "the arXiv ID of the paper"
+                }
+            },
+            "required": [
+                "prompt"
+            ]
+        }
+    },
+    "FLIGHT": {
+        "name": "get_Round_trip_flight_price",
+        "description": "Get round-trip ticket prices between two cities for the next six months. Use two city names as parameters. The name of the citys must be in Chinese.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "departcity": {
+                    "type": "string",
+                    "description": "the chinese name of departure city. e.g. 上海"
+                },
+                "arrivalcity": {
+                    "type": "string",
+                    "description": "the chinese name of arrival city. e.g. 北京"
+                }
+            },
+            "required": [
+                "departcity",
+                "arrivalcity"
+            ]
+        }
+    },
+}
+
+
+if __name__ == "__main__":
+    import json
+    tools_list = {"tools": [{"type": "function", "function": function_call_list[key]} for key in function_call_list.keys() if key != "base"]}
+    print(json.dumps(tools_list, indent=4, ensure_ascii=False))

beswarm/aient/test/claude.py

@@ -0,0 +1,32 @@
+# import os
+# import sys
+# print(os.path.dirname(os.path.abspath(__file__)))
+# sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+from .chatgpt import function_call_list
+def gpt2claude_tools_json(json_dict):
+    import copy
+    json_dict = copy.deepcopy(json_dict)
+    keys_to_change = {
+        "parameters": "input_schema",
+    }
+    for old_key, new_key in keys_to_change.items():
+        if old_key in json_dict:
+            if new_key:
+                json_dict[new_key] = json_dict.pop(old_key)
+            else:
+                json_dict.pop(old_key)
+        else:
+            if new_key and "description" in json_dict.keys():
+                json_dict[new_key] = {
+                    "type": "object",
+                    "properties": {}
+                }
+    if "tools" in json_dict.keys():
+        json_dict["tool_choice"] = {
+            "type": "auto"
+        }
+    return json_dict
+
+claude_tools_list = {f"{key}": gpt2claude_tools_json(function_call_list[key]) for key in function_call_list.keys()}
+if __name__ == "__main__":
+    print(claude_tools_list)

beswarm/aient/test/test.py

@@ -0,0 +1,2 @@
+a = "v1"
+print(a.split("v1"))

beswarm/aient/test/test_API.py

@@ -0,0 +1,6 @@
+def replace_with_asterisk(string, start=15, end=40):
+    return string[:start] + '*' * (end - start) + string[end:]
+
+original_string = "sk-zIuWeeuWY8vNCVhhHCXLroNmA6QhBxnv0ARMFcODVQwwqGRg"
+result = replace_with_asterisk(original_string)
+print(result)

beswarm/aient/test/test_Deepbricks.py

@@ -0,0 +1,20 @@
+import os
+from datetime import datetime
+
+from aient.models import chatgpt
+from aient.utils import prompt
+
+API = os.environ.get('API', None)
+API_URL = os.environ.get('API_URL', None)
+GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4o')
+LANGUAGE = os.environ.get('LANGUAGE', 'Simplified Chinese')
+
+current_date = datetime.now()
+Current_Date = current_date.strftime("%Y-%m-%d")
+
+systemprompt = os.environ.get('SYSTEMPROMPT', prompt.system_prompt.format(LANGUAGE, Current_Date))
+
+bot = chatgpt(api_key=API, api_url=API_URL, engine=GPT_ENGINE, system_prompt=systemprompt)
+# for text in bot.ask_stream("你好"):
+for text in bot.ask_stream("arXiv:2311.17132 讲了什么？"):
+    print(text, end="")

beswarm/aient/test/test_Web_crawler.py

@@ -0,0 +1,262 @@
+import re
+import os
+os.system('cls' if os.name == 'nt' else 'clear')
+import time
+import requests
+from bs4 import BeautifulSoup
+
+def Web_crawler(url: str, isSearch=False) -> str:
+    """返回链接网址url正文内容，必须是合法的网址"""
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+    }
+    result = ''
+    try:
+        requests.packages.urllib3.disable_warnings()
+        response = requests.get(url, headers=headers, verify=False, timeout=3, stream=True)
+        if response.status_code == 404:
+            print("Page not found:", url)
+            return ""
+            # return "抱歉，网页不存在，目前无法访问该网页。@Trash@"
+        content_length = int(response.headers.get('Content-Length', 0))
+        if content_length > 5000000:
+            print("Skipping large file:", url)
+            return result
+        try:
+            soup = BeautifulSoup(response.text.encode(response.encoding), 'xml', from_encoding='utf-8')
+        except:
+            soup = BeautifulSoup(response.text.encode(response.encoding), 'html.parser', from_encoding='utf-8')
+        # print("soup", soup)
+
+        for script in soup(["script", "style"]):
+            script.decompose()
+
+        table_contents = ""
+        tables = soup.find_all('table')
+        for table in tables:
+            table_contents += table.get_text()
+            table.decompose()
+
+        # body_text = "".join(soup.find('body').get_text().split('\n'))
+        body = soup.find('body')
+        if body:
+            body_text = body.get_text(separator=' ', strip=True)
+        else:
+            body_text = soup.get_text(separator=' ', strip=True)
+
+        result = table_contents + body_text
+        if result == '' and not isSearch:
+            result = ""
+            # result = "抱歉，可能反爬虫策略，目前无法访问该网页。@Trash@"
+        if result.count("\"") > 1000:
+            result = ""
+    except Exception as e:
+        print('\033[31m')
+        print("error: url", url)
+        print("error", e)
+        print('\033[0m')
+        result = "抱歉，目前无法访问该网页。"
+    # print("url content", result + "\n\n")
+    print(result)
+    return result
+
+import lxml.html
+from lxml.html.clean import Cleaner
+import httpx
+def get_body(url):
+    body = lxml.html.fromstring(httpx.get(url).text).xpath('//body')[0]
+    body = Cleaner(javascript=True, style=True).clean_html(body)
+    return ''.join(lxml.html.tostring(c, encoding='unicode') for c in body)
+
+import re
+import httpx
+import lxml.html
+from lxml.html.clean import Cleaner
+from html2text import HTML2Text
+from textwrap import dedent
+
+def url_to_markdown(url):
+    # 获取并清理网页内容
+    def get_body(url):
+        try:
+            text = httpx.get(url, verify=False, timeout=5).text
+            if text == "":
+                return "抱歉，目前无法访问该网页。"
+            # body = lxml.html.fromstring(text).xpath('//body')
+
+            doc = lxml.html.fromstring(text)
+            # 检查是否是GitHub raw文件格式（body > pre）
+            if doc.xpath('//body/pre'):
+                return text  # 直接返回原始文本，保留格式
+
+            body = doc.xpath('//body')
+            if body == [] and text != "":
+                body = text
+                return f'<pre>{body}</pre>'
+                # return body
+            else:
+                body = body[0]
+                body = Cleaner(javascript=True, style=True).clean_html(body)
+                return ''.join(lxml.html.tostring(c, encoding='unicode') for c in body)
+        except Exception as e:
+            print('\033[31m')
+            print("error: url", url)
+            print("error", e)
+            print('\033[0m')
+            return "抱歉，目前无法访问该网页。"
+
+    # 将HTML转换为Markdown
+    def get_md(cts):
+        h2t = HTML2Text(bodywidth=5000)
+        h2t.ignore_links = True
+        h2t.mark_code = True
+        h2t.ignore_images = True
+        res = h2t.handle(cts)
+
+        def _f(m):
+            return f'```\n{dedent(m.group(1))}\n```'
+
+        return re.sub(r'\[code]\s*\n(.*?)\n\[/code]', _f, res or '', flags=re.DOTALL).strip()
+
+    # 获取网页内容
+    body_content = get_body(url)
+
+    # 转换为Markdown
+    markdown_content = get_md(body_content)
+
+    return markdown_content
+
+def jina_ai_Web_crawler(url: str, isSearch=False) -> str:
+    """返回链接网址url正文内容，必须是合法的网址"""
+    headers = {
+        "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/58.0.3029.110 Safari/537.3"
+    }
+    result = ''
+    try:
+        requests.packages.urllib3.disable_warnings()
+        url = "https://r.jina.ai/" + url
+        response = requests.get(url, headers=headers, verify=False, timeout=5, stream=True)
+        if response.status_code == 404:
+            print("Page not found:", url)
+            return "抱歉，网页不存在，目前无法访问该网页。@Trash@"
+        content_length = int(response.headers.get('Content-Length', 0))
+        if content_length > 5000000:
+            print("Skipping large file:", url)
+            return result
+        soup = BeautifulSoup(response.text.encode(response.encoding), 'lxml', from_encoding='utf-8')
+        table_contents = ""
+        tables = soup.find_all('table')
+        for table in tables:
+            table_contents += table.get_text()
+            table.decompose()
+        body = "".join(soup.find('body').get_text().split('\n'))
+        result = table_contents + body
+        if result == '' and not isSearch:
+            result = "抱歉，可能反爬虫策略，目前无法访问该网页。@Trash@"
+        if result.count("\"") > 1000:
+            result = ""
+    except Exception as e:
+        print('\033[31m')
+        print("error: url", url)
+        print("error", e)
+        print('\033[0m')
+        result = "抱歉，目前无法访问该网页。"
+    print(result + "\n\n")
+    return result
+
+
+def get_url_content(url: str) -> str:
+    """
+    比较 url_to_markdown 和 jina_ai_Web_crawler 的结果，选择更好的内容
+
+    :param url: 要爬取的网页URL
+    :return: 选择的更好的内容
+    """
+    markdown_content = url_to_markdown(url)
+    print(markdown_content)
+    print('-----------------------------')
+    jina_content = jina_ai_Web_crawler(url)
+    print('-----------------------------')
+
+    # 定义评分函数
+    def score_content(content):
+        # 1. 内容长度
+        length_score = len(content)
+
+        # 2. 是否包含错误信息
+        error_penalty = 1000 if "抱歉" in content or "@Trash@" in content else 0
+
+        # 3. 内容的多样性（可以通过不同类型的字符来粗略估计）
+        diversity_score = len(set(content))
+
+        # 4. 特殊字符比例（过高可能意味着格式问题）
+        special_char_ratio = len(re.findall(r'[^a-zA-Z0-9\u4e00-\u9fff\s]', content)) / len(content)
+        special_char_penalty = 500 if special_char_ratio > 0.1 else 0
+
+        return length_score + diversity_score - error_penalty - special_char_penalty
+
+    if markdown_content == "":
+        markdown_score = -2000
+    else:
+        markdown_score = score_content(markdown_content)
+    if jina_content == "":
+        jina_score = -2000
+    else:
+        jina_score = score_content(jina_content)
+
+    print(f"url_to_markdown 得分： {markdown_score}")
+    print(f"jina_ai_Web_crawler 得分： {jina_score}")
+
+    if markdown_score > jina_score:
+        print("选择 url_to_markdown 的结果")
+        return markdown_content
+    elif markdown_score == jina_score and jina_score < 0:
+        print("两者都无法访问")
+        return ""
+    else:
+        print("选择 jina_ai_Web_crawler 的结果")
+        return jina_content
+
+start_time = time.time()
+# for url in ['https://www.zhihu.com/question/557257320', 'https://job.achi.idv.tw/2021/12/05/what-is-the-403-forbidden-error-how-to-fix-it-8-methods-explained/', 'https://www.lifewire.com/403-forbidden-error-explained-2617989']:
+# for url in ['https://developer.mozilla.org/en-US/docs/Web/HTTP/Status/403']:
+# for url in ['https://www.hostinger.com/tutorials/what-is-403-forbidden-error-and-how-to-fix-it']:
+# for url in ['https://beebom.com/what-is-403-forbidden-error-how-to-fix/']:
+# for url in ['https://www.lifewire.com/403-forbidden-error-explained-2617989']:
+# for url in ['https://www.usnews.com/news/best-countries/articles/2022-02-24/explainer-why-did-russia-invade-ukraine']:
+# for url in ['https://github.com/EAimTY/tuic']:
+# TODO 没办法访问
+# for url in ['https://s.weibo.com/top/summary?cate=realtimehot']:
+# for url in ['https://www.microsoft.com/en-us/security/blog/2023/05/24/volt-typhoon-targets-us-critical-infrastructure-with-living-off-the-land-techniques/']:
+# for url in ['https://tophub.today/n/KqndgxeLl9']:
+# for url in ['https://support.apple.com/zh-cn/HT213931']:
+# for url in ["https://zeta.zeabur.app"]:
+# for url in ["https://www.anthropic.com/research/probes-catch-sleeper-agents"]:
+# for url in ['https://finance.sina.com.cn/stock/roll/2023-06-26/doc-imyyrexk4053724.shtml']:
+# for url in ['https://s.weibo.com/top/summary?cate=realtimehot']:
+# for url in ['https://tophub.today/n/KqndgxeLl9', 'https://www.whatsonweibo.com/', 'https://www.trendingonweibo.com/?ref=producthunt', 'https://www.trendingonweibo.com/', 'https://www.statista.com/statistics/1377073/china-most-popular-news-on-weibo/']:
+# for url in ['https://www.usnews.com/news/entertainment/articles/2023-12-22/china-drafts-new-rules-proposing-restrictions-on-online-gaming']:
+# for url in ['https://developer.aliyun.com/article/721836']:
+# for url in ['https://cn.aliyun.com/page-source/price/detail/machinelearning_price']:
+# for url in ['https://mp.weixin.qq.com/s/Itad7Y-QBcr991JkF3SrIg']:
+# for url in ['https://zhidao.baidu.com/question/317577832.html']:
+# for url in ['https://www.cnn.com/2023/09/06/tech/huawei-mate-60-pro-phone/index.html']:
+# for url in ['https://www.reddit.com/r/China_irl/comments/15qojkh/46%E6%9C%88%E5%A4%96%E8%B5%84%E5%AF%B9%E4%B8%AD%E5%9B%BD%E7%9B%B4%E6%8E%A5%E6%8A%95%E8%B5%84%E5%87%8F87/', 'https://www.apple.com.cn/job-creation/Apple_China_CSR_Report_2020.pdf', 'https://hdr.undp.org/system/files/documents/hdr2013chpdf.pdf']:
+# for url in ['https://www.airuniversity.af.edu/JIPA/Display/Article/3111127/the-uschina-trade-war-vietnam-emerges-as-the-greatest-winner/']:
+# for url in ['https://zhuanlan.zhihu.com/p/646786536']:
+# for url in ['https://zh.wikipedia.org/wiki/%E4%BF%84%E7%BE%85%E6%96%AF%E5%85%A5%E4%BE%B5%E7%83%8F%E5%85%8B%E8%98%AD']:
+for url in ['https://raw.githubusercontent.com/yym68686/ChatGPT-Telegram-Bot/main/README.md']:
+# for url in ['https://raw.githubusercontent.com/openai/openai-python/main/src/openai/api_requestor.py']:
+# for url in ['https://stock.finance.sina.com.cn/usstock/quotes/aapl.html']:
+    # Web_crawler(url)
+    # print(get_body(url))
+    # print('-----------------------------')
+    # jina_ai_Web_crawler(url)
+    # print('-----------------------------')
+    # print(url_to_markdown(url))
+    # print('-----------------------------')
+    best_content = get_url_content(url)
+end_time = time.time()
+run_time = end_time - start_time
+# 打印运行时间
+print(f"程序运行时间：{run_time}秒")

beswarm/aient/test/test_aiwaves.py

@@ -0,0 +1,25 @@
+import os
+from aient.models import chatgpt
+
+API = os.environ.get('API', None)
+API_URL = os.environ.get('API_URL', None)
+GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4o')
+
+systemprompt = (
+    "你是一位旅行规划专家。你需要帮助用户规划旅行行程，给出合理的行程安排。"
+    "- 如果用户提及要从一个城市前往另外一个城市，必须使用 get_Round_trip_flight_price 查询两个城市半年内往返机票价格信息。"
+    "- 在规划行程之前，必须使用 get_city_tarvel_info 查询城市的景点旅行攻略信息。"
+    "- 查询攻略后，你需要分析用户个性化需求。充分考虑用户的年龄，情侣，家庭，朋友，儿童，独自旅行等情况。排除不适合用户个性化需求的景点。之后输出符合用户需求的景点。"
+    "- 综合用户游玩时间，适合用户个性化需求的旅游城市景点，机票信息和预算，给出真实准确的旅游行程，包括游玩时长、景点之间的交通方式和移动距离，每天都要给出总的游玩时间。"
+    "- 根据查到的景点介绍结合你自己的知识，每个景点必须包含你推荐的理由和景点介绍。介绍景点用户游玩的景点，景点介绍尽量丰富精彩，吸引用户眼球，不要直接复述查到的景点介绍。"
+    "- 每个景点都要标注游玩时间、景点之间的交通方式和移动距离还有生动的景点介绍"
+    "- 尽量排满用户的行程，不要有太多空闲时间。"
+)
+bot = chatgpt(api_key=API, api_url=API_URL, engine=GPT_ENGINE, system_prompt=systemprompt)
+for text in bot.ask_stream("我在上海想去重庆旅游，我只有2000元预算，我想在重庆玩一周，你能帮我规划一下吗？"):
+# for text in bot.ask_stream("我在广州市，想周一去香港，周四早上回来，是去游玩，请你帮我规划整个行程。包括细节，如交通，住宿，餐饮，价格，等等，最好细节到每天各个部分的时间，花费，等等，尽量具体，用户一看就能直接执行的那种"):
+# for text in bot.ask_stream("上海有哪些好玩的地方？"):
+# for text in bot.ask_stream("just say test"):
+# for text in bot.ask_stream("我在上海想去重庆旅游，我只有2000元预算，我想在重庆玩一周，你能帮我规划一下吗？"):
+# for text in bot.ask_stream("我在上海想去重庆旅游，我有一天的时间。你能帮我规划一下吗？"):
+    print(text, end="")

beswarm/aient/test/test_aiwaves_arxiv.py

@@ -0,0 +1,19 @@
+import os
+from datetime import datetime
+
+from aient.models import chatgpt
+from aient.utils import prompt
+
+API = os.environ.get('API', None)
+API_URL = os.environ.get('API_URL', None)
+GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4o')
+LANGUAGE = os.environ.get('LANGUAGE', 'Simplified Chinese')
+
+current_date = datetime.now()
+Current_Date = current_date.strftime("%Y-%m-%d")
+
+systemprompt = os.environ.get('SYSTEMPROMPT', prompt.system_prompt.format(LANGUAGE, Current_Date))
+
+bot = chatgpt(api_key=API, api_url=API_URL, engine=GPT_ENGINE, system_prompt=systemprompt)
+for text in bot.ask_stream("arXiv:2311.17132 讲了什么？"):
+    print(text, end="")

beswarm/aient/test/test_ask_gemini.py

@@ -0,0 +1,8 @@
+import os
+from aient.models import gemini
+
+GOOGLE_AI_API_KEY = os.environ.get('GOOGLE_AI_API_KEY', None)
+
+bot = gemini(api_key=GOOGLE_AI_API_KEY, engine='gemini-2.0-flash-exp')
+for text in bot.ask_stream("give me some example code of next.js to build a modern web site"):
+    print(text, end="")

beswarm/aient/test/test_class.py

@@ -0,0 +1,17 @@
+    # return e
+def j(e, f):
+    e(f)
+    # return e
+class a:
+    def __init__(self) -> None:
+        self.b = [1, 2, 3]
+    def d(self, e):
+        e.append(4)
+    def c(self):
+        j(self.d, self.b)
+        return self.b
+
+k = a()
+print(k.b)
+print(k.c())
+print(k.b)

beswarm/aient/test/test_claude.py

@@ -0,0 +1,23 @@
+import os
+from datetime import datetime
+
+from aient.models import chatgpt, claude3
+from aient.utils import prompt
+
+API = os.environ.get('API', None)
+CLAUDE_API = os.environ.get('claude_api_key', None)
+API_URL = os.environ.get('API_URL', None)
+GPT_ENGINE = os.environ.get('GPT_ENGINE', 'gpt-4o')
+LANGUAGE = os.environ.get('LANGUAGE', 'Simplified Chinese')
+
+current_date = datetime.now()
+Current_Date = current_date.strftime("%Y-%m-%d")
+
+systemprompt = os.environ.get('SYSTEMPROMPT', prompt.system_prompt.format(LANGUAGE, Current_Date))
+
+# bot = chatgpt(api_key=API, api_url=API_URL, engine=GPT_ENGINE, system_prompt=systemprompt)
+bot = claude3(api_key=CLAUDE_API, engine=GPT_ENGINE, system_prompt=systemprompt)
+for text in bot.ask_stream("arXiv:2210.10716 这篇文章讲了啥"):
+# for text in bot.ask_stream("今天的微博热搜有哪些？"):
+# for text in bot.ask_stream("你现在是什么版本？"):
+    print(text, end="")

beswarm/aient/test/test_claude_zh_char.py

@@ -0,0 +1,26 @@
+def is_surrounded_by_chinese(text, index):
+    left_char = text[index - 1]
+    if 0 < index < len(text) - 1:
+        right_char = text[index + 1]
+        return '\u4e00' <= left_char <= '\u9fff' or '\u4e00' <= right_char <= '\u9fff'
+    if index == len(text) - 1:
+        return '\u4e00' <= left_char <= '\u9fff'
+    return False
+
+def replace_char(string, index, new_char):
+    return string[:index] + new_char + string[index+1:]
+
+def claude_replace(text):
+    Punctuation_mapping = {",": "，", ":": "：", "!": "！", "?": "？", ";": "；"}
+    key_list = list(Punctuation_mapping.keys())
+    for i in range(len(text)):
+        if is_surrounded_by_chinese(text, i) and (text[i] in key_list):
+            text = replace_char(text, i, Punctuation_mapping[text[i]])
+    return text
+
+text = '''
+你好！我是一名人工智能助手，很高兴见到你。有什么我可以帮助你的吗？无论是日常问题还是专业领域，我都会尽我所能为你解答。让我们开始愉快的交流吧!'''
+
+if __name__ == '__main__':
+    new_text = claude_replace(text)
+    print(new_text)

beswarm/aient/test/test_ddg_search.py

@@ -0,0 +1,50 @@
+from itertools import islice
+from duckduckgo_search import DDGS
+
+# def getddgsearchurl(query, max_results=4):
+#     try:
+#         webresult = DDGS().text(query, max_results=max_results)
+#         if webresult == None:
+#             return []
+#         urls = [result['href'] for result in webresult]
+#     except Exception as e:
+#         print('\033[31m')
+#         print("duckduckgo error", e)
+#         print('\033[0m')
+#         urls = []
+#     # print("ddg urls", urls)
+#     return urls
+
+def getddgsearchurl(query, max_results=4):
+    try:
+        results = []
+        with DDGS() as ddgs:
+            ddgs_gen = ddgs.text(query, safesearch='Off', timelimit='y', backend="lite")
+            for r in islice(ddgs_gen, max_results):
+                results.append(r)
+        urls = [result['href'] for result in results]
+    except Exception as e:
+        print('\033[31m')
+        print("duckduckgo error", e)
+        print('\033[0m')
+        urls = []
+    return urls
+
+def search_answers(keywords, max_results=4):
+    results = []
+    with DDGS() as ddgs:
+        # 使用DuckDuckGo搜索关键词
+        ddgs_gen = ddgs.answers(keywords)
+        # 从搜索结果中获取最大结果数
+        for r in islice(ddgs_gen, max_results):
+            results.append(r)
+
+    # 返回一个json响应，包含搜索结果
+    return {'results': results}
+
+
+if __name__ == '__main__':
+    # 搜索关键词
+    query = "OpenAI"
+    print(getddgsearchurl(query))
+    # print(search_answers(query))