From 592bbaab8696785b4f58094becd83b716b09fea5 Mon Sep 17 00:00:00 2001 From: Johannes Date: Wed, 26 Apr 2023 08:12:10 +0200 Subject: added error checking to the remaining methods all the methods within Phind should now raise a value error if cf_clearance is missing --- phind/__init__.py | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/phind/__init__.py b/phind/__init__.py index bd54d82a..9f000c6f 100644 --- a/phind/__init__.py +++ b/phind/__init__.py @@ -53,8 +53,10 @@ class PhindResponse: class Search: def create(prompt: str, actualSearch: bool = True, language: str = 'en') -> dict: # None = no search - if user_agent == '' or cf_clearance == '': - raise ValueError('user_agent and cf_clearance must be set, refer to documentation') + if user_agent == '': + raise ValueError('user_agent must be set, refer to documentation') + if cf_clearance == '' : + raise ValueError('cf_clearance must be set, refer to documentation') if not actualSearch: return { @@ -106,8 +108,11 @@ class Completion: codeContext: str = '', language: str = 'en') -> PhindResponse: - if user_agent == '': + if user_agent == '' : raise ValueError('user_agent must be set, refer to documentation') + + if cf_clearance == '' : + raise ValueError('cf_clearance must be set, refer to documentation') if results is None: results = Search.create(prompt, actualSearch = True) @@ -235,6 +240,8 @@ class StreamingCompletion: if user_agent == '': raise ValueError('user_agent must be set, refer to documentation') + if cf_clearance == '' : + raise ValueError('cf_clearance must be set, refer to documentation') if results is None: results = Search.create(prompt, actualSearch = True) -- cgit v1.2.3