Commit 0012a2a0 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent 6604d5b3
Loading
Loading
Loading
Loading
+14 −11
Original line number Diff line number Diff line
@@ -9,13 +9,10 @@
  # dependencies
  eval-type-backport,
  httpx,
  invoke,
  opentelemetry-api,
  opentelemetry-exporter-otlp-proto-http,
  opentelemetry-sdk,
  opentelemetry-semantic-conventions,
  pydantic,
  python-dateutil,
  pyyaml,
  typing-inspection,

  # optional-dependencies
@@ -26,19 +23,20 @@
  requests,

  # tests
  opentelemetry-sdk,
  pytestCheckHook,
}:

buildPythonPackage (finalAttrs: {
  pname = "mistralai";
  version = "1.12.3";
  version = "2.0.4";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "mistralai";
    repo = "client-python";
    tag = "v${finalAttrs.version}";
    hash = "sha256-DaCAO4/DoskFuRgNcoLWBuRdnxXfYhsgY/WZu/s1wk0=";
    hash = "sha256-SLPLj9rp8TTOSE3ldobBFU1+MpffzH1Bpshw+7LLUvU=";
  };

  preBuild = ''
@@ -50,18 +48,15 @@ buildPythonPackage (finalAttrs: {
  ];

  pythonRelaxDeps = [
    "opentelemetry-exporter-otlp-proto-http"
    "opentelemetry-semantic-conventions"
  ];
  dependencies = [
    eval-type-backport
    httpx
    invoke
    opentelemetry-api
    opentelemetry-exporter-otlp-proto-http
    opentelemetry-sdk
    opentelemetry-semantic-conventions
    pydantic
    python-dateutil
    pyyaml
    typing-inspection
  ];

@@ -80,7 +75,15 @@ buildPythonPackage (finalAttrs: {
  pythonImportsCheck = [ "mistralai" ];

  nativeCheckInputs = [
    opentelemetry-sdk
    pytestCheckHook
  ]
  ++ finalAttrs.passthru.optional-dependencies.agents
  ++ finalAttrs.passthru.optional-dependencies.gcp;

  disabledTests = [
    # AssertionError: <Response [200 OK]> is not an instance of <class 'mistralai.extra.observability.otel.TracedResponse'>
    "TestOtelTracing"
  ];

  meta = {