alpaka 0.1.13__tar.gz

alpaka-0.1.13/PKG-INFO

@@ -0,0 +1,30 @@
+Metadata-Version: 2.1
+Name: alpaka
+Version: 0.1.13
+Summary: 
+Author: jhnnsrs
+Author-email: jhnnsrs@gmail.com
+Requires-Python: >=3.10,<4.0
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: aiohttp (>=3.9.5,<4.0.0)
+Requires-Dist: koil (>=1.0.0)
+Requires-Dist: ollama (>=0.3.3,<0.4.0)
+Description-Content-Type: text/markdown
+
+# alpaka
+
+[![codecov](https://codecov.io/gh/jhnnsrs/akuire/branch/master/graph/badge.svg?token=UGXEA2THBV)](https://codecov.io/gh/jhnnsrs/akuire)
+[![PyPI version](https://badge.fury.io/py/akuire.svg)](https://pypi.org/project/akuire/)
+[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://pypi.org/project/akuire/)
+![Maintainer](https://img.shields.io/badge/maintainer-jhnnsrs-blue)
+[![PyPI pyversions](https://img.shields.io/pypi/pyversions/akuire.svg)](https://pypi.python.org/pypi/akuire/)
+[![PyPI status](https://img.shields.io/pypi/status/akuire.svg)](https://pypi.python.org/pypi/akuire/)
+
+Alpaka is the arkitekt compüability layer around ollama and allows your arkitekt apps to connect to an ollama server
+
+
+
+

alpaka-0.1.13/README.md

@@ -0,0 +1,13 @@
+# alpaka
+
+[![codecov](https://codecov.io/gh/jhnnsrs/akuire/branch/master/graph/badge.svg?token=UGXEA2THBV)](https://codecov.io/gh/jhnnsrs/akuire)
+[![PyPI version](https://badge.fury.io/py/akuire.svg)](https://pypi.org/project/akuire/)
+[![Maintenance](https://img.shields.io/badge/Maintained%3F-yes-green.svg)](https://pypi.org/project/akuire/)
+![Maintainer](https://img.shields.io/badge/maintainer-jhnnsrs-blue)
+[![PyPI pyversions](https://img.shields.io/pypi/pyversions/akuire.svg)](https://pypi.python.org/pypi/akuire/)
+[![PyPI status](https://img.shields.io/pypi/status/akuire.svg)](https://pypi.python.org/pypi/akuire/)
+
+Alpaka is the arkitekt compüability layer around ollama and allows your arkitekt apps to connect to an ollama server
+
+
+

alpaka-0.1.13/alpaka/__arkitekt__.py

@@ -0,0 +1,45 @@
+from alpaka.alpaka import Alpaka
+from herre_next import Herre
+from fakts_next import Fakts
+from typing import Optional
+
+from arkitekt_next.base_models import Manifest
+
+from arkitekt_next.service_registry import (
+    Params,
+)
+from arkitekt_next.base_models import Requirement
+
+
+class ArkitektAlpaka(Alpaka):
+    endpoint_url: str = "fake_url"
+    fakts: Fakts
+    herre: Herre
+
+    async def aconnect(self, *args, **kwargs):
+        endpoint_url = await self.fakts.get("alpaka.endpoint_url")
+        self.endpoint_url = endpoint_url
+        await super().aconnect(*args, **kwargs)
+
+    class Config:
+        arbitrary_types_allowed = True
+
+
+def init_services(service_builder_registry):
+    def build_arkitekt_next_alpaka(
+        fakts: Fakts, herre: Herre, params: Params, manifest: Manifest
+    ):
+        return ArkitektAlpaka(
+            fakts=fakts,
+            herre=herre,
+        )
+
+    service_builder_registry.register(
+        "alpaka",
+        build_arkitekt_next_alpaka,
+        Requirement(
+            key="alpaka",
+            service="io.ollama.ollama",
+            description="An instance of Ollama to chat with",
+        ),
+    )

alpaka-0.1.13/alpaka/__init__.py

@@ -0,0 +1,2 @@
+from .alpaka import Alpaka
+from .funcs import chat, achat

alpaka-0.1.13/alpaka/alpaka.py

@@ -0,0 +1,41 @@
+from typing import TypeVar
+from koil.composition.base import KoiledModel
+from alpaka.vars import current_alpaka
+import asyncio
+from ollama import AsyncClient
+
+
+T = TypeVar("T")
+
+
+class Alpaka(KoiledModel):
+    endpoint_url: str
+    _client: AsyncClient = None
+
+    async def aconnect(self, **kwargs):
+        self._client = AsyncClient(self.endpoint_url)
+
+    async def chat(self, *args, **kwargs):
+        if not self._client:
+            await self.aconnect()
+
+        return await self._client.chat(*args, **kwargs)
+
+    async def pull(self, *args, **kwargs):
+        if not self._client:
+            await self.aconnect()
+
+        return await self._client.pull(*args, **kwargs)
+
+    async def __aenter__(self: T) -> T:
+        current_alpaka.set(self)
+        return await super().__aenter__()
+
+    async def __aexit__(self, exc_type, exc_value, traceback):
+        await super().__aexit__(exc_type, exc_value, traceback)
+        if self._client:
+            print("Closing client???", self._client)
+        current_alpaka.set(None)
+
+    class Config:
+        arbitrary_types_allowed = True

alpaka-0.1.13/alpaka/funcs.py

@@ -0,0 +1,21 @@
+from .vars import get_current_alpaka
+from koil import unkoil
+
+
+async def achat(*args, **kwargs):
+    alpaka = get_current_alpaka()
+    return await alpaka.chat(*args, **kwargs)
+
+
+def chat(*args, **kwargs):
+    return unkoil(achat, *args, **kwargs)
+
+
+async def apull(*args, **kwargs):
+    alpaka = get_current_alpaka()
+    print(args, kwargs)
+    return await alpaka.pull(*args, **kwargs)
+
+
+def pull(*args, **kwargs):
+    return unkoil(apull, *args, **kwargs)

alpaka-0.1.13/alpaka/vars.py

@@ -0,0 +1,12 @@
+from contextvars import ContextVar
+
+
+current_alpaka = ContextVar("current_alpaka", default=None)
+
+
+def get_current_alpaka():
+    return current_alpaka.get()
+
+
+def set_current_alpaka(value):
+    current_alpaka.set(value)

alpaka-0.1.13/pyproject.toml

@@ -0,0 +1,31 @@
+[tool.poetry]
+name = "alpaka"
+version = "0.1.13"
+description = ""
+authors = ["jhnnsrs <jhnnsrs@gmail.com>"]
+readme = "README.md"
+packages = [{ include = "alpaka" }]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+aiohttp = "^3.9.5"
+koil = ">=1.0.0"
+ollama = "^0.3.3"
+
+
+[tool.poetry.group.dev.dependencies]
+pytest-aiohttp = "^1.0.5"
+pytest-cov = "^5.0.0"
+
+[tool.poetry.scripts]
+akuire = "akuire.__main__:main"
+
+[tool.pytest.ini_options]
+markers = [
+    "composition: test composition",
+    "external: test external",
+]
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"