ollama

Author: mamico

setup.py

@@ -56,6 +56,10 @@
         "plone.restapi",
         "plone.volto",
         "requests",
+        "ollama",
+        "deepl",
+        "boto3",
+        "openai",
     ],
     extras_require={
         "test": [
@@ -67,15 +71,6 @@
             "pytest-cov",
             "pytest-plone>=0.5.0",
         ],
-        "deepl": [
-            "deepl",
-        ],
-        "aws": [
-            "boto3",
-        ],
-        "deepseek": [
-            "openai",
-        ],
         "libretranslate": [],
     },
     entry_points="""

src/collective/translators/configure.zcml

@@ -37,5 +37,6 @@
       package=".deepseek"
       zcml:condition="installed openai"
       />
+  <include package=".ollama" />
 
 </configure>

src/collective/translators/interfaces.py

@@ -10,7 +10,7 @@ class IBrowserLayer(IDefaultBrowserLayer):
     """Marker interface that defines a browser layer."""
 
 
-class BaseControlPanel(Interface):
+class IBaseControlPanel(Interface):
     enabled = schema.Bool(
         title=_("Enabled"),
         description=_("Translation service enabled"),
@@ -45,23 +45,23 @@ class BaseControlPanel(Interface):
     )
 
 
-class IDeeplControlPanel(BaseControlPanel):
+class IDeeplControlPanel(IBaseControlPanel):
     api_key = schema.TextLine(
         title=_("API Key"),
         description=_("The API key for the Deepl translation service."),
         required=False,
     )
 
 
-class IDeepSeekControlPanel(BaseControlPanel):
+class IDeepSeekControlPanel(IBaseControlPanel):
     api_key = schema.TextLine(
         title=_("API Key"),
         description=_("The API key for the DeepSeek service."),
         required=False,
     )
 
 
-class IAWSTranslateControlPanel(BaseControlPanel):
+class IAWSTranslateControlPanel(IBaseControlPanel):
     access_key = schema.TextLine(
         title=_("Access key"),
         description=_("The key for the access to AWS Translate service."),
@@ -82,7 +82,7 @@ class IAWSTranslateControlPanel(BaseControlPanel):
     )
 
 
-class ILibreTranslateControlPanel(BaseControlPanel):
+class ILibreTranslateControlPanel(IBaseControlPanel):
     api_key = schema.TextLine(
         title=_("API Key"),
         description=_("The API key for the Libre translate translation service."),

src/collective/translators/ollama/__init__.py

@@ -0,0 +1,28 @@
+<configure
+    xmlns="http://namespaces.zope.org/zope"
+    xmlns:browser="http://namespaces.zope.org/browser"
+    xmlns:i18n="http://namespaces.zope.org/i18n"
+    xmlns:plone="http://namespaces.plone.org/plone"
+    i18n_domain="collective.translators"
+    >
+
+  <utility
+      provides="plone.app.multilingual.interfaces.IExternalTranslationService"
+      name="ollama"
+      component=".utility.Ollama"
+      />
+
+  <browser:page
+      name="ollama-translator-controlpanel"
+      for="plone.base.interfaces.IPloneSiteRoot"
+      class=".controlpanel.ControlPanel"
+      permission="cmf.ManagePortal"
+      layer="..interfaces.IBrowserLayer"
+      />
+
+  <adapter
+      factory=".controlpanel.RegistryConfigletPanel"
+      name="ollama-controlpanel"
+      />
+
+</configure>

src/collective/translators/ollama/configure.zcml

@@ -0,0 +1,35 @@
+# from collective.translators import _
+# from zope import schema
+from collective.translators import _
+from collective.translators.interfaces import IBaseControlPanel
+from plone.app.registry.browser import controlpanel
+from plone.restapi.controlpanels import RegistryConfigletPanel
+from zope.component import adapter
+from zope.interface import Interface
+
+
+class IControlPanel(IBaseControlPanel):
+    pass
+    # api_key = schema.TextLine(
+    #     title=_("API Key"),
+    #     description=_("The API key for the DeepSeek service."),
+    #     required=False,
+    # )
+
+
+class ControlPanel(controlpanel.RegistryEditForm):
+    id = "OllamaControlPanel"
+    label = _("Ollama Translation Service")
+    schema = IControlPanel
+
+
+@adapter(Interface, Interface)
+class RegistryConfigletPanel(RegistryConfigletPanel):
+    """Ollama control panel"""
+
+    schema = IControlPanel
+    schema_prefix = "ollama"
+    configlet_id = "ollama-controlpanel"
+    configlet_category_id = "Products"
+    title = _("Ollama Settings")
+    group = "Products"

src/collective/translators/ollama/controlpanel.py

@@ -0,0 +1,41 @@
+from .controlpanel import IControlPanel
+from plone import api
+
+import ollama
+
+
+class OllamaFactory:
+    order = 30
+    model = "zongwei/gemma3-translator:1b"
+    prompt = "Translate the following text from {source_language} to {target_language}: {content}"
+
+    def is_available(self):
+        value = api.portal.get_registry_record(name="enabled", interface=IControlPanel)
+        return value
+
+    def available_languages(self):
+        return []
+
+    def translate_content(self, content, source_language, target_language):
+        try:
+            response = ollama.chat(
+                model=self.model,
+                messages=[
+                    {
+                        "role": "user",
+                        "content": self.prompt.format(
+                            source_language=source_language,
+                            target_language=target_language,
+                            content=content,
+                        ),
+                    }
+                ],
+            )
+            return response.message.content
+        # TODO: specific exception
+        except ollama.ResponseError:
+            # TODO: show_message here is useless, remove and return the error as
+            return None
+
+
+Ollama = OllamaFactory()

src/collective/translators/ollama/utility.py

@@ -0,0 +1,13 @@
+<?xml version="1.0" encoding="utf-8"?>
+<registry xmlns:i18n="http://xml.zope.org/namespaces/i18n"
+          i18n:domain="collective.translators"
+>
+
+  <records interface="collective.translators.ollama.controlpanel.IControlPanel">
+    <value key="enabled">False</value>
+    <value key="order">30</value>
+    <value key="source_languages" />
+    <value key="target_languages" />
+  </records>
+
+</registry>