github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-08 12:30:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore(model): Modify siliconflow name (#2194)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-12-12 16:47:14 +08:00
committed by GitHub
parent 4da1809b31
commit dab55493f5
10 changed files with 48 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
dbgpt/model/adapter/proxy_adapter.py
View File

@@ -349,7 +349,7 @@ class SiliconFlowProxyLLMModelAdapter(ProxyLLMModelAdapter):
return True
def do_match(self, lower_model_name_or_path: Optional[str] = None):
return lower_model_name_or_path == "silicon_flow_proxyllm"
return lower_model_name_or_path == "siliconflow_proxyllm"
def get_llm_client_class(
self, params: ProxyModelParameters
@@ -359,9 +359,9 @@ class SiliconFlowProxyLLMModelAdapter(ProxyLLMModelAdapter):
return SiliconFlowLLMClient
def get_async_generate_stream_function(self, model, model_path: str):
from dbgpt.model.proxy.llms.siliconflow import silicon_flow_generate_stream
from dbgpt.model.proxy.llms.siliconflow import siliconflow_generate_stream
return silicon_flow_generate_stream
return siliconflow_generate_stream
register_model_adapter(OpenAIProxyLLMModelAdapter)