mirror of
https://github.com/Aider-AI/aider.git
synced 2025-06-12 23:54:59 +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):
|
def github_copilot_token_to_open_ai_key(self):
|
||||||
# check to see if there's an openai api key
|
# check to see if there's an openai api key
|
||||||
# If so, check to see if it's expire
|
# 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 (
|
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
|
import requests
|
||||||
|
|
||||||
headers = {
|
headers = {
|
||||||
'Authorization': f"Bearer {os.environ['GITHUB_COPILOT_TOKEN']}",
|
"Authorization": f"Bearer {os.environ['GITHUB_COPILOT_TOKEN']}",
|
||||||
'Editor-Version': self.extra_params['extra_headers']['Editor-Version'],
|
"Editor-Version": self.extra_params["extra_headers"]["Editor-Version"],
|
||||||
'Copilot-Integration-Id': self.extra_params['extra_headers']['Copilot-Integration-Id'],
|
"Copilot-Integration-Id": self.extra_params["extra_headers"][
|
||||||
'Content-Type': 'application/json',
|
"Copilot-Integration-Id"
|
||||||
|
],
|
||||||
|
"Content-Type": "application/json",
|
||||||
}
|
}
|
||||||
res = requests.get("https://api.github.com/copilot_internal/v2/token", headers=headers)
|
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):
|
def send_completion(self, messages, functions, stream, temperature=None):
|
||||||
if os.environ.get("AIDER_SANITY_CHECK_TURNS"):
|
if os.environ.get("AIDER_SANITY_CHECK_TURNS"):
|
||||||
|
@ -935,7 +939,7 @@ class Model(ModelSettings):
|
||||||
kwargs["messages"] = messages
|
kwargs["messages"] = messages
|
||||||
|
|
||||||
# Are we using github copilot?
|
# 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()
|
self.github_copilot_token_to_open_ai_key()
|
||||||
|
|
||||||
res = litellm.completion(**kwargs)
|
res = litellm.completion(**kwargs)
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue