feat(model): Add OllamaLLMClient to proxy module

eosphoros-ai · Apr 28, 2024 · e0c296f · e0c296f
1 parent b025d59
commit e0c296f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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",
 ]