Spaces:
Paused
Paused
Update api_connector.py
Browse files- api_connector.py +6 -6
api_connector.py
CHANGED
|
@@ -1,5 +1,5 @@
|
|
| 1 |
import requests
|
| 2 |
-
from
|
| 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 |
-
|
| 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 |
-
|
| 87 |
self.refresh_auth_token(api_name, auth_config, session)
|
| 88 |
continue
|
| 89 |
response.raise_for_status()
|
| 90 |
-
|
| 91 |
return response.json()
|
| 92 |
except requests.Timeout:
|
| 93 |
fallback = intent_def.get("fallback_timeout_message", "This operation is currently unavailable.")
|
| 94 |
-
|
| 95 |
return {"fallback": fallback}
|
| 96 |
except Exception as e:
|
| 97 |
-
|
| 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}
|