zach commited on
Commit
c0b34a2
·
1 Parent(s): a5cafbd

Fix tenacity imports in anthropic_api.py and hume_api.py

Browse files
src/integrations/anthropic_api.py CHANGED
@@ -25,7 +25,7 @@ from typing import Union, List
25
  # Third-Party Library Imports
26
  from anthropic import Anthropic
27
  from anthropic.types import Message, ModelParam, TextBlock
28
- from tenacity import retry, stop_after_attempt, wait_fixed
29
  # Local Application Imports
30
  from src.config import logger
31
  from src.utils import truncate_text, validate_env_var
 
25
  # Third-Party Library Imports
26
  from anthropic import Anthropic
27
  from anthropic.types import Message, ModelParam, TextBlock
28
+ from tenacity import retry, stop_after_attempt, wait_fixed, before_log, after_log
29
  # Local Application Imports
30
  from src.config import logger
31
  from src.utils import truncate_text, validate_env_var
src/integrations/hume_api.py CHANGED
@@ -24,7 +24,7 @@ from dataclasses import dataclass
24
  from typing import Optional
25
  # Third-Party Library Imports
26
  import requests
27
- from tenacity import retry, stop_after_attempt, wait_fixed
28
  # Local Application Imports
29
  from src.config import logger
30
  from src.utils import validate_env_var, truncate_text
 
24
  from typing import Optional
25
  # Third-Party Library Imports
26
  import requests
27
+ from tenacity import retry, stop_after_attempt, wait_fixed, before_log, after_log
28
  # Local Application Imports
29
  from src.config import logger
30
  from src.utils import validate_env_var, truncate_text