1
0
Fork 0

added error checking to the remaining methods

all the methods within Phind should now raise a value error if
cf_clearance is missing
This commit is contained in:
Johannes 2023-04-26 08:12:10 +02:00
parent 89cbaf7b61
commit 592bbaab86

View file

@ -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,9 +108,12 @@ 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)