Skip to content

YEP-2933 Implement cached access token endpoint for yepcode run #5

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sergio-rs merged 1 commit into from
Jul 7, 2025
Merged

YEP-2933 Implement cached access token endpoint for yepcode run #5

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 13 additions & 10 deletions yepcode_run/api/yepcode_api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -65,10 +65,6 @@ def __init__(self, config: YepCodeApiConfig = None):
"timeout": 60000,
**config_dict,
}
if not final_config.get("auth_url"):
final_config["auth_url"] = (
f"{final_config['api_host']}/auth/realms/yepcode/protocol/openid-connect/token"
)

if (
not final_config.get("access_token")
Expand Down Expand Up @@ -110,6 +106,7 @@ def __init__(self, config: YepCodeApiConfig = None):
self.api_host = final_config.get("api_host")
self.client_id = final_config.get("client_id")
self.client_secret = final_config.get("client_secret")
self.api_token = final_config.get("api_token")
self.auth_url = final_config.get("auth_url")
self.team_id = final_config.get("team_id")
self.access_token = final_config.get("access_token")
Expand All @@ -118,6 +115,8 @@ def __init__(self, config: YepCodeApiConfig = None):
self.client_id = self._client_id_from_access_token()
if not self.team_id and self.client_id:
self.team_id = self._team_id_from_client_id()
if not self.auth_url:
self.auth_url = self._get_auth_url()

def get_client_id(self) -> str:
if not self.client_id:
Expand Down Expand Up @@ -154,23 +153,27 @@ def _team_id_from_client_id(self) -> str:
def _get_base_url(self) -> str:
return f"{self.api_host}/api/{self.team_id}/rest"

def _get_auth_url(self) -> str:
return f"{self._get_base_url()}/auth/token"

def _get_access_token(self) -> str:
if not self.client_id or not self.client_secret:
if not self.api_token and (not self.client_id or not self.client_secret):
raise ValueError(
"AccessToken has expired. Provide a new one or enable automatic refreshing by providing an apiToken or clientId and clientSecret."
)
try:
auth_str = base64.b64encode(
api_token = (
self.api_token
or f"sk-{base64.b64encode(
f"{self.client_id}:{self.client_secret}".encode()
).decode()
).decode()}"
)

response = requests.post(
self.auth_url,
headers={
"Authorization": f"Basic {auth_str}",
"Content-Type": "application/x-www-form-urlencoded",
"x-api-token": api_token,
},
data="grant_type=client_credentials",
timeout=self.timeout,
)

Expand Down