2024-03-15 10:01:03 +01:00
|
|
|
{ flake-inputs, pkgs, python3, runCommand }:
|
|
|
|
let
|
|
|
|
dream2nix = flake-inputs.dream2nix;
|
|
|
|
module = { dream2nix, config, lib, ... }: {
|
|
|
|
imports = [ dream2nix.modules.dream2nix.pip ];
|
|
|
|
name = "llm-env";
|
|
|
|
deps.python = python3;
|
2024-06-01 10:39:14 +02:00
|
|
|
version = "0.14";
|
2024-03-15 10:01:03 +01:00
|
|
|
pip.requirementsList = [
|
2024-06-01 10:39:14 +02:00
|
|
|
"llm==0.14"
|
2024-03-15 10:01:03 +01:00
|
|
|
"llm-claude-3==0.3"
|
2024-04-20 10:13:51 +02:00
|
|
|
"llm-replicate==0.3.1"
|
2024-06-01 10:39:14 +02:00
|
|
|
"llm-gemini==0.1a4"
|
|
|
|
"llm-cmd==0.1a0"
|
2024-03-15 10:01:03 +01:00
|
|
|
];
|
|
|
|
pip.flattenDependencies = true;
|
|
|
|
public = config.pip.env;
|
|
|
|
paths.projectRoot = ./..;
|
|
|
|
paths.package = "pkgs/llm";
|
|
|
|
};
|
|
|
|
packages = dream2nix.lib.evalModules {
|
|
|
|
packageSets.nixpkgs = pkgs;
|
|
|
|
modules = [ module ];
|
|
|
|
};
|
|
|
|
|
|
|
|
pyEnv = packages.config.public.pyEnv;
|
|
|
|
in
|
|
|
|
runCommand "llm" {} ''
|
|
|
|
mkdir -p $out/bin
|
|
|
|
cp ${pyEnv}/bin/llm $_
|
|
|
|
''
|