github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-22 03:41:43 +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,
auto_register_resource,
)
from dbgpt.core.interface.parameter import LLMDeployModelParameters
from dbgpt.model.proxy.base import (
AsyncGenerateStreamFunction,
GenerateStreamFunction,
ProxyLLMClient,
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.util.i18n_utils import _
@ -31,7 +31,7 @@ logger = logging.getLogger(__name__)
show_in_ui=False,
)
@dataclass
class OllamaDeployModelParameters(OpenAICompatibleDeployModelParameters):
class OllamaDeployModelParameters(LLMDeployModelParameters):
"""Deploy model parameters for Ollama."""
provider: str = "proxy/ollama"