From 1c95252d10f10f52578fe4c8e55f8492949098b4 Mon Sep 17 00:00:00 2001
From: Aaron Bieber <aaron@bolddaemon.com>
Date: Wed, 24 Apr 2024 14:29:20 -0600
Subject: [PATCH] europa: enable ollama, add a watcher for enabling
 acceleration

---
 hosts/europa/default.nix  | 12 +++++++++++-
 lib/default.nix           |  9 +++++++++
 pull_requests/306375.json | 20 ++++++++++++++++++++
 3 files changed, 40 insertions(+), 1 deletion(-)
 create mode 100644 pull_requests/306375.json

diff --git a/hosts/europa/default.nix b/hosts/europa/default.nix
index 84d041c..1348d2c 100644
--- a/hosts/europa/default.nix
+++ b/hosts/europa/default.nix
@@ -7,7 +7,7 @@
 }:
 let
   inherit (inputs.stable.legacyPackages.${pkgs.system}) chirp beets;
-  inherit (xinlib) jobToUserService;
+  inherit (xinlib) jobToUserService prIsOpen;
   peerixUser =
     if builtins.hasAttr "peerix" config.users.users
     then config.users.users.peerix.name
@@ -190,6 +190,10 @@ in
   services.xinCA = { enable = false; };
 
   services = {
+    ollama = {
+      enable = true;
+      acceleration = prIsOpen.str 306375 "rocm";
+    };
     rkvm.server = {
       enable = true;
       settings = {
@@ -301,6 +305,11 @@ in
     user.services =
       lib.listToAttrs (builtins.map jobToUserService jobs);
     services = {
+      ollama = {
+        environment = {
+          OLLAMA_ORIGINS = "*";
+        };
+      };
       "whytailscalewhy" = {
         description = "Tailscale restart on resume";
         wantedBy = [ "post-resume.target" ];
@@ -367,6 +376,7 @@ in
       nix-top
       nmap
       obsidian
+      ollama
       openscad
       picocom
       proton-caller
diff --git a/lib/default.nix b/lib/default.nix
index 0fc6c35..9327743 100644
--- a/lib/default.nix
+++ b/lib/default.nix
@@ -8,6 +8,15 @@ let
     in
     prStatus;
   prIsOpen = {
+    str = pr: a:
+      let
+        prStatus = getPrStatus pr;
+      in
+      if prStatus.status == "open"
+      then 
+        lib.warn
+          "PR: ${toString pr} (${prStatus.title}) is open.. disabling option" null 
+      else a;
     option = pr: a:
       let
         prStatus = getPrStatus pr;
diff --git a/pull_requests/306375.json b/pull_requests/306375.json
new file mode 100644
index 0000000..be878bc
--- /dev/null
+++ b/pull_requests/306375.json
@@ -0,0 +1,20 @@
+{
+  "branches": [
+    "master",
+    "staging",
+    "staging-next"
+  ],
+  "error": "",
+  "pull_request": 306375,
+  "release": "unstable",
+  "status": "open",
+  "status_info": {
+    "master": true,
+    "nixos-unstable": false,
+    "nixos-unstable-small": false,
+    "nixpkgs-unstable": false,
+    "staging": true,
+    "staging-next": true
+  },
+  "title": "rocmPackages: use CMake 3.9.2 when building clr"
+}