初始化

function_generator.py

@@ -0,0 +1,236 @@
+import datetime
+import json
+import os
+import subprocess
+
+import bs4
+import requests
+
+wolfram_api_key = None
+always_authenticate = False
+
+def gen_functions(functions_def: list) -> list:
+    function_params_list = []
+    for user_function in functions_def:
+        try:
+            function = {
+                "name": user_function["name"],
+                "description": "\n".join([line.strip() for line in user_function["description"].strip().split("\n")]),
+                "parameters": {
+                    "type": "object",
+                    "properties": user_function["params"]
+                },
+                "requires": list(user_function["params"].keys())
+            }
+            function_params_list.append(function)
+        except KeyError:
+            print(f"函数 {user_function.get('name', '未知')} 无法识别")
+    return function_params_list
+
+def get_time(args: dict=None) -> dict:
+    if args is None:
+        args = {}
+    return json.dumps({"time": str(datetime.datetime.now())})
+
+def run_cmd(args: dict=None) -> dict:
+    if args is None:
+        args = {}
+    cmd = args.get("cmd", "")
+    print(f"请求执行指令：{cmd}, 你是否授权这一行为? [Y/n/用户反馈] ", end="")
+    authorization = "y" if always_authenticate else input("")
+    if authorization.lower() == "n":
+        return json.dumps({"result": "User rejected the request"})
+    elif authorization.lower() == "y" or authorization == "":
+        try:
+            result = subprocess.run(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True, encoding='gbk')
+            return json.dumps(
+                {
+                    "result": f"成功运行了指令：{cmd} {result.stdout.strip()}"
+                    if result.returncode == 0 and result.stdout is not None
+                    else f"运行指令时发生了错误：{cmd} {result.stderr.strip()}"
+                }
+            )
+        except Exception as e:
+            return json.dumps({"result": str(e)})
+    else:
+        return json.dumps({"result": f"User rejected the request with a feedback: {authorization}"})
+
+def wolframalpha(args: dict=None) -> dict:
+    if args is None:
+        args = {}
+    query = args.get("query", "")
+    full_results = args.get("full_results", False)
+    base_url = "http://api.wolframalpha.com/v2/query" if full_results else "http://api.wolframalpha.com/v1/result"
+    if query == "":
+        return json.dumps({"error": "query is empty"})
+    if wolfram_api_key is None:
+        return json.dumps({"error": "wolfram_api_key is empty"})
+
+    params = {
+        "input": query,
+        "format": "plaintext",
+        "appid": wolfram_api_key
+    }
+    if full_results:
+        params["output"] = "JSON"
+
+    try:
+        response = requests.get(base_url, params=params)
+        response.raise_for_status()
+        result = response.json() if full_results else {"result": response.text}
+        return json.dumps(result)
+    except requests.exceptions.RequestException as e:
+        return json.dumps({"error": f"Request failed: {e}"})
+
+def spider(args: dict=None) -> dict:
+    if args is None:
+        args = {}
+    url = args.get("url", "")
+    print(f"尝试爬取内容: {url}")
+    response = requests.get(url)
+    soup = bs4.BeautifulSoup(response.text, "html.parser")
+    text = ' '.join(soup.stripped_strings)
+    return json.dumps({"result": text})
+
+def get_absolute_path(path):
+    return path if os.path.isabs(path) else os.path.abspath(path)
+
+def write(args: dict=None) -> dict:
+    if args is None:
+        args = {}
+    result = {'result': ""}
+    try:
+        path = args['path']
+        content = args['content']
+        encoding = args.get('encoding', 'utf-8')
+        if path == "":
+            path = "./"
+        path = get_absolute_path(path)
+        print(f"请求保存文件到：{path}, 你是否授权这一行为? [Y/n/用户反馈] ", end="")
+        authorization = "y" if always_authenticate else input("")
+        if authorization.lower() == "n":
+            result['result'] = "User rejected the request"
+            return json.dumps(result)
+        elif authorization.lower() == "y" or authorization == "":
+            os.makedirs(os.path.dirname(path), exist_ok=True)
+            # file deepcode ignore BinaryWrite: encoding be decided by AI
+            with open(path, 'w', encoding=encoding) as f:
+                f.write(content)
+            result['result'] = f"File saved successfully at {path}"
+        else:
+            result['result'] = f"User rejected the request with a feedback: {authorization}"
+            return json.dumps(result)
+    except Exception as e:
+        result['result'] = f"An error occurred while saving the file: {str(e)}"
+    return json.dumps(result)
+
+def read(args):
+    result = {'success': False, 'content': None, 'error': None}
+    try:
+        path = args['path']
+        encoding = args['encoding'] if 'encoding' in args else 'utf-8'
+        print(f"请求读取文件：{path}, 你是否授权这一行为? [Y/n/用户反馈] ", end="")
+        authorization = "y" if always_authenticate else input("")
+        if authorization.lower() == "n":
+            result['error'] = "User rejected the request"
+            return json.dumps(result)
+        elif authorization.lower() == "y" or authorization == "":
+            with open(path, 'r', encoding=encoding) as f:
+                content = f.read()
+            result['success'] = True
+            result['content'] = content
+        else:
+            result['error'] = f"User rejected the request with a feedback: {authorization}"
+            return json.dumps(result)
+    except Exception as e:
+        result['error'] = str(e)
+    return json.dumps(result)
+
+functions_def = [
+    {
+        "name": "get_time",
+        "user_friendly_name": "获取时间",
+        "description": "Get the current date time in YYYY-MM-DD HH:mm:SS.millisecond",
+        "params": {},
+        "function": get_time
+    },
+    {
+        "name": "run_cmd",
+        "user_friendly_name": "运行指令",
+        "description": """run a Windows command and get command output
+                        Your command will be be running in a safe environment.
+                        For file paths that has spaces, use double quotes to wrap the path.""",
+        "params": {
+            "cmd": {
+                "type": "string",
+                "description": "The command you would like to execute"
+            }
+        },
+        "function": run_cmd
+    },
+    {
+        "name": "wolframalpha",
+        "user_friendly_name": "Wolfram Alpha",
+        "description": """to get math answers using natural language
+                        If the function fails, notify the user and try solve the problem""",
+        "params": {
+            "query": {
+                "type": "string",
+                "description": "describe your math question in natural language, you MUST use english language."
+            },
+            "full_results": {
+                "type": "boolean",
+                "description": "set to true will return the answer in full, vice versa."
+            }
+        },
+        "function": wolframalpha
+    },
+    {
+        "name": "spider",
+        "user_friendly_name": "爬虫",
+        "description": "for getting text from a dedicated url",
+        "params": {
+            "url": {
+                "type": "string",
+                "description": "your dedicated url"
+            }
+        },
+        "function": spider
+    },
+    {
+        "name": "read",
+        "user_friendly_name": "读取文件",
+        "description": "read local file from a given path and encoding",
+        "params": {
+            "path": {
+                "type": "string",
+                "description": "Path to the file you would like to read"
+            },
+            "encoding":{
+                "type": "string",
+                "description": "The Encoding of the file you would like to read"
+            }
+        },
+        "function": read
+    },
+    {
+        "name": "write",
+        "user_friendly_name": "写入文件",
+        "description": "write data to local file from a given path and encoding",
+        "params": {
+            "path": {
+                "type": "string",
+                "description": "Path to the file you would like to write"
+            },
+            "encoding":{
+                "type": "string",
+                "description": "The Encoding of the file you would like to write"
+            },
+            "content":{
+                "type": "string",
+                "description": "Content you would like to write to the file"
+            }
+        },
+        "function": write
+    }
+]