ciyidogan commited on
Commit
d7e064f
Β·
verified Β·
1 Parent(s): 315cdec

Update api_connector.py

Browse files
Files changed (1) hide show
  1. api_connector.py +6 -6
api_connector.py CHANGED
@@ -1,5 +1,5 @@
1
  import requests
2
- from utils import log
3
 
4
  class APIConnector:
5
  def __init__(self, service_config):
@@ -58,7 +58,7 @@ class APIConnector:
58
  new_refresh_token = json_resp.get("refresh_token", session.auth_tokens[api_name].get("refresh_token"))
59
  session.auth_tokens[api_name] = {"token": token, "refresh_token": new_refresh_token}
60
 
61
- log(f"πŸ” Refreshed auth token for {api_name}")
62
  return token
63
 
64
  def call_api(self, intent_def, session):
@@ -83,17 +83,17 @@ class APIConnector:
83
  try:
84
  response = requests.request(method, url, headers=headers, json=body, timeout=timeout)
85
  if response.status_code == 401 and auth_config and attempt < retry_count:
86
- log(f"πŸ” Token expired for {api_name}, refreshing...")
87
  self.refresh_auth_token(api_name, auth_config, session)
88
  continue
89
  response.raise_for_status()
90
- log(f"βœ… API call successful: {api_name}")
91
  return response.json()
92
  except requests.Timeout:
93
  fallback = intent_def.get("fallback_timeout_message", "This operation is currently unavailable.")
94
- log(f"⚠️ API timeout for {api_name} β†’ {fallback}")
95
  return {"fallback": fallback}
96
  except Exception as e:
97
- log(f"❌ API call error for {api_name}: {e}")
98
  fallback = intent_def.get("fallback_error_message", "An error occurred during the operation.")
99
  return {"fallback": fallback}
 
1
  import requests
2
+ from logger import log_info, log_error, log_warning, log_debug
3
 
4
  class APIConnector:
5
  def __init__(self, service_config):
 
58
  new_refresh_token = json_resp.get("refresh_token", session.auth_tokens[api_name].get("refresh_token"))
59
  session.auth_tokens[api_name] = {"token": token, "refresh_token": new_refresh_token}
60
 
61
+ log_info(f"πŸ” Refreshed auth token for {api_name}")
62
  return token
63
 
64
  def call_api(self, intent_def, session):
 
83
  try:
84
  response = requests.request(method, url, headers=headers, json=body, timeout=timeout)
85
  if response.status_code == 401 and auth_config and attempt < retry_count:
86
+ log_info(f"πŸ” Token expired for {api_name}, refreshing...")
87
  self.refresh_auth_token(api_name, auth_config, session)
88
  continue
89
  response.raise_for_status()
90
+ log_info(f"βœ… API call successful: {api_name}")
91
  return response.json()
92
  except requests.Timeout:
93
  fallback = intent_def.get("fallback_timeout_message", "This operation is currently unavailable.")
94
+ log_warning(f"⚠️ API timeout for {api_name} β†’ {fallback}")
95
  return {"fallback": fallback}
96
  except Exception as e:
97
+ log_error(f"❌ API call error for {api_name}", e)
98
  fallback = intent_def.get("fallback_error_message", "An error occurred during the operation.")
99
  return {"fallback": fallback}