From a485cfb180094f14b3fe27ea218d2890033768a3 Mon Sep 17 00:00:00 2001
From: "t.me/xtekky" <98614666+xtekky@users.noreply.github.com>
Date: Thu, 20 Apr 2023 15:34:19 +0100
Subject: unfinished api's (help is welcome)

---
 unfinished/bard/typings.py | 15 +++++++++++++++
 1 file changed, 15 insertions(+)
 create mode 100644 unfinished/bard/typings.py

(limited to 'unfinished/bard/typings.py')

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]
-- 
cgit v1.2.3