Add ollama based local model services in AgentScope. (#60)

* Add ollama api

* bug fix

* include ollama in __init__.py (agentscope.models package)

* bug fix

* Add unit test for ollama

* bug fix and reformat

* add package ollama

* reformat

* add logging directory management

* reformat

* re-correct ollama test with single instance

* re-correct ollama test with single instance

* reformat

setup.py

@@ -44,12 +44,14 @@
     "tiktoken",
     "Pillow",
     "requests",
-    "openai>=1.3.0",
     "numpy",
     "Flask==3.0.0",
     "Flask-Cors==4.0.0",
     "Flask-SocketIO==5.3.6",
+    # TODO: move into other requires
     "dashscope==1.14.1",
+    "openai>=1.3.0",
+    "ollama>=0.1.7",
 ]
 
 distribute_requires = minimal_requires + rpc_requires

src/agentscope/models/__init__.py

@@ -22,6 +22,11 @@
     DashScopeImageSynthesisWrapper,
     DashScopeTextEmbeddingWrapper,
 )
+from .ollama_model import (
+    OllamaChatWrapper,
+    OllamaEmbeddingWrapper,
+    OllamaGenerationWrapper,
+)
 
 
 __all__ = [
@@ -39,6 +44,9 @@
     "DashScopeChatWrapper",
     "DashScopeImageSynthesisWrapper",
     "DashScopeTextEmbeddingWrapper",
+    "OllamaChatWrapper",
+    "OllamaEmbeddingWrapper",
+    "OllamaGenerationWrapper",
 ]
 
 _MODEL_CONFIGS: dict[str, dict] = {}
@@ -97,7 +105,10 @@ def load_model_by_config_name(config_name: str) -> ModelWrapperBase:
         )
 
     model_type = config.model_type
-    return _get_model_wrapper(model_type=model_type)(**config)
+
+    kwargs = {k: v for k, v in config.items() if k != "model_type"}
+
+    return _get_model_wrapper(model_type=model_type)(**kwargs)
 
 
 def clear_model_configs() -> None: