feat: add gemini api parse method

dsdanielpark · Mar 20, 2024 · bf1d558 · bf1d558
1 parent 1abaf33
commit bf1d558
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/bardapi/core.py b/bardapi/core.py
@@ -7,7 +7,9 @@
 import string
 import uuid
 import requests
+import gemini
 from typing import Optional
+from gemini.src.model.parser.response_parser import ResponseParser
 
 # from urllib.parse import parse_qs, urlparse
 try:
@@ -281,6 +283,13 @@ def get_answer(
             proxies=self.proxies,
         )
 
+        try:
+            parser = ResponseParser(cookies=self.cookies)
+            return  parser.parse(resp)
+        except:
+            pass
+
+
         # Post-processing of response
         resp_dict = json.loads(resp.content.splitlines()[-5])[0][2]