From e0c296fd1801003730344de751031750bdb37e65 Mon Sep 17 00:00:00 2001
From: Fangyin Cheng <staneyffer@gmail.com>
Date: Sun, 28 Apr 2024 16:11:23 +0800
Subject: [PATCH] feat(model): Add OllamaLLMClient to proxy module

---
 dbgpt/model/proxy/__init__.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/dbgpt/model/proxy/__init__.py b/dbgpt/model/proxy/__init__.py
index 831456fbd..1c953ba23 100644
--- a/dbgpt/model/proxy/__init__.py
+++ b/dbgpt/model/proxy/__init__.py
@@ -11,6 +11,7 @@ def __lazy_import(name):
         "ZhipuLLMClient": "dbgpt.model.proxy.llms.zhipu",
         "YiLLMClient": "dbgpt.model.proxy.llms.yi",
         "MoonshotLLMClient": "dbgpt.model.proxy.llms.moonshot",
+        "OllamaLLMClient": "dbgpt.model.proxy.llms.ollama",
     }
 
     if name in module_path:
@@ -33,4 +34,5 @@ def __getattr__(name):
     "SparkLLMClient",
     "YiLLMClient",
     "MoonshotLLMClient",
+    "OllamaLLMClient",
 ]