pyrox.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 setuptools,
6 llm,
7 llm-command-r,
8 cohere,
9 pytestCheckHook,
10 pytest-recording,
11 writableTmpDirAsHomeHook,
12}:
13
14buildPythonPackage rec {
15 pname = "llm-command-r";
16 version = "0.3.1";
17 pyproject = true;
18
19 src = fetchFromGitHub {
20 owner = "simonw";
21 repo = "llm-command-r";
22 tag = version;
23 hash = "sha256-PxICRds9NJQP64HwoL7Oxd39yaIrMdAyQEbhaumJCgo=";
24 };
25
26 build-system = [ setuptools ];
27
28 dependencies = [
29 cohere
30 llm
31 ];
32
33 nativeCheckInputs = [
34 pytestCheckHook
35 pytest-recording
36 writableTmpDirAsHomeHook
37 ];
38
39 pythonImportsCheck = [ "llm_command_r" ];
40
41 passthru.tests = llm.mkPluginTest llm-command-r;
42
43 meta = {
44 description = "Access the Cohere Command R family of models";
45 homepage = "https://github.com/simonw/llm-command-r";
46 changelog = "https://github.com/simonw/llm-command-r/releases/tag/${version}/CHANGELOG.md";
47 license = lib.licenses.asl20;
48 maintainers = with lib.maintainers; [ philiptaron ];
49 };
50}