divi 0.0.1.dev9__tar.gz → 0.0.1.dev10__tar.gz

{divi-0.0.1.dev9 → divi-0.0.1.dev10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: divi
-Version: 0.0.1.dev9
+Version: 0.0.1.dev10
 Summary: The Agent Platform for Observability & Evaluation
 License-File: LICENSE
 Requires-Python: >=3.11

{divi-0.0.1.dev9 → divi-0.0.1.dev10}/divi/__init__.py

@@ -5,5 +5,5 @@ from . import proto
 name: str = "divi"
 run: Optional[Run] = None
 
-__version__ = "0.0.1.dev9"
+__version__ = "0.0.1.dev10"
 __all__ = ["init", "finish", "proto"]

{divi-0.0.1.dev9 → divi-0.0.1.dev10}/scripts/hatch_build.py

@@ -1,11 +1,17 @@
 import re
+import shutil
+import pathlib
 import sysconfig
+import subprocess
 
 from hatchling.builders.hooks.plugin.interface import BuildHookInterface
 
 
 class CustomBuildHook(BuildHookInterface):
     def initialize(self, version, build_data):
+        target_path = pathlib.Path("core", "main.go")
+        output_path = pathlib.Path("divi", "bin", "core")
+        self._go_build(target_path, output_path)
         build_data["tag"] = f"py3-none-{self._get_platform_tag()}"
         build_data["pure_python"] = False
 
@@ -45,3 +51,31 @@ class CustomBuildHook(BuildHookInterface):
         if os == "linux":
             parts[0] = "manylinux1"
         return "_".join(parts)
+
+    def _go_build(
+        self,
+        target_path: pathlib.PurePath,
+        output_path: pathlib.PurePath,
+    ):
+        """go build -o output_path target_path"""
+        output_flags = ["-o", str(output_path)]
+        subprocess.check_call(
+            [
+                str(self._get_and_require_go_binary()),
+                "build",
+                *output_flags,
+                str(target_path),
+            ],
+        )
+
+    def _get_and_require_go_binary(self) -> pathlib.Path:
+        go = shutil.which("go")
+
+        if not go:
+            self.app.abort(
+                "Did not find the 'go' binary. You need Go to build wandb"
+                " from source. See https://go.dev/doc/install.",
+            )
+            raise AssertionError("unreachable")
+
+        return pathlib.Path(go)

{divi-0.0.1.dev9 → divi-0.0.1.dev10}/scripts/proto_build.py

@@ -1,13 +1,10 @@
 #!/usr/bin/env python
 """Compile proto files to python and go files."""
 
-import importlib.metadata
 import os
-import pathlib
 
 import grpc_tools  # type: ignore
 from grpc_tools import protoc  # type: ignore
-from packaging import version
 
 
 proto_root = os.path.join(os.path.dirname(grpc_tools.__file__), "_proto")
@@ -23,9 +20,9 @@ for proto_file in [
             proto_root,
             "-I",
             ".",
-            f"--python_out=.",
-            f"--pyi_out=.",
-            f"--go_out=.",
+            "--python_out=.",
+            "--pyi_out=.",
+            "--go_out=.",
             f"divi/proto/{proto_file}",
         )
     )
@@ -42,11 +39,11 @@ for proto_file in [
             proto_root,
             "-I",
             ".",
-            f"--python_out=.",
-            f"--pyi_out=.",
-            f"--grpc_python_out=.",
-            f"--go_out=.",
-            f"--go-grpc_out=.",
+            "--python_out=.",
+            "--pyi_out=.",
+            "--grpc_python_out=.",
+            "--go_out=.",
+            "--go-grpc_out=.",
             f"divi/proto/{proto_file}",
         )
     )