v0.2.1 files refactored

Author: Austin5925

.flake8

@@ -0,0 +1,3 @@
+[flake8]
+ignore = E203
+max-line-length = 100

.gitignore

@@ -128,6 +128,8 @@ ENV/
 env.bak/
 venv.bak/
 
+.idea/
+
 # Spyder project settings
 .spyderproject
 .spyproject

.idea/fastApiProject.iml

@@ -30,8 +30,8 @@ repos:
     hooks:
       - id: isort
         args: [ "--profile", "black" ]
-#  - repo: https://github.com/psf/black
-#    rev: 22.3.0
-#    hooks:
-#      - id: black
-#        args: [ --line-length=88 ]
+  - repo: https://github.com/psf/black
+    rev: 22.3.0
+    hooks:
+      - id: black
+        args: [ --line-length=100 ]

.pre-commit-config.yaml

@@ -0,0 +1,5 @@
+def draw_charts(sql_result: dict) -> None:
+    """
+    画图
+    """
+    pass

src/chartsInfo.py

@@ -0,0 +1,72 @@
+import json
+import os
+from typing import AsyncGenerator, List
+
+from dotenv import load_dotenv
+from httpx import AsyncClient
+
+load_dotenv()
+api_key = os.getenv("API_KEY")
+api_url = os.getenv("API_URL")
+
+
+GPT_MODEL = "gpt-4"
+GPT_TEMPERATURE = 0.3
+
+
+async def request(val: List[dict[str, str]]) -> AsyncGenerator[dict, None]:
+    print("request")
+    """
+    请求API
+    """
+    url = api_url
+    headers = {
+        "Content-Type": "application/json",
+        "Authorization": "Bearer " + api_key,
+    }
+    params = {
+        "model": GPT_MODEL,
+        "messages": val,
+        "max_tokens": 3000,
+        "temperature": GPT_TEMPERATURE,
+        "top_p": 1,
+        "n": 1,
+        "stream": True,
+    }
+    # if call_function:
+    #     params["function_call"] = "auto"
+    #     params["function"] = [{"name": "chat_stream",
+    #                            "description": "Generate the sql using
+    #                            the given question",
+    #                            "parameters": {
+    #                                    "type": "object",
+    #                                    "properties": {
+    #                                        "question": {
+    #                                            "type": "string",
+    #                                            "description":
+    #                                            "The paraphrased question"
+    #                                        },
+    #                                    },
+    #                                    "required": ["question"]
+    #                                },
+    #                            },
+    #                           ]
+    async with AsyncClient() as client:
+        async with client.stream("POST", url, headers=headers, json=params, timeout=60) as response:
+            async for line in response.aiter_lines():
+                if not line.strip():
+                    continue
+                line = line.replace("data: ", "")
+                try:
+                    data = json.loads(line)
+                except Exception:
+                    # finish_reason maybe function_call, but it doesn't matter
+                    data = {"choices": [{"finish_reason": "stop"}]}
+                if data.get("choices")[0].get("finish_reason") is not None:
+                    return
+                delta = data.get("choices")[0].get("delta")
+                # 下面就不用get delta字段了
+                if delta is None:  # 跳过空值
+                    continue
+                yield delta
+                # pprint.pprint(response.json())