Spaces:
Runtime error
Runtime error
File size: 1,453 Bytes
c1f3831 e2937e5 c1f3831 63de821 c1f3831 63de821 db88ae6 bca1508 db88ae6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 |
import json
import requests
BASE_URL = "https://caffeinecrew-techdocs.hf.space"
def get_access_token(data, return_refresh_token=False):
url = BASE_URL + "/auth/login"
headers = {
"accept": "application/json",
}
data = json.dumps(data)
response = requests.post(url, data=data, headers=headers)
access_token = response.json()['access_token']
if return_refresh_token:
refresh_token = response.json()['refresh_token']
return access_token, refresh_token
return access_token
def request_inference(config, code_block, max_retries=1):
if max_retries == 0:
return None
url = BASE_URL+"/api/inference"
headers={"accept":"application/json", "Authorization": f"Bearer {config['access_token']}"}
code_input = code_block
response = requests.post(url=url, headers=headers, params={'code_block':code_input, 'api_key':config['api_key']})
if response.status_code == 200:
return response.json()["docstr"]
else:
data = {
"username":config['username'],
"password":config['password']
}
print("Encountered error retrying...")
config.update({"access_token":get_access_token(data)})
return request_inference(config, code_block, max_retries=max_retries-1)
def update_file(file_path, docstr_code):
with open(file_path, "w",errors='ignore') as file:
file.write(docstr_code)
|