summaryrefslogtreecommitdiffstats
path: root/experimental/bard/typings.py
diff options
context:
space:
mode:
Diffstat (limited to 'experimental/bard/typings.py')
-rw-r--r--experimental/bard/typings.py15
1 files changed, 15 insertions, 0 deletions
diff --git a/experimental/bard/typings.py b/experimental/bard/typings.py
new file mode 100644
index 00000000..69239762
--- /dev/null
+++ b/experimental/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]