diff options
Diffstat (limited to 'unfinished/bard/typings.py')
-rw-r--r-- | unfinished/bard/typings.py | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/unfinished/bard/typings.py b/unfinished/bard/typings.py new file mode 100644 index 00000000..69239762 --- /dev/null +++ b/unfinished/bard/typings.py @@ -0,0 +1,15 @@ +class BardResponse: + def __init__(self, json_dict): + 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', [])] + + class BardChoice: + def __init__(self, choice_dict): + self.id = choice_dict.get('id') + self.content = choice_dict.get('content')[0] |