mirror of https://github.com/xtekky/gpt4free.git
Merge pull request #1301 from xtekky/go
Add proxy support / fix streaming
This commit is contained in:
commit
89a56d0c34
|
@ -23,31 +23,29 @@ class Pi(BaseProvider):
|
|||
**kwargs
|
||||
) -> CreateResult:
|
||||
if not scraper:
|
||||
scraper = cls.get_scraper()
|
||||
scraper = cls.get_scraper(proxy)
|
||||
if not conversation:
|
||||
conversation = cls.start_conversation(scraper)
|
||||
answer = cls.ask(scraper, messages, conversation)
|
||||
|
||||
last_answer = 0
|
||||
for line in answer:
|
||||
if "text" in line:
|
||||
yield line["text"][last_answer:]
|
||||
last_answer = len(line["text"])
|
||||
yield line["text"]
|
||||
|
||||
def get_scraper():
|
||||
scraper = create_scraper(
|
||||
def get_scraper(proxy: str):
|
||||
return create_scraper(
|
||||
browser={
|
||||
'browser': 'chrome',
|
||||
'platform': 'windows',
|
||||
'desktop': True
|
||||
},
|
||||
sess=session()
|
||||
)
|
||||
scraper.headers = {
|
||||
headers={
|
||||
'Accept': '*/*',
|
||||
'Accept-Encoding': 'deflate,gzip,br',
|
||||
},
|
||||
proxies={
|
||||
"https": proxy
|
||||
}
|
||||
return scraper
|
||||
)
|
||||
|
||||
def start_conversation(scraper: CloudScraper):
|
||||
response = scraper.post('https://pi.ai/api/chat/start', data="{}", headers={
|
||||
|
|
Loading…
Reference in New Issue