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:
parent
89cbaf7b61
commit
592bbaab86
1 changed files with 10 additions and 3 deletions
|
@ -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)
|
||||
|
|
Loading…
Reference in a new issue