XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Add model_aliases to HarProvider

36f5fc70
hlohaus <983577+hlohaus@users.noreply.github.com>
提交于

代码差异

2 个文件 +6 -3
Modified g4f/Provider/PollinationsAI.py +1 -0
@@ -127,6 +127,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
127 127 "grok-3-mini": "grok",
128 128
129 129 ### Audio Models ###
130 "gpt-4o-audio": "openai-audio",
130 131 "gpt-4o-mini-audio": "openai-audio",
131 132
132 133 ### Image Models ###
Modified g4f/Provider/har/__init__.py +5 -3
@@ -3,6 +3,7 @@ from __future__ import annotations
3 3 import os
4 4 import json
5 5 import uuid
6 import random
6 7 from urllib.parse import urlparse
7 8
8 9 from ...typing import AsyncResult, Messages, MediaListType
@@ -14,6 +15,7 @@ from ...errors import ResponseError
14 15 from ..base_provider import AsyncGeneratorProvider, ProviderModelMixin
15 16 from ..helper import get_last_user_message
16 17 from ..openai.har_file import get_headers
18 from ..LegacyLMArena import LegacyLMArena
17 19
18 20 class HarProvider(AsyncGeneratorProvider, ProviderModelMixin):
19 21 label = "LMArena (Har)"
@@ -21,9 +23,7 @@ class HarProvider(AsyncGeneratorProvider, ProviderModelMixin):
21 23 api_endpoint = "/queue/join?"
22 24 working = True
23 25 default_model = "chatgpt-4o-latest-20250326"
24 model_aliases = {
25 "claude-3.7-sonnet": "claude-3-7-sonnet-20250219",
26 }
26 model_aliases = LegacyLMArena.model_aliases
27 27 vision_models = [
28 28 "o3-2025-04-16",
29 29 "o4-mini-2025-04-16",
@@ -155,6 +155,8 @@ class HarProvider(AsyncGeneratorProvider, ProviderModelMixin):
155 155 yield new_content
156 156 if model in cls.model_aliases:
157 157 model = cls.model_aliases[model]
158 if isinstance(model, list):
159 model = random.choice(model)
158 160 prompt = get_last_user_message(messages)
159 161 async with StreamSession(impersonate="chrome") as session:
160 162 if conversation is None: