Switch Python to unstable and add pydantic-ai

Pydantic-ai is an agent framework that seems simple and good: well-typed with
pydantic, tool usage is just an `@tool` decorator on a function, and so
on. While building these I realized there were some deps they needed that were
already in nixpkgs unstable, so I just switched to that instead of trying to
backport all the versions and stuff.

6 files changed, 238 insertions, 99 deletions

diff --git a/Omni/Bild/Deps/Python.nix b/Omni/Bild/Deps/Python.nix
index 3a0562d..e7c796d 100644
--- a/Omni/Bild/Deps/Python.nix
+++ b/Omni/Bild/Deps/Python.nix
@@ -8,6 +8,10 @@
   "nltk"
   "ollama"
   "openai"
+  "pydantic"
+  "pydantic-ai"
+  "pydantic-ai-slim"
+  "pydantic-graph"
   "requests"
   "slixmpp"
   "sqids"
diff --git a/Omni/Bild/Deps/logfire-api.nix b/Omni/Bild/Deps/logfire-api.nix
new file mode 100644
index 0000000..af6eedf
--- /dev/null
+++ b/Omni/Bild/Deps/logfire-api.nix
@@ -0,0 +1,24 @@
+{
+  lib,
+  buildPythonPackage,
+  sources,
+  hatchling,
+  pythonOlder,
+}:
+buildPythonPackage rec {
+  pname = "logfire-api";
+  version = sources.logfire.rev;
+  pyproject = true;
+  disabled = pythonOlder "3.8";
+  src = sources.logfire;
+  sourceRoot = "logfire-src/logfire-api";
+  build-system = [hatchling];
+  pythonImportsCheck = ["logfire_api"];
+  meta = {
+    description = "Shim for the Logfire SDK which does nothing unless Logfire is installed";
+    homepage = "https://pypi.org/project/logfire-api/";
+    changelog = "https://github.com/pydantic/logfire/releases/tag/v${version}";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [bsima];
+  };
+}
diff --git a/Omni/Bild/Deps/openai-python.nix b/Omni/Bild/Deps/openai-python.nix
deleted file mode 100644
index 79db11c..0000000
--- a/Omni/Bild/Deps/openai-python.nix
+++ /dev/null
@@ -1,99 +0,0 @@
-{
-  lib,
-  buildPythonPackage,
-  pythonOlder,
-  # build-system
-  hatchling,
-  hatch-fancy-pypi-readme,
-  # dependencies
-  anyio,
-  distro,
-  httpx,
-  jiter,
-  pydantic,
-  sniffio,
-  tqdm,
-  typing-extensions,
-  numpy,
-  pandas,
-  pandas-stubs,
-  # check deps
-  pytestCheckHook,
-  dirty-equals,
-  inline-snapshot,
-  nest-asyncio,
-  pytest-asyncio,
-  pytest-mock,
-  respx,
-  sources,
-}:
-buildPythonPackage rec {
-  pname = "openai";
-  version = sources.openai-python.version;
-  pyproject = true;
-
-  disabled = pythonOlder "3.8";
-
-  src = sources.openai-python;
-
-  build-system = [
-    hatchling
-    hatch-fancy-pypi-readme
-  ];
-
-  dependencies = [
-    anyio
-    distro
-    httpx
-    jiter
-    pydantic
-    sniffio
-    tqdm
-    typing-extensions
-  ];
-
-  optional-dependencies = {
-    datalib = [
-      numpy
-      pandas
-      pandas-stubs
-    ];
-  };
-
-  pythonImportsCheck = ["openai"];
-
-  nativeCheckInputs = [
-    pytestCheckHook
-    dirty-equals
-    inline-snapshot
-    nest-asyncio
-    pytest-asyncio
-    pytest-mock
-    respx
-  ];
-
-  pytestFlagsArray = [
-    "-W"
-    "ignore::DeprecationWarning"
-  ];
-
-  disabledTests = [
-    # Tests make network requests
-    "test_copy_build_request"
-    "test_basic_attribute_access_works"
-  ];
-
-  disabledTestPaths = [
-    # Test makes network requests
-    "tests/api_resources"
-  ];
-
-  meta = with lib; {
-    description = "Python client library for the OpenAI API";
-    homepage = "https://github.com/openai/openai-python";
-    changelog = "https://github.com/openai/openai-python/releases/tag/v${version}";
-    license = licenses.mit;
-    maintainers = with maintainers; [malo];
-    mainProgram = "openai";
-  };
-}
diff --git a/Omni/Bild/Deps/pydantic-ai-slim.nix b/Omni/Bild/Deps/pydantic-ai-slim.nix
new file mode 100644
index 0000000..067508b
--- /dev/null
+++ b/Omni/Bild/Deps/pydantic-ai-slim.nix
@@ -0,0 +1,90 @@
+{
+  lib,
+  buildPythonPackage,
+  hatchling,
+  pydantic,
+  logfire-api,
+  httpx,
+  eval-type-backport,
+  griffe,
+  pydantic-graph,
+  pythonOlder,
+  sources,
+  writeTextFile,
+}: let
+  version = sources.pydantic-ai.version;
+  pyproject_toml = writeTextFile {
+    name = "pyproject.toml";
+    text = ''
+      [build-system]
+      requires = ["hatchling"]
+      build-backend = "hatchling.build"
+
+      [project]
+      name = "pydantic-ai-slim"
+      version = "${version}"
+      description = "Agent Framework / shim to use Pydantic with LLMs, slim package"
+      authors = [{ name = "Samuel Colvin", email = "samuel@pydantic.dev" }]
+      license = "MIT"
+      readme = "README.md"
+      requires-python = ">=3.9"
+      dependencies = [
+          "eval-type-backport>=0.2.0",
+          "griffe>=1.3.2",
+          "httpx>=0.27",
+          "pydantic>=2.10",
+          "pydantic-graph==0.1.9",
+          "exceptiongroup; python_version < '3.11'",
+          "opentelemetry-api>=1.28.0",
+          "typing-inspection>=0.4.0",
+      ]
+
+      [tool.hatch.metadata]
+      allow-direct-references = true
+
+      [project.scripts]
+      pai = "pydantic_ai._cli:app"
+
+      [tool.hatch.build.targets.wheel]
+      packages = ["pydantic_ai"]
+    '';
+  };
+in
+  buildPythonPackage rec {
+    pname = "pydantic-ai-slim";
+    inherit version;
+    pyproject = true;
+    disabled = pythonOlder "3.8";
+    src = sources.pydantic-ai;
+    build-system = [hatchling];
+    sourceRoot = "pydantic-ai-src/pydantic_ai_slim";
+    dependencies = [
+      pydantic
+      logfire-api
+      httpx
+      eval-type-backport
+      griffe
+      pydantic-graph
+    ];
+    nativeCheckInputs = [
+      pydantic
+      logfire-api
+      httpx
+      eval-type-backport
+      griffe
+      pydantic-graph
+    ];
+    preBuild = ''
+      cp ${pyproject_toml} ./pyproject.toml
+    '';
+    pythonImportsCheck = [
+      "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere]"
+    ];
+    meta = {
+      description = "Graph and finite state machine library";
+      homepage = "https://github.com/pydantic/pydantic-ai";
+      changelog = "https://github.com/pydantic/pydantic-ai/releases/tag/v${version}";
+      license = lib.licenses.mit;
+      maintainers = with lib.maintainers; [bsima];
+    };
+  }
diff --git a/Omni/Bild/Deps/pydantic-ai.nix b/Omni/Bild/Deps/pydantic-ai.nix
new file mode 100644
index 0000000..399649d
--- /dev/null
+++ b/Omni/Bild/Deps/pydantic-ai.nix
@@ -0,0 +1,75 @@
+{
+  lib,
+  buildPythonPackage,
+  hatchling,
+  pydantic-ai-slim,
+  pythonOlder,
+  pytest-vcr,
+  dirty-equals,
+  sources,
+  writeTextFile,
+}: let
+  version = sources.pydantic-ai.version;
+  pyproject_toml = writeTextFile {
+    name = "pyproject.toml";
+    text = ''
+      [build-system]
+      requires = ["hatchling"]
+      build-backend = "hatchling.build"
+
+      [project]
+      name = "pydantic-ai"
+      version = "${version}"
+      description = "Agent Framework / shim to use Pydantic with LLMs"
+      authors = [
+          { name = "Samuel Colvin", email = "samuel@pydantic.dev" },
+          { name = "Marcelo Trylesinski", email = "marcelotryle@gmail.com" },
+          { name = "David Montague", email = "david@pydantic.dev" },
+          { name = "Alex Hall", email = "alex@pydantic.dev" },
+      ]
+      license = "MIT"
+      readme = "README.md"
+      requires-python = ">=3.9"
+      dependencies = [
+        "pydantic-ai-slim[openai,vertexai,groq,anthropic,mistral,cohere,bedrock,cli,mcp,evals]==${version}",
+      ]
+
+      [project.urls]
+      Homepage = "https://ai.pydantic.dev"
+      Source = "https://github.com/pydantic/pydantic-ai"
+      Documentation = "https://ai.pydantic.dev"
+      Changelog = "https://github.com/pydantic/pydantic-ai/releases"
+
+      [project.scripts]
+      pai = "pydantic_ai._cli:app"
+    '';
+  };
+in
+  buildPythonPackage rec {
+    pname = "pydantic-ai";
+    inherit version;
+    pyproject = true;
+    disabled = pythonOlder "3.8";
+    src = sources.pydantic-ai;
+    build-system = [hatchling];
+    dependencies = [pydantic-ai-slim];
+    nativeCheckInputs = [
+      pydantic-ai-slim
+      pytest-vcr
+      dirty-equals
+      # pytestCheckHook
+    ];
+    preBuild = ''
+      cp ${pyproject_toml} ./pyproject.toml
+    '';
+    pythonImportsCheck = [
+      "pydantic_ai"
+    ];
+    meta = {
+      description = "Agent Framework / shim to use Pydantic with LLMs";
+      homepage = "https://github.com/pydantic/pydantic-ai";
+      changelog = "https://github.com/pydantic/pydantic-ai/releases/tag/v${version}";
+      license = lib.licenses.mit;
+      maintainers = with lib.maintainers; [bsima];
+    };
+  }
diff --git a/Omni/Bild/Deps/pydantic-graph.nix b/Omni/Bild/Deps/pydantic-graph.nix
new file mode 100644
index 0000000..e2797b9
--- /dev/null
+++ b/Omni/Bild/Deps/pydantic-graph.nix
@@ -0,0 +1,45 @@
+{
+  lib,
+  buildPythonPackage,
+  hatchling,
+  pydantic,
+  logfire-api,
+  httpx,
+  opentelemetry-api,
+  pythonOlder,
+  sources,
+}:
+buildPythonPackage rec {
+  pname = "pydantic-graph";
+  version = sources.pydantic-ai.version;
+  pyproject = true;
+  disabled = pythonOlder "3.8";
+  src = sources.pydantic-ai;
+  sourceRoot = "pydantic-ai-src/pydantic_graph";
+  build-system = [hatchling];
+  dependencies = [
+    pydantic
+    logfire-api
+    httpx
+    opentelemetry-api
+  ];
+  nativeCheckInputs = [
+    pydantic
+    logfire-api
+    httpx
+  ];
+  pythonRelaxDeps = true;
+  postPatch = ''
+    substituteInPlace pyproject.toml \
+      --replace-fail ', "uv-dynamic-versioning>=0.7.0"' "" \
+      --replace-fail 'dynamic = ["version"]' 'version = "${version}"'
+  '';
+  pythonImportsCheck = ["pydantic_graph"];
+  meta = {
+    description = "PydanticAI core logic with minimal required dependencies.";
+    homepage = "https://github.com/pydantic/pydantic-ai";
+    changelog = "https://github.com/pydantic/pydantic-ai/releases/tag/v${version}";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [bsima];
+  };
+}