diff options
author | t.me/xtekky <98614666+xtekky@users.noreply.github.com> | 2023-06-24 03:47:00 +0200 |
---|---|---|
committer | t.me/xtekky <98614666+xtekky@users.noreply.github.com> | 2023-06-24 03:47:00 +0200 |
commit | 5db58fd87f230fbe5bae599bb4b120ab42cad3be (patch) | |
tree | 770be13bca77c5d04dfe3265f378431df788706f /unfinished/bard/typings.py | |
parent | Merge pull request #664 from LopeKinz/main (diff) | |
download | gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.gz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.bz2 gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.lz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.xz gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.tar.zst gpt4free-5db58fd87f230fbe5bae599bb4b120ab42cad3be.zip |
Diffstat (limited to 'unfinished/bard/typings.py')
-rw-r--r-- | unfinished/bard/typings.py | 54 |
1 files changed, 0 insertions, 54 deletions
diff --git a/unfinished/bard/typings.py b/unfinished/bard/typings.py deleted file mode 100644 index 75b73bf9..00000000 --- a/unfinished/bard/typings.py +++ /dev/null @@ -1,54 +0,0 @@ -from typing import Dict, List, Union - - -class BardResponse: - def __init__(self, json_dict: Dict[str, Union[str, List]]) -> None: - """ - Initialize a BardResponse object. - - :param json_dict: A dictionary containing the JSON response data. - """ - self.json = json_dict - - self.content = json_dict.get('content') - self.conversation_id = json_dict.get('conversation_id') - self.response_id = json_dict.get('response_id') - self.factuality_queries = json_dict.get('factualityQueries', []) - self.text_query = json_dict.get('textQuery', []) - self.choices = [self.BardChoice(choice) - for choice in json_dict.get('choices', [])] - - def __repr__(self) -> str: - """ - Return a string representation of the BardResponse object. - - :return: A string representation of the BardResponse object. - """ - return f"BardResponse(conversation_id={self.conversation_id}, response_id={self.response_id}, content={self.content})" - - def filter_choices(self, keyword: str) -> List['BardChoice']: - """ - Filter the choices based on a keyword. - - :param keyword: The keyword to filter choices by. - :return: A list of filtered BardChoice objects. - """ - return [choice for choice in self.choices if keyword.lower() in choice.content.lower()] - - class BardChoice: - def __init__(self, choice_dict: Dict[str, str]) -> None: - """ - Initialize a BardChoice object. - - :param choice_dict: A dictionary containing the choice data. - """ - self.id = choice_dict.get('id') - self.content = choice_dict.get('content')[0] - - def __repr__(self) -> str: - """ - Return a string representation of the BardChoice object. - - :return: A string representation of the BardChoice object. - """ - return f"BardChoice(id={self.id}, content={self.content})" |