ciyidogan commited on
Commit
3e9ad17
·
verified ·
1 Parent(s): 681e4ee

Update api_executor.py

Browse files
Files changed (1) hide show
  1. api_executor.py +5 -5
api_executor.py CHANGED
@@ -27,7 +27,7 @@ def _get_variable_value(session: Session, var_path: str) -> Any:
27
  if len(parts) >= 3:
28
  token_api = parts[1]
29
  token_field = parts[2]
30
- token_data = session.auth_tokens.get(token_api, {})
31
  return token_data.get(token_field)
32
  elif var_path.startswith("config."):
33
  attr_name = var_path.split(".", 1)[1]
@@ -136,7 +136,7 @@ def _fetch_token(api: APIConfig, session: Session) -> None:
136
  raise ValueError(f"Token path {api.auth.response_token_path} not found in response")
137
 
138
  # Store in session
139
- session.auth_tokens[api.name] = {
140
  "token": token,
141
  "expiry": time.time() + 3500, # ~1 hour
142
  "refresh_token": json_data.get("refresh_token")
@@ -153,7 +153,7 @@ def _refresh_token(api: APIConfig, session: Session) -> bool:
153
  if not api.auth or not api.auth.token_refresh_endpoint:
154
  return False
155
 
156
- token_info = session.auth_tokens.get(api.name, {})
157
  if not token_info.get("refresh_token"):
158
  return False
159
 
@@ -180,7 +180,7 @@ def _refresh_token(api: APIConfig, session: Session) -> bool:
180
  raise ValueError(f"Token path {api.auth.response_token_path} not found in refresh response")
181
 
182
  # Update session
183
- session.auth_tokens[api.name] = {
184
  "token": token,
185
  "expiry": time.time() + 3500,
186
  "refresh_token": json_data.get("refresh_token", token_info["refresh_token"])
@@ -198,7 +198,7 @@ def _ensure_token(api: APIConfig, session: Session) -> None:
198
  if not api.auth or not api.auth.enabled:
199
  return
200
 
201
- token_info = session.auth_tokens.get(api.name)
202
 
203
  # No token yet
204
  if not token_info:
 
27
  if len(parts) >= 3:
28
  token_api = parts[1]
29
  token_field = parts[2]
30
+ token_data = session._auth_tokens.get(token_api, {})
31
  return token_data.get(token_field)
32
  elif var_path.startswith("config."):
33
  attr_name = var_path.split(".", 1)[1]
 
136
  raise ValueError(f"Token path {api.auth.response_token_path} not found in response")
137
 
138
  # Store in session
139
+ session._auth_tokens[api.name] = {
140
  "token": token,
141
  "expiry": time.time() + 3500, # ~1 hour
142
  "refresh_token": json_data.get("refresh_token")
 
153
  if not api.auth or not api.auth.token_refresh_endpoint:
154
  return False
155
 
156
+ token_info = session._auth_tokens.get(api.name, {})
157
  if not token_info.get("refresh_token"):
158
  return False
159
 
 
180
  raise ValueError(f"Token path {api.auth.response_token_path} not found in refresh response")
181
 
182
  # Update session
183
+ session._auth_tokens[api.name] = {
184
  "token": token,
185
  "expiry": time.time() + 3500,
186
  "refresh_token": json_data.get("refresh_token", token_info["refresh_token"])
 
198
  if not api.auth or not api.auth.enabled:
199
  return
200
 
201
+ token_info = session._auth_tokens.get(api.name)
202
 
203
  # No token yet
204
  if not token_info: