x

contrib/builders/ollama.py

@@ -0,0 +1,38 @@
+from typing import TYPE_CHECKING
+
+
+if TYPE_CHECKING:
+
+    from fakts.base_models import LinkingContext 
+    from contrib.backends.docker_backend import DockerServiceDescriptor, SelfServiceDescriptor
+
+
+def _create_base_url(self: "SelfServiceDescriptor", context: "LinkingContext", descriptor: "DockerServiceDescriptor", inside_port="80/tcp"):
+    try:
+        outside_port = descriptor.port_map[inside_port]
+    except KeyError:
+        raise Exception(f"Service {descriptor.internal_host} does not expose port {inside_port} only exposes ports: " + str(descriptor.port_map.keys()))
+
+    protocol = "https" if context.request.is_secure else "http"
+    inside_base_url = f"{protocol}://{descriptor.internal_host}:{inside_port.split('/')[0]}"
+    outside_base_url = f"{protocol}://{context.request.host}:{outside_port}"
+
+    # Depending on how the service is accessed, we need to return the correct base_url
+    if context.request.host == self.internal_host:
+        base_url = inside_base_url
+    else:
+        base_url = outside_base_url
+
+    return base_url
+
+
+
+def ollama(self: "SelfServiceDescriptor", context: "LinkingContext", descriptor: "DockerServiceDescriptor"):
+
+    base_url = _create_base_url(self, context, descriptor, inside_port="11434/tcp")
+
+
+
+    return { "endpoint_url": base_url }
+
+