mirror of https://github.com/THUDM/CodeGeeX.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
# encoding:utf-8
|
|
|
|
import requests
|
|
import json
|
|
|
|
'''
|
|
Code Generation
|
|
'''
|
|
API_KEY = "" # Get from Tianqi console. 从控制台获取
|
|
API_SECRET = "" # Get from Tianqi console. 从控制台获取
|
|
PROMPT = "from typing import List\n\ndef has_close_elements(numbers: List[float], threshold: float) -> bool:\n \"\"\" Check if in given list of numbers, are any two numbers closer to each other than\n given threshold.\n >>> has_close_elements([1.0, 2.0, 3.0], 0.5)\n False\n >>> has_close_elements([1.0, 2.8, 3.0, 4.0, 5.0, 2.0], 0.3)\n True\n \"\"\"\n"
|
|
NUMBER = 3
|
|
LANG = "Python"
|
|
request_url = "https://tianqi.aminer.cn/api/v2/"
|
|
api = 'multilingual_code_generate'
|
|
|
|
# Request is in json format. 指定请求参数格式为json
|
|
headers = {'Content-Type': 'application/json'}
|
|
request_url = request_url + api
|
|
data = {
|
|
"apikey": API_KEY,
|
|
"apisecret": API_SECRET,
|
|
"prompt":PROMPT,
|
|
"n":NUMBER,
|
|
"lang":LANG
|
|
}
|
|
|
|
def main():
|
|
response = requests.post(request_url, headers=headers, data=json.dumps(data))
|
|
if response:
|
|
print(response.json())
|
|
|
|
if __name__ == '__main__':
|
|
main()
|