diff --git a/pkgs/llm/llm-claude-3/default.nix b/pkgs/llm/llm-anthropic/default.nix similarity index 72% rename from pkgs/llm/llm-claude-3/default.nix rename to pkgs/llm/llm-anthropic/default.nix index a3a2492..883952d 100644 --- a/pkgs/llm/llm-claude-3/default.nix +++ b/pkgs/llm/llm-anthropic/default.nix @@ -12,15 +12,15 @@ }: buildPythonPackage rec { - pname = "llm-claude-3"; - version = "0.8"; + pname = "llm-anthropic"; + version = "0.12"; pyproject = true; src = fetchFromGitHub { owner = "simonw"; - repo = "llm-claude-3"; + repo = "llm-anthropic"; rev = version; - hash = "sha256-XhmxUo+nM6el17AVRUq+RLT5SEl+Q0eWhouU9WDZJl0="; + hash = "sha256-7+5j5jZBFfaaqnfjvLTI+mz1PUuG8sB5nD59UCpJuR4="; }; nativeBuildInputs = [ diff --git a/pkgs/llm/llm-gemini/default.nix b/pkgs/llm/llm-gemini/default.nix new file mode 100644 index 0000000..42fd8ca --- /dev/null +++ b/pkgs/llm/llm-gemini/default.nix @@ -0,0 +1,35 @@ +{ + buildPythonPackage, + fetchFromGitHub, + + # build-system + setuptools, + wheel, + + # deps + httpx, + ijson, + ... +}: + +buildPythonPackage rec { + pname = "llm-gemini"; + version = "0.10"; + pyproject = true; + + src = fetchFromGitHub { + owner = "simonw"; + repo = "llm-gemini"; + rev = version; + hash = "sha256-+ghsBvEY8GQAphdvG7Rdu3T/7yz64vmkuA1VGvqw1fU="; + }; + + nativeBuildInputs = [ + setuptools + wheel + ]; + + dependencies = [ httpx ijson ]; + + dontCheckRuntimeDeps = true; +} diff --git a/pkgs/llm/llm-with-plugins/default.nix b/pkgs/llm/llm-with-plugins/default.nix index 000c971..3477b58 100644 --- a/pkgs/llm/llm-with-plugins/default.nix +++ b/pkgs/llm/llm-with-plugins/default.nix @@ -5,12 +5,14 @@ let inherit (pkgs) python3Packages; llm = python3Packages.callPackage ../llm { }; - llm-claude-3 = python3Packages.callPackage ../llm-claude-3 { }; + llm-anthropic = python3Packages.callPackage ../llm-anthropic { }; + llm-gemini = python3Packages.callPackage ../llm-gemini { }; pyWithLlm = ( pkgs.python3.withPackages (_: [ llm - llm-claude-3 + llm-anthropic + llm-gemini ]) ); in diff --git a/pkgs/llm/llm/default.nix b/pkgs/llm/llm/default.nix index f32a108..2aa8367 100644 --- a/pkgs/llm/llm/default.nix +++ b/pkgs/llm/llm/default.nix @@ -22,7 +22,7 @@ let llm = buildPythonPackage rec { pname = "llm"; - version = "0.17.1"; + version = "0.21"; pyproject = true; build-system = [ setuptools ]; @@ -33,7 +33,7 @@ let owner = "simonw"; repo = "llm"; rev = "refs/tags/${version}"; - hash = "sha256-6OO0SIIxChM5HRJoUM4CYGbsINmc3i+iyL/oahLHhrY="; + hash = "sha256-gxmhdczgbcvbWJQTy+gek499C/3jm9WL5vKZmaGVWgU="; }; # patches = [ ./001-disable-install-uninstall-commands.patch ];