From b206a1eb6364e519741ceb89e3f3b63910b4e1ef Mon Sep 17 00:00:00 2001
From: Raju Komati <komatiraju032@gmail.com>
Date: Fri, 28 Apr 2023 02:27:06 +0530
Subject: update openai_rev module add test file  with simple use

---
 openai_rev/forefront/__init__.py | 18 ++++++++++++++----
 1 file changed, 14 insertions(+), 4 deletions(-)

(limited to 'openai_rev/forefront/__init__.py')

diff --git a/openai_rev/forefront/__init__.py b/openai_rev/forefront/__init__.py
index bef10e9e..10202615 100644
--- a/openai_rev/forefront/__init__.py
+++ b/openai_rev/forefront/__init__.py
@@ -1,9 +1,9 @@
 from json import loads
 from re import match
 from time import time, sleep
+from typing import Generator, Optional
 from uuid import uuid4
 
-from altair.vegalite.v3 import Generator
 from fake_useragent import UserAgent
 from requests import post
 from tls_client import Session
@@ -14,7 +14,7 @@ from .models import ForeFrontResponse
 
 class Account:
     @staticmethod
-    def create(proxy=None, logging=False):
+    def create(proxy: Optional[str] = None, logging: bool = False):
         proxies = {'http': 'http://' + proxy, 'https': 'http://' + proxy} if proxy else False
 
         start = time()
@@ -57,7 +57,8 @@ class Account:
         while True:
             sleep(1)
             for _ in mail_client.fetch_inbox():
-                print(mail_client.get_message_content(_['id']))
+                if logging:
+                    print(mail_client.get_message_content(_['id']))
                 mail_token = match(r'(\d){5,6}', mail_client.get_message_content(_['id'])).group(0)
 
             if mail_token:
@@ -166,7 +167,9 @@ class Completion:
         default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0',  # default
         model='gpt-4',
     ) -> ForeFrontResponse:
+        text = ''
         final_response = None
+        res = list(StreamingCompletion.create(token=token, prompt=prompt))
         for response in StreamingCompletion.create(
             token=token,
             chat_id=chat_id,
@@ -175,6 +178,13 @@ class Completion:
             default_persona=default_persona,
             model=model,
         ):
-            final_response = response
+            if response:
+                final_response = response
+                text += response.text
+
+        if final_response:
+            final_response.text = text
+        else:
+            raise Exception('Unable to get the response, Please try again')
 
         return final_response
-- 
cgit v1.2.3