mirror of
https://github.com/qodo-ai/pr-agent.git
synced 2025-07-04 21:00:40 +08:00
5dfd696c2b
```pipx run -e --spec . gen-api-key``` - Add shell script as an alternative to Python for API key generation
61 lines
2.4 KiB
Python
61 lines
2.4 KiB
Python
import requests
|
|
import json
|
|
import webbrowser
|
|
|
|
BASE_URL="https://api.cli.qodo.ai"
|
|
QM_CMD_ENDPOINT_KEY_NAME="qm_cmd_endpoint"
|
|
|
|
def get_api_key():
|
|
try:
|
|
url = f"{BASE_URL}/v1/auth/cli-auth?key_name={QM_CMD_ENDPOINT_KEY_NAME}"
|
|
# Make HTTP request
|
|
response = requests.get(url, stream=True, # For SSE (Server-Sent Events)
|
|
headers={'Accept': 'text/event-stream'})
|
|
|
|
# Parse Server-Sent Events format
|
|
# Look for line:
|
|
# data: {"auth_url":"https:\/\/auth.qodo.ai\/?extensionId=Codium.codium&extensionQuery=<SOME UUID>&uriScheme=cli"}
|
|
opened_auth_url = False
|
|
for line in response.iter_lines(decode_unicode=True):
|
|
if line and line.startswith('data:'):
|
|
data_json = line.replace('data: ', '', 1)
|
|
try:
|
|
data = json.loads(data_json)
|
|
if not opened_auth_url:
|
|
if 'auth_url' not in data:
|
|
print(f"Expected an auth_url, but couldn't find one.")
|
|
return None
|
|
print(f"Got url: {data['auth_url']}.")
|
|
# Unescape the URL
|
|
auth_url = data['auth_url'].replace('\\/', '/')
|
|
webbrowser.open(auth_url)
|
|
opened_auth_url = True
|
|
else:
|
|
api_key = data.get('api_key', None)
|
|
print(f"Got api key: {api_key}.")
|
|
if not api_key:
|
|
print(f"Expected an api_key, but couldn't find one.")
|
|
return api_key
|
|
except json.JSONDecodeError as e:
|
|
print(f"Error parsing JSON: {e}")
|
|
return None
|
|
|
|
print("No data line found in response")
|
|
return None
|
|
|
|
except requests.exceptions.RequestException as e:
|
|
print(f"Error making HTTP request: {e}")
|
|
return None
|
|
|
|
def run():
|
|
"""This is the function serving as this file's entry point, as defined in pyproject.toml"""
|
|
print("Attempting to generate an API Key...")
|
|
|
|
api_key = get_api_key()
|
|
if api_key:
|
|
print(f"Please use the following API Key: {api_key} when invoking QM endpoint.")
|
|
print(f"For example: curl -i --header \"Authorization: ApiKey {api_key}\" -X POST https://<QM ENDPOINT SERVER ADDR>/api/v1/qodo_merge_cmd")
|
|
|
|
if __name__ == "__main__":
|
|
run()
|