Spaces:
Building
Building
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}
|