gmakstutis/ragflow
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add gpt 5.2 / pro

Browse source
This commit is contained in:
Billy Bao 2025-12-12 16:40:53 +08:00
parent 0fcb1680fd
commit 17df7ceebc
3 changed files with 25 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
conf/llm_factories.json
View file

@ -7,6 +7,20 @@
"status": "1",
"rank": "999",
"llm": [
{
"llm_name": "gpt-5.2-pro",
"tags": "LLM,CHAT,400k,IMAGE2TEXT",
"max_tokens": 400000,
"model_type": "chat",
"is_tools": true
},
{
"llm_name": "gpt-5.2",
"tags": "LLM,CHAT,400k,IMAGE2TEXT",
"max_tokens": 400000,
"model_type": "chat",
"is_tools": true
},
{
"llm_name": "gpt-5.1",
"tags": "LLM,CHAT,400k,IMAGE2TEXT",

4
rag/llm/__init__.py
View file

@ -54,6 +54,7 @@ class SupportedLiteLLMProvider(StrEnum):
MiniMax = "MiniMax"
DeerAPI = "DeerAPI"
GPUStack = "GPUStack"
OpenAI = "OpenAI"
FACTORY_DEFAULT_BASE_URL = {
@ -78,6 +79,7 @@ FACTORY_DEFAULT_BASE_URL = {
SupportedLiteLLMProvider.ZHIPU_AI: "https://open.bigmodel.cn/api/paas/v4",
SupportedLiteLLMProvider.MiniMax: "https://api.minimaxi.com/v1",
SupportedLiteLLMProvider.DeerAPI: "https://api.deerapi.com/v1",
SupportedLiteLLMProvider.OpenAI: "https://api.openai.com/v1",
}
@ -113,6 +115,8 @@ LITELLM_PROVIDER_PREFIX = {
SupportedLiteLLMProvider.MiniMax: "openai/",
SupportedLiteLLMProvider.DeerAPI: "openai/",
SupportedLiteLLMProvider.GPUStack: "openai/",
SupportedLiteLLMProvider.OpenAI: "openai/",
}
ChatModel = globals().get("ChatModel", {})

13
rag/llm/chat_model.py
View file

@ -487,13 +487,13 @@ class Base(ABC):
assert False, "Shouldn't be here."
class GptTurbo(Base):
_FACTORY_NAME = "OpenAI"
# class GptTurbo(Base):
# _FACTORY_NAME = "OpenAI"
def __init__(self, key, model_name="gpt-3.5-turbo", base_url="https://api.openai.com/v1", **kwargs):
if not base_url:
base_url = "https://api.openai.com/v1"
super().__init__(key, model_name, base_url, **kwargs)
# def __init__(self, key, model_name="gpt-3.5-turbo", base_url="https://api.openai.com/v1", **kwargs):
# if not base_url:
# base_url = "https://api.openai.com/v1"
# super().__init__(key, model_name, base_url, **kwargs)
class XinferenceChat(Base):
@ -1227,6 +1227,7 @@ class LiteLLMBase(ABC):
"MiniMax",
"DeerAPI",
"GPUStack",
"OpenAI",
]
def __init__(self, key, model_name, base_url=None, **kwargs):