返回提交历史
Modified
g4f/Provider/hf_space/BAAI_Ling.py
+6
-5
XFEstudio/gpt4free
Fix code review issues in BAAI_Ling provider
Co-authored-by: hlohaus <983577+hlohaus@users.noreply.github.com>
04e300d7
代码差异
1 个文件
+6
-5
@@ -5,7 +5,7 @@ import json
5
5
import uuid
6
6
7
7
from ...typing import AsyncResult, Messages
8
from ...providers.response import JsonConversation, Reasoning
8
from ...providers.response import JsonConversation
9
9
from ...requests.raise_for_status import raise_for_status
10
10
from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
11
11
from ..helper import format_prompt, get_last_user_message
@@ -22,11 +22,10 @@ class BAAI_Ling(AsyncGeneratorProvider, ProviderModelMixin):
22
22
supports_message_history = False
23
23
24
24
default_model = "ling-1t"
25
models = [default_model]
26
25
model_aliases = {
27
"ling-1t": default_model,
28
26
"ling": default_model,
29
27
}
28
models = [default_model]
30
29
31
30
@classmethod
32
31
async def create_async_generator(
@@ -37,10 +36,11 @@ class BAAI_Ling(AsyncGeneratorProvider, ProviderModelMixin):
37
36
conversation: JsonConversation = None,
38
37
**kwargs
39
38
) -> AsyncResult:
40
if conversation is None or not hasattr(conversation, 'session_hash'):
39
is_new_conversation = conversation is None or not hasattr(conversation, 'session_hash')
40
if is_new_conversation:
41
41
conversation = JsonConversation(session_hash=str(uuid.uuid4()).replace('-', '')[:12])
42
42
43
prompt = format_prompt(messages) if conversation is None else get_last_user_message(messages)
43
prompt = format_prompt(messages) if is_new_conversation else get_last_user_message(messages)
44
44
45
45
headers = {
46
46
'accept': '*/*',
@@ -62,6 +62,7 @@ class BAAI_Ling(AsyncGeneratorProvider, ProviderModelMixin):
62
62
async with aiohttp.ClientSession() as session:
63
63
async with session.post(cls.api_endpoint, headers=headers, json=payload, proxy=proxy) as response:
64
64
await raise_for_status(response)
65
# Response body must be consumed for the request to complete
65
66
await response.json()
66
67
67
68
data_url = f'{cls.url}/gradio_api/queue/data?session_hash={conversation.session_hash}'