mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-11 23:25:01 +00:00
lint
This commit is contained in:
parent
ef2986a231
commit
8304029b92
1 changed files with 12 additions and 8 deletions
|
@ -878,20 +878,24 @@ class Model(ModelSettings):
|
|||
def github_copilot_token_to_open_ai_key(self):
|
||||
# check to see if there's an openai api key
|
||||
# If so, check to see if it's expire
|
||||
openai_api_key = 'OPENAI_API_KEY'
|
||||
openai_api_key = "OPENAI_API_KEY"
|
||||
|
||||
if openai_api_key not in os.environ or (
|
||||
int(dict(x.split("=") for x in os.environ[openai_api_key].split(";"))['exp']) < int(datetime.now().timestamp())
|
||||
int(dict(x.split("=") for x in os.environ[openai_api_key].split(";"))["exp"])
|
||||
< int(datetime.now().timestamp())
|
||||
):
|
||||
import requests
|
||||
|
||||
headers = {
|
||||
'Authorization': f"Bearer {os.environ['GITHUB_COPILOT_TOKEN']}",
|
||||
'Editor-Version': self.extra_params['extra_headers']['Editor-Version'],
|
||||
'Copilot-Integration-Id': self.extra_params['extra_headers']['Copilot-Integration-Id'],
|
||||
'Content-Type': 'application/json',
|
||||
"Authorization": f"Bearer {os.environ['GITHUB_COPILOT_TOKEN']}",
|
||||
"Editor-Version": self.extra_params["extra_headers"]["Editor-Version"],
|
||||
"Copilot-Integration-Id": self.extra_params["extra_headers"][
|
||||
"Copilot-Integration-Id"
|
||||
],
|
||||
"Content-Type": "application/json",
|
||||
}
|
||||
res = requests.get("https://api.github.com/copilot_internal/v2/token", headers=headers)
|
||||
os.environ[openai_api_key] = res.json()['token']
|
||||
os.environ[openai_api_key] = res.json()["token"]
|
||||
|
||||
def send_completion(self, messages, functions, stream, temperature=None):
|
||||
if os.environ.get("AIDER_SANITY_CHECK_TURNS"):
|
||||
|
@ -935,7 +939,7 @@ class Model(ModelSettings):
|
|||
kwargs["messages"] = messages
|
||||
|
||||
# Are we using github copilot?
|
||||
if 'GITHUB_COPILOT_TOKEN' in os.environ:
|
||||
if "GITHUB_COPILOT_TOKEN" in os.environ:
|
||||
self.github_copilot_token_to_open_ai_key()
|
||||
|
||||
res = litellm.completion(**kwargs)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue