diff options
| author | Ryan Jordan <ryjordan@gmail.com> | 2023-09-06 02:39:57 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-09-06 02:39:57 +0200 |
| commit | f81e618958318a092ca4c70a1b3ea15260bda97c (patch) | |
| tree | 3ce022a8d719011268da7f1a0befc97bf32a60d8 /g4f/Provider/AItianhu.py | |
| parent | feat(docker): add Docker and Docker Compose support (diff) | |
| parent | ~ | Merge pull request #869 from ahobsonsayers/add-console-script (diff) | |
| download | gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.gz gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.bz2 gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.lz gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.xz gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.tar.zst gpt4free-f81e618958318a092ca4c70a1b3ea15260bda97c.zip | |
Diffstat (limited to 'g4f/Provider/AItianhu.py')
| -rw-r--r-- | g4f/Provider/AItianhu.py | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/g4f/Provider/AItianhu.py b/g4f/Provider/AItianhu.py index e8e5714a..0982d3c6 100644 --- a/g4f/Provider/AItianhu.py +++ b/g4f/Provider/AItianhu.py @@ -1,3 +1,5 @@ +from __future__ import annotations + import json import requests @@ -7,7 +9,7 @@ from .base_provider import BaseProvider class AItianhu(BaseProvider): - url = "https://www.aitianhu.com/api/chat-process" + url = "https://www.aitianhu.com/" working = False supports_gpt_35_turbo = True @@ -15,13 +17,10 @@ class AItianhu(BaseProvider): def create_completion( model: str, messages: list[dict[str, str]], - stream: bool, - **kwargs: Any, - ) -> CreateResult: - base = "" - for message in messages: - base += "%s: %s\n" % (message["role"], message["content"]) - base += "assistant:" + stream: bool, **kwargs: Any) -> CreateResult: + + base = "\n".join(f"{message['role']}: {message['content']}" for message in messages) + base += "\nassistant: " headers = { "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/114.0.0.0 Safari/537.36" |