Native ollama provider

src/mcp_agent/llm/model_factory.py

@@ -17,6 +17,7 @@
 from mcp_agent.llm.providers.augmented_llm_generic import GenericAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_google_native import GoogleNativeAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_google_oai import GoogleOaiAugmentedLLM
+from mcp_agent.llm.providers.augmented_llm_ollama import OllamaAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_openai import OpenAIAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_openrouter import OpenRouterAugmentedLLM
 from mcp_agent.llm.providers.augmented_llm_tensorzero import TensorZeroAugmentedLLM
@@ -144,6 +145,7 @@ class ModelFactory:
         Provider.TENSORZERO: TensorZeroAugmentedLLM,
         Provider.AZURE: AzureOpenAIAugmentedLLM,
         Provider.ALIYUN: AliyunAugmentedLLM,
+        Provider.OLLAMA: OllamaAugmentedLLM,  # Add this line
     }
 
     # Mapping of special model names to their specific LLM classes
@@ -260,4 +262,4 @@ def factory(
             llm: AugmentedLLMProtocol = llm_class(**llm_args)
             return llm
 
-        return factory
+        return factory

src/mcp_agent/llm/provider_types.py

@@ -14,6 +14,7 @@ class Provider(Enum):
     GENERIC = "generic"
     GOOGLE_OAI = "googleoai"  # For Google through OpenAI libraries
     GOOGLE = "google"  # For Google GenAI native library
+    OLLAMA = "ollama"  # Add native Ollama
     OPENAI = "openai"
     OPENROUTER = "openrouter"
     TENSORZERO = "tensorzero"  # For TensorZero Gateway