Add ollama provider and update version to 0.1.2

CHANGELOG.md

@@ -1,6 +1,10 @@
 Release History
 ===============
 
+## 0.1.2 (2024-10-29)
+
+- Add ollama provider.
+
 ## 0.1.1 (2024-10-29)
 
 - Fix Groq provider.

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "simplemind"
-version = "0.1.1"
+version = "0.1.2"
 description = "An experimental client for AI providers that intends to replace LangChain and LangGraph for most common use cases."
 readme = "README.md"
 requires-python = ">=3.11"

test_ollama.py

@@ -1,66 +0,0 @@
-import os
-import unittest
-from unittest import mock
-import simplemind as sm
-from pydantic import BaseModel
-
-class TestOllama(unittest.TestCase):
-
-  def test_generate_text(self):
-    result = sm.generate_text(prompt="What is the meaning of life?", llm_provider="ollama", llm_model="llama3.2")
-    self.assertGreater(len(result), 0)
-    self.assertIsNotNone(result)
-
-  def test_create_conversation(self):
-    conversation = sm.create_conversation(llm_provider="ollama", llm_model="llama3.2")
-    conversation.add_message("user", "Remember the number 42.")
-    result = conversation.send()
-    self.assertIsNotNone(result)
-    self.assertGreaterEqual(len(result.text), 0)
-    self.assertIsInstance(result, sm.models.Message)
-
-  def test_memory(self):
-    class SimpleMemoryPlugin:
-      def __init__(self):
-          self.memories = [
-              "the earth has fictionally been destroyed.",
-              "the moon is made of cheese.",
-          ]
-
-      def yield_memories(self):
-          return (m for m in self.memories)
-
-      def send_hook(self, conversation: sm.Conversation):
-          for m in self.yield_memories():
-              conversation.prepend_system_message(role="system", text=m)
-    
-    conversation = sm.create_conversation(llm_provider="ollama", llm_model="llama3.2")
-
-    conversation.add_message(
-        role="user",
-        text="Write a poem about the moon",
-    )
-    self.assertGreater(len(conversation.messages), 0)
-    conversation.add_plugin(SimpleMemoryPlugin())
-    result = conversation.send()
-    self.assertGreater(len(conversation.messages), 2)
-    self.assertIsNotNone(result)
-    self.assertIsNotNone(result.text)
-    self.assertGreater(len(result.text), 0)
-    self.assertIsInstance(result, sm.models.Message)
-
-  def test_structure_response(self):
-      class Poem(BaseModel):
-          title: str
-          content: str
-      # Test for NotImplementedError
-      with self.assertRaises(NotImplementedError):
-          sm.generate_data(
-              prompt="Write a poem about love",
-              llm_provider="ollama",
-              llm_model="llama3.2",
-              response_model=Poem)
-
-
-if __name__ == '__main__':
-  unittest.main()