github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-28 22:37:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Modify ollama config

Browse Source
This commit is contained in:
Fangyin Cheng 2025-03-04 18:45:52 +08:00
parent c8e252c4de
commit a22de08936
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
packages/dbgpt-core/src/dbgpt/model/proxy/llms/ollama.py
View File

@ -9,13 +9,13 @@ from dbgpt.core.awel.flow import (
ResourceCategory, ResourceCategory,
auto_register_resource, auto_register_resource,
) )
from dbgpt.core.interface.parameter import LLMDeployModelParameters
from dbgpt.model.proxy.base import ( from dbgpt.model.proxy.base import (
AsyncGenerateStreamFunction, AsyncGenerateStreamFunction,
GenerateStreamFunction, GenerateStreamFunction,
ProxyLLMClient, ProxyLLMClient,
register_proxy_model_adapter, register_proxy_model_adapter,
) )
from dbgpt.model.proxy.llms.chatgpt import OpenAICompatibleDeployModelParameters
from dbgpt.model.proxy.llms.proxy_model import ProxyModel, parse_model_request from dbgpt.model.proxy.llms.proxy_model import ProxyModel, parse_model_request
from dbgpt.util.i18n_utils import _ from dbgpt.util.i18n_utils import _
@ -31,7 +31,7 @@ logger = logging.getLogger(__name__)
show_in_ui=False, show_in_ui=False,
) )
@dataclass @dataclass
class OllamaDeployModelParameters(OpenAICompatibleDeployModelParameters): class OllamaDeployModelParameters(LLMDeployModelParameters):
"""Deploy model parameters for Ollama.""" """Deploy model parameters for Ollama."""
provider: str = "proxy/ollama" provider: str = "proxy/ollama"