api_key_auth_factory.py 495 B

123456789101112131415
  1. from services.auth.firecrawl import FirecrawlAuth
  2. from services.auth.jina import JinaAuth
  3. class ApiKeyAuthFactory:
  4. def __init__(self, provider: str, credentials: dict):
  5. if provider == "firecrawl":
  6. self.auth = FirecrawlAuth(credentials)
  7. elif provider == "jinareader":
  8. self.auth = JinaAuth(credentials)
  9. else:
  10. raise ValueError("Invalid provider")
  11. def validate_credentials(self):
  12. return self.auth.validate_credentials()