wandb 0.21.2__py3-none-macosx_12_0_arm64.whl → 0.21.4__py3-none-macosx_12_0_arm64.whl

wandb/integration/dspy/dspy.py

@@ -0,0 +1,422 @@
+"""DSPy ↔ Weights & Biases integration."""
+
+from __future__ import annotations
+
+import logging
+import os
+from collections.abc import Mapping, Sequence
+from typing import Any, Literal
+
+import wandb
+import wandb.util
+from wandb.sdk.wandb_run import Run
+
+dspy = wandb.util.get_module(
+    name="dspy",
+    required=(
+        "To use the W&B DSPy integration you need to have the `dspy` "
+        "python package installed.  Install it with `uv pip install dspy`."
+    ),
+    lazy=False,
+)
+if dspy is not None:
+    assert dspy.__version__ >= "3.0.0", (
+        "DSPy 3.0.0 or higher is required. You have " + dspy.__version__
+    )
+
+
+logger = logging.getLogger(__name__)
+
+
+def _flatten_rows(rows: list[dict[str, Any]]) -> list[dict[str, Any]]:
+    """Flatten a list of nested row dicts into flat key/value dicts.
+
+    Args:
+        rows (list[dict[str, Any]]): List of nested dictionaries to flatten.
+
+    Returns:
+        list[dict[str, Any]]: List of flattened dictionaries.
+
+    """
+
+    def _flatten(
+        d: dict[str, Any], parent_key: str = "", sep: str = "."
+    ) -> dict[str, Any]:
+        items = []
+        for k, v in d.items():
+            new_key = f"{parent_key}{sep}{k}" if parent_key else k
+            if isinstance(v, dict):
+                items.extend(_flatten(v, new_key, sep=sep).items())
+            else:
+                items.append((new_key, v))
+        return dict(items)
+
+    return [_flatten(row) for row in rows]
+
+
+class WandbDSPyCallback(dspy.utils.BaseCallback):
+    """W&B callback for tracking DSPy evaluation and optimization.
+
+    This callback logs evaluation scores, per-step predictions (optional), and
+    a table capturing the DSPy program signature over time. It can also save
+    the best program as a W&B Artifact for reproducibility.
+
+    Examples:
+        Basic usage within DSPy settings:
+
+        ```python
+        import dspy
+        import wandb
+        from wandb.integration.dspy import WandbDSPyCallback
+
+        with wandb.init(project="dspy-optimization") as run:
+            dspy.settings.callbacks.append(WandbDSPyCallback(run=run))
+            # Run your DSPy optimization/evaluation
+        ```
+    """
+
+    def __init__(self, log_results: bool = True, run: Run | None = None) -> None:
+        """Initialize the callback.
+
+        Args:
+            log_results (bool): Whether to log per-evaluation prediction tables.
+            run (Run | None): Optional W&B run to use. Defaults to the
+                current global run if available.
+
+        Raises:
+            wandb.Error: If no active run is provided or found.
+        """
+        # If no run is provided, use the current global run if available.
+        if run is None:
+            if wandb.run is None:
+                raise wandb.Error(
+                    "You must call `wandb.init()` before instantiating WandbDSPyCallback()."
+                )
+            run = wandb.run
+
+        self.log_results = log_results
+
+        with wandb.wandb_lib.telemetry.context(run=run) as tel:
+            tel.feature.dspy_callback = True
+
+        self._run = run
+        self._did_log_config: bool = False
+        self._program_info: dict[str, Any] = {}
+        self._program_table: wandb.Table | None = None
+        self._row_idx: int = 0
+
+    def _flatten_dict(
+        self, nested: Any, parent_key: str = "", sep: str = "."
+    ) -> dict[str, Any]:
+        """Recursively flatten arbitrarily nested mappings and sequences.
+
+        Args:
+            nested (Any): Nested structure of mappings/lists to flatten.
+            parent_key (str): Prefix to prepend to keys in the flattened output.
+            sep (str): Key separator for nested fields.
+
+        Returns:
+            dict[str, Any]: Flattened dictionary representation.
+        """
+        flat: dict[str, Any] = {}
+
+        def _walk(obj: Any, base: str) -> None:
+            if isinstance(obj, Mapping):
+                for k, v in obj.items():
+                    new_key = f"{base}{sep}{k}" if base else str(k)
+                    _walk(v, new_key)
+            elif isinstance(obj, Sequence) and not isinstance(
+                obj, (str, bytes, bytearray)
+            ):
+                for idx, v in enumerate(obj):
+                    new_key = f"{base}{sep}{idx}" if base else str(idx)
+                    _walk(v, new_key)
+            else:
+                # Base can be empty only if the top-level is a scalar; guard against that.
+                key = base if base else ""
+                if key:
+                    flat[key] = obj
+
+        _walk(nested, parent_key)
+        return flat
+
+    def _extract_fields(self, fields: list[dict[str, Any]]) -> dict[str, str]:
+        """Convert signature fields to a flat mapping of strings.
+
+        Note:
+            The input is expected to be a dict-like mapping from field names to
+            field metadata. Values are stringified for logging.
+
+        Args:
+            fields (list[dict[str, Any]]): Mapping of field name to metadata.
+
+        Returns:
+            dict[str, str]: Mapping of field name to string value.
+        """
+        return {k: str(v) for k, v in fields.items()}
+
+    def _extract_program_info(self, program_obj: Any) -> dict[str, Any]:
+        """Extract signature-related info from a DSPy program.
+
+        Attempts to read the program signature, instructions, input and output
+        fields from a DSPy `Predict` parameter if available.
+
+        Args:
+            program_obj (Any): DSPy program/module instance.
+
+        Returns:
+            dict[str, Any]: Flattened dictionary of signature metadata.
+        """
+        info_dict = {}
+
+        if program_obj is None:
+            return info_dict
+
+        try:
+            sig = next(
+                param.signature
+                for _, param in program_obj.named_parameters()
+                if isinstance(param, dspy.Predict)
+            )
+
+            if getattr(sig, "signature", None):
+                info_dict["signature"] = sig.signature
+            if getattr(sig, "instructions", None):
+                info_dict["instructions"] = sig.instructions
+            if getattr(sig, "input_fields", None):
+                input_fields = sig.input_fields
+                info_dict["input_fields"] = self._extract_fields(input_fields)
+            if getattr(sig, "output_fields", None):
+                output_fields = sig.output_fields
+                info_dict["output_fields"] = self._extract_fields(output_fields)
+
+            return self._flatten_dict(info_dict)
+        except Exception as e:
+            logger.warning(
+                "Failed to extract program info from Evaluate instance: %s", e
+            )
+        return info_dict
+
+    def on_evaluate_start(
+        self,
+        call_id: str,
+        instance: Any,
+        inputs: dict[str, Any],
+    ) -> None:
+        """Handle start of a DSPy evaluation call.
+
+        Logs non-private fields from the evaluator instance to W&B config and
+        captures program signature info for later logging.
+
+        Args:
+            call_id (str): Unique identifier for the evaluation call.
+            instance (Any): The evaluation instance (e.g., `dspy.Evaluate`).
+            inputs (dict[str, Any]): Inputs passed to the evaluation (may
+                include a `program` key with the DSPy program).
+        """
+        if not self._did_log_config:
+            instance_vars = vars(instance) if hasattr(instance, "__dict__") else {}
+            serializable = {
+                k: v for k, v in instance_vars.items() if not k.startswith("_")
+            }
+            if "devset" in serializable:
+                # we don't want to log the devset in the config
+                del serializable["devset"]
+
+            self._run.config.update(serializable)
+            self._did_log_config = True
+
+        # 2) Build/append program signature tables from the 'program' inputs
+        if program_obj := inputs.get("program"):
+            self._program_info = self._extract_program_info(program_obj)
+
+    def on_evaluate_end(
+        self,
+        call_id: str,
+        outputs: Any | None,
+        exception: Exception | None = None,
+    ) -> None:
+        """Handle end of a DSPy evaluation call.
+
+        If available, logs a numeric `score` metric and (optionally) per-step
+        prediction tables. Always appends a row to the program-signature table.
+
+        Args:
+            call_id (str): Unique identifier for the evaluation call.
+            outputs (Any | None): Evaluation outputs; supports
+                `dspy.evaluate.evaluate.EvaluationResult`.
+            exception (Exception | None): Exception raised during evaluation, if any.
+        """
+        # The `BaseCallback` does not define the interface for the `outputs` parameter,
+        # Currently, we know of `EvaluationResult` which is a subclass of `dspy.Prediction`.
+        # We currently support this type and will warn the user if a different type is passed.
+        score: float | None = None
+        if exception is None:
+            if isinstance(outputs, dspy.evaluate.evaluate.EvaluationResult):
+                # log the float score as a wandb metric
+                score = outputs.score
+                wandb.log({"score": float(score)}, step=self._row_idx)
+
+                # Log the predictions as a separate table for each eval end.
+                # We know that results if of type `list[tuple["dspy.Example", "dspy.Example", Any]]`
+                results = outputs.results
+                if self.log_results:
+                    rows = self._parse_results(results)
+                    if rows:
+                        self._log_predictions_table(rows)
+            else:
+                wandb.termwarn(
+                    f"on_evaluate_end received unexpected outputs type: {type(outputs)}. "
+                    "Expected dspy.evaluate.evaluate.EvaluationResult; skipping logging score and `log_results`."
+                )
+        else:
+            wandb.termwarn(
+                f"on_evaluate_end received exception: {exception}. "
+                "Skipping logging score and `log_results`."
+            )
+
+        # Log the program signature iteratively
+        if self._program_table is None:
+            columns = ["step", *self._program_info.keys()]
+            if isinstance(score, float):
+                columns.append("score")
+            self._program_table = wandb.Table(columns=columns, log_mode="INCREMENTAL")
+
+        if self._program_table is not None:
+            values = list(self._program_info.values())
+            if isinstance(score, float):
+                values.append(score)
+
+            self._program_table.add_data(
+                self._row_idx,
+                *values,
+            )
+            self._run.log(
+                {"program_signature": self._program_table}, step=self._row_idx
+            )
+
+        self._row_idx += 1
+
+    def _parse_results(
+        self,
+        results: list[tuple[dspy.Example, dspy.Prediction | dspy.Completions, bool]],
+    ) -> list[dict[str, Any]]:
+        """Normalize evaluation results into serializable row dicts.
+
+        Args:
+            results (list[tuple]): Sequence of `(example, prediction, is_correct)`
+                tuples from DSPy evaluation.
+
+        Returns:
+            list[dict[str, Any]]: Rows with `example`, `prediction`, `is_correct`.
+        """
+        _rows: list[dict[str, Any]] = []
+        for example, prediction, is_correct in results:
+            if isinstance(prediction, dspy.Prediction):
+                prediction_dict = prediction.toDict()
+            if isinstance(prediction, dspy.Completions):
+                prediction_dict = prediction.items()
+
+            row: dict[str, Any] = {
+                "example": example.toDict(),
+                "prediction": prediction_dict,
+                "is_correct": is_correct,
+            }
+            _rows.append(row)
+
+        return _rows
+
+    def _log_predictions_table(self, rows: list[dict[str, Any]]) -> None:
+        """Log a W&B Table of predictions for the current evaluation step.
+
+        Args:
+            rows (list[dict[str, Any]]): Prediction rows to log.
+        """
+        rows = _flatten_rows(rows)
+        columns = list(rows[0].keys())
+
+        data: list[list[Any]] = [list(row.values()) for row in rows]
+
+        preds_table = wandb.Table(columns=columns, data=data, log_mode="IMMUTABLE")
+        self._run.log({f"predictions_{self._row_idx}": preds_table}, step=self._row_idx)
+
+    def log_best_model(
+        self,
+        model: dspy.Module,
+        *,
+        save_program: bool = True,
+        save_dir: str | None = None,
+        filetype: Literal["json", "pkl"] = "json",
+        aliases: Sequence[str] = ("best", "latest"),
+        artifact_name: str = "dspy-program",
+    ) -> None:
+        """Save and log the best DSPy program as a W&B Artifact.
+
+        You can choose to save the full program (architecture + state) or only
+        the state to a single file (JSON or pickle).
+
+        Args:
+            model (dspy.Module): DSPy module to save.
+            save_program (bool): Save full program directory if True; otherwise
+                save only the state file. Defaults to `True`.
+            save_dir (str): Directory to store program files before logging. Defaults to a
+                subdirectory `dspy_program` within the active run's files directory
+                (i.e., `wandb.run.dir`).
+            filetype (Literal["json", "pkl"]): State file format when
+                `save_program` is False. Defaults to `json`.
+            aliases (Sequence[str]): Aliases for the logged Artifact version. Defaults to `("best", "latest")`.
+            artifact_name (str): Base name for the Artifact. Defaults to `dspy-program`.
+
+        Examples:
+            Save the complete program and add aliases:
+
+            ```python
+            callback.log_best_model(
+                optimized_program, save_program=True, aliases=("best", "production")
+            )
+            ```
+
+            Save only the state as JSON:
+
+            ```python
+            callback.log_best_model(
+                optimized_program, save_program=False, filetype="json"
+            )
+            ```
+        """
+        # Derive metadata to help discoverability in the UI
+        info_dict = self._extract_program_info(model)
+        metadata = {
+            "dspy_version": getattr(dspy, "__version__", "unknown"),
+            "module_class": model.__class__.__name__,
+            **info_dict,
+        }
+        artifact = wandb.Artifact(
+            name=f"{artifact_name}-{self._run.id}",
+            type="model",
+            metadata=metadata,
+        )
+
+        # Resolve and normalize the save directory in a cross-platform way
+        if save_dir is None:
+            save_dir = os.path.join(self._run.dir, "dspy_program")
+        save_dir = os.path.normpath(save_dir)
+
+        try:
+            os.makedirs(save_dir, exist_ok=True)
+        except Exception as exc:
+            wandb.termwarn(
+                f"Could not create or access directory '{save_dir}': {exc}. Skipping artifact logging."
+            )
+            return
+        # Save per requested mode
+        if save_program:
+            model.save(save_dir, save_program=True)
+            artifact.add_dir(save_dir)
+        else:
+            filename = f"program.{filetype}"
+            file_path = os.path.join(save_dir, filename)
+            model.save(file_path, save_program=False)
+            artifact.add_file(file_path)
+
+        self._run.log_artifact(artifact, aliases=list(aliases))

wandb/integration/weave/weave.py

@@ -10,6 +10,7 @@ The integration can be disabled by setting the WANDB_DISABLE_WEAVE environment v
 
 from __future__ import annotations
 
+import importlib.util
 import os
 import sys
 import threading
@@ -21,6 +22,31 @@ _weave_init_lock = threading.Lock()
 _DISABLE_WEAVE = "WANDB_DISABLE_WEAVE"
 _WEAVE_PACKAGE_NAME = "weave"
 
+# This list is adapted from https://github.com/wandb/weave/blob/master/weave/integrations/__init__.py
+_AVAILABLE_WEAVE_INTEGRATIONS = [
+    "anthropic",
+    "autogen",
+    "cohere",
+    "crewai",
+    "dspy",
+    "google.genai",
+    "groq",
+    "huggingface_hub.inference",
+    "instructor",
+    "langchain",
+    "litellm",
+    "llama_index",
+    "mcp",
+    "mistral",
+    "notdiamond",
+    "openai",
+    "agents",
+    "smolagents",
+    "verdict",
+    "verifiers",
+    "vertexai",
+]
+
 
 def setup(entity: str | None, project: str | None) -> None:
     """Set up automatic Weave initialization for the current W&B run.
@@ -43,6 +69,7 @@ def setup(entity: str | None, project: str | None) -> None:
     # If weave is not yet imported, we can't init it from here.  Instead, we'll
     # rely on the weave library itself to detect a run and init itself.
     if _WEAVE_PACKAGE_NAME not in sys.modules:
+        _maybe_suggest_weave_installation()
         return
 
     # If weave has already been imported, initialize immediately
@@ -61,3 +88,31 @@ def setup(entity: str | None, project: str | None) -> None:
             weave.init(project_path)
         except Exception as e:
             wandb.termwarn(f"Failed to automatically initialize Weave: {e}")
+
+
+def _maybe_suggest_weave_installation() -> None:
+    """Suggest Weave installation or import if any target library is imported."""
+    imported_libs = [lib for lib in _AVAILABLE_WEAVE_INTEGRATIONS if lib in sys.modules]
+    if not imported_libs:
+        return
+
+    weave_spec = importlib.util.find_spec(_WEAVE_PACKAGE_NAME)
+    if weave_spec is None:
+        # Weave is not installed
+        msg = (
+            "Use W&B Weave for improved LLM call tracing. Install Weave with "
+            "`pip install weave` then add `import weave` to the top of your script."
+        )
+    else:
+        # Weave is installed but not imported
+        msg = (
+            "Use W&B Weave for improved LLM call tracing. Weave is installed "
+            "but not imported. Add `import weave` to the top of your script."
+        )
+
+    wandb.termlog(f"Detected [{', '.join(imported_libs)}] in use.", repeat=False)
+    wandb.termlog(msg, repeat=False)
+    wandb.termlog(
+        "For more information, check out the docs at: https://weave-docs.wandb.ai/",
+        repeat=False,
+    )

wandb/proto/v3/wandb_server_pb2.py

@@ -15,9 +15,10 @@ _sym_db = _symbol_database.Default()
 from wandb.proto import wandb_base_pb2 as wandb_dot_proto_dot_wandb__base__pb2
 from wandb.proto import wandb_internal_pb2 as wandb_dot_proto_dot_wandb__internal__pb2
 from wandb.proto import wandb_settings_pb2 as wandb_dot_proto_dot_wandb__settings__pb2
+from wandb.proto import wandb_sync_pb2 as wandb_dot_proto_dot_wandb__sync__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1ewandb/proto/wandb_server.proto\x12\x0ewandb_internal\x1a\x1cwandb/proto/wandb_base.proto\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\"k\n\x19ServerAuthenticateRequest\x12\x0f\n\x07\x61pi_key\x18\x01 \x01(\t\x12\x10\n\x08\x62\x61se_url\x18\x02 \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"w\n\x1aServerAuthenticateResponse\x12\x16\n\x0e\x64\x65\x66\x61ult_entity\x18\x01 \x01(\t\x12\x14\n\x0c\x65rror_status\x18\x02 \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"D\n\x15ServerShutdownRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x18\n\x16ServerShutdownResponse\"B\n\x13ServerStatusRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x16\n\x14ServerStatusResponse\"r\n\x17ServerInformInitRequest\x12*\n\x08settings\x18\x01 \x01(\x0b\x32\x18.wandb_internal.Settings\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1a\n\x18ServerInformInitResponse\"s\n\x18ServerInformStartRequest\x12*\n\x08settings\x18\x01 \x01(\x0b\x32\x18.wandb_internal.Settings\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1b\n\x19ServerInformStartResponse\"H\n\x19ServerInformFinishRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1c\n\x1aServerInformFinishResponse\"H\n\x19ServerInformAttachRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"u\n\x1aServerInformAttachResponse\x12*\n\x08settings\x18\x01 \x01(\x0b\x32\x18.wandb_internal.Settings\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"H\n\x19ServerInformDetachRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1c\n\x1aServerInformDetachResponse\"]\n\x1bServerInformTeardownRequest\x12\x11\n\texit_code\x18\x01 \x01(\x05\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1e\n\x1cServerInformTeardownResponse\"\xfb\x04\n\rServerRequest\x12\x12\n\nrequest_id\x18\n \x01(\t\x12\x30\n\x0erecord_publish\x18\x01 \x01(\x0b\x32\x16.wandb_internal.RecordH\x00\x12\x34\n\x12record_communicate\x18\x02 \x01(\x0b\x32\x16.wandb_internal.RecordH\x00\x12>\n\x0binform_init\x18\x03 \x01(\x0b\x32\'.wandb_internal.ServerInformInitRequestH\x00\x12\x42\n\rinform_finish\x18\x04 \x01(\x0b\x32).wandb_internal.ServerInformFinishRequestH\x00\x12\x42\n\rinform_attach\x18\x05 \x01(\x0b\x32).wandb_internal.ServerInformAttachRequestH\x00\x12\x42\n\rinform_detach\x18\x06 \x01(\x0b\x32).wandb_internal.ServerInformDetachRequestH\x00\x12\x46\n\x0finform_teardown\x18\x07 \x01(\x0b\x32+.wandb_internal.ServerInformTeardownRequestH\x00\x12@\n\x0cinform_start\x18\x08 \x01(\x0b\x32(.wandb_internal.ServerInformStartRequestH\x00\x12\x41\n\x0c\x61uthenticate\x18\t \x01(\x0b\x32).wandb_internal.ServerAuthenticateRequestH\x00\x42\x15\n\x13server_request_type\"\x91\x05\n\x0eServerResponse\x12\x12\n\nrequest_id\x18\n \x01(\t\x12\x34\n\x12result_communicate\x18\x02 \x01(\x0b\x32\x16.wandb_internal.ResultH\x00\x12H\n\x14inform_init_response\x18\x03 \x01(\x0b\x32(.wandb_internal.ServerInformInitResponseH\x00\x12L\n\x16inform_finish_response\x18\x04 \x01(\x0b\x32*.wandb_internal.ServerInformFinishResponseH\x00\x12L\n\x16inform_attach_response\x18\x05 \x01(\x0b\x32*.wandb_internal.ServerInformAttachResponseH\x00\x12L\n\x16inform_detach_response\x18\x06 \x01(\x0b\x32*.wandb_internal.ServerInformDetachResponseH\x00\x12P\n\x18inform_teardown_response\x18\x07 \x01(\x0b\x32,.wandb_internal.ServerInformTeardownResponseH\x00\x12J\n\x15inform_start_response\x18\x08 \x01(\x0b\x32).wandb_internal.ServerInformStartResponseH\x00\x12K\n\x15\x61uthenticate_response\x18\t \x01(\x0b\x32*.wandb_internal.ServerAuthenticateResponseH\x00\x42\x16\n\x14server_response_typeB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1ewandb/proto/wandb_server.proto\x12\x0ewandb_internal\x1a\x1cwandb/proto/wandb_base.proto\x1a wandb/proto/wandb_internal.proto\x1a wandb/proto/wandb_settings.proto\x1a\x1cwandb/proto/wandb_sync.proto\"k\n\x19ServerAuthenticateRequest\x12\x0f\n\x07\x61pi_key\x18\x01 \x01(\t\x12\x10\n\x08\x62\x61se_url\x18\x02 \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"w\n\x1aServerAuthenticateResponse\x12\x16\n\x0e\x64\x65\x66\x61ult_entity\x18\x01 \x01(\t\x12\x14\n\x0c\x65rror_status\x18\x02 \x01(\t\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"D\n\x15ServerShutdownRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x18\n\x16ServerShutdownResponse\"B\n\x13ServerStatusRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x16\n\x14ServerStatusResponse\"r\n\x17ServerInformInitRequest\x12*\n\x08settings\x18\x01 \x01(\x0b\x32\x18.wandb_internal.Settings\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1a\n\x18ServerInformInitResponse\"H\n\x19ServerInformFinishRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1c\n\x1aServerInformFinishResponse\"H\n\x19ServerInformAttachRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"u\n\x1aServerInformAttachResponse\x12*\n\x08settings\x18\x01 \x01(\x0b\x32\x18.wandb_internal.Settings\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"H\n\x19ServerInformDetachRequest\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1c\n\x1aServerInformDetachResponse\"]\n\x1bServerInformTeardownRequest\x12\x11\n\texit_code\x18\x01 \x01(\x05\x12+\n\x05_info\x18\xc8\x01 \x01(\x0b\x32\x1b.wandb_internal._RecordInfo\"\x1e\n\x1cServerInformTeardownResponse\"\xee\x05\n\rServerRequest\x12\x12\n\nrequest_id\x18\n \x01(\t\x12\x30\n\x0erecord_publish\x18\x01 \x01(\x0b\x32\x16.wandb_internal.RecordH\x00\x12\x34\n\x12record_communicate\x18\x02 \x01(\x0b\x32\x16.wandb_internal.RecordH\x00\x12>\n\x0binform_init\x18\x03 \x01(\x0b\x32\'.wandb_internal.ServerInformInitRequestH\x00\x12\x42\n\rinform_finish\x18\x04 \x01(\x0b\x32).wandb_internal.ServerInformFinishRequestH\x00\x12\x42\n\rinform_attach\x18\x05 \x01(\x0b\x32).wandb_internal.ServerInformAttachRequestH\x00\x12\x42\n\rinform_detach\x18\x06 \x01(\x0b\x32).wandb_internal.ServerInformDetachRequestH\x00\x12\x46\n\x0finform_teardown\x18\x07 \x01(\x0b\x32+.wandb_internal.ServerInformTeardownRequestH\x00\x12\x41\n\x0c\x61uthenticate\x18\t \x01(\x0b\x32).wandb_internal.ServerAuthenticateRequestH\x00\x12:\n\tinit_sync\x18\x0b \x01(\x0b\x32%.wandb_internal.ServerInitSyncRequestH\x00\x12\x31\n\x04sync\x18\x0c \x01(\x0b\x32!.wandb_internal.ServerSyncRequestH\x00\x12>\n\x0bsync_status\x18\r \x01(\x0b\x32\'.wandb_internal.ServerSyncStatusRequestH\x00\x42\x15\n\x13server_request_typeJ\x04\x08\x08\x10\t\"\x98\x06\n\x0eServerResponse\x12\x12\n\nrequest_id\x18\n \x01(\t\x12\x34\n\x12result_communicate\x18\x02 \x01(\x0b\x32\x16.wandb_internal.ResultH\x00\x12H\n\x14inform_init_response\x18\x03 \x01(\x0b\x32(.wandb_internal.ServerInformInitResponseH\x00\x12L\n\x16inform_finish_response\x18\x04 \x01(\x0b\x32*.wandb_internal.ServerInformFinishResponseH\x00\x12L\n\x16inform_attach_response\x18\x05 \x01(\x0b\x32*.wandb_internal.ServerInformAttachResponseH\x00\x12L\n\x16inform_detach_response\x18\x06 \x01(\x0b\x32*.wandb_internal.ServerInformDetachResponseH\x00\x12P\n\x18inform_teardown_response\x18\x07 \x01(\x0b\x32,.wandb_internal.ServerInformTeardownResponseH\x00\x12K\n\x15\x61uthenticate_response\x18\t \x01(\x0b\x32*.wandb_internal.ServerAuthenticateResponseH\x00\x12\x44\n\x12init_sync_response\x18\x0b \x01(\x0b\x32&.wandb_internal.ServerInitSyncResponseH\x00\x12;\n\rsync_response\x18\x0c \x01(\x0b\x32\".wandb_internal.ServerSyncResponseH\x00\x12H\n\x14sync_status_response\x18\r \x01(\x0b\x32(.wandb_internal.ServerSyncStatusResponseH\x00\x42\x16\n\x14server_response_typeJ\x04\x08\x08\x10\tB\x1bZ\x19\x63ore/pkg/service_go_protob\x06proto3')
 
 
 
@@ -29,8 +30,6 @@ _SERVERSTATUSREQUEST = DESCRIPTOR.message_types_by_name['ServerStatusRequest']
 _SERVERSTATUSRESPONSE = DESCRIPTOR.message_types_by_name['ServerStatusResponse']
 _SERVERINFORMINITREQUEST = DESCRIPTOR.message_types_by_name['ServerInformInitRequest']
 _SERVERINFORMINITRESPONSE = DESCRIPTOR.message_types_by_name['ServerInformInitResponse']
-_SERVERINFORMSTARTREQUEST = DESCRIPTOR.message_types_by_name['ServerInformStartRequest']
-_SERVERINFORMSTARTRESPONSE = DESCRIPTOR.message_types_by_name['ServerInformStartResponse']
 _SERVERINFORMFINISHREQUEST = DESCRIPTOR.message_types_by_name['ServerInformFinishRequest']
 _SERVERINFORMFINISHRESPONSE = DESCRIPTOR.message_types_by_name['ServerInformFinishResponse']
 _SERVERINFORMATTACHREQUEST = DESCRIPTOR.message_types_by_name['ServerInformAttachRequest']
@@ -97,20 +96,6 @@ ServerInformInitResponse = _reflection.GeneratedProtocolMessageType('ServerInfor
   })
 _sym_db.RegisterMessage(ServerInformInitResponse)
 
-ServerInformStartRequest = _reflection.GeneratedProtocolMessageType('ServerInformStartRequest', (_message.Message,), {
-  'DESCRIPTOR' : _SERVERINFORMSTARTREQUEST,
-  '__module__' : 'wandb.proto.wandb_server_pb2'
-  # @@protoc_insertion_point(class_scope:wandb_internal.ServerInformStartRequest)
-  })
-_sym_db.RegisterMessage(ServerInformStartRequest)
-
-ServerInformStartResponse = _reflection.GeneratedProtocolMessageType('ServerInformStartResponse', (_message.Message,), {
-  'DESCRIPTOR' : _SERVERINFORMSTARTRESPONSE,
-  '__module__' : 'wandb.proto.wandb_server_pb2'
-  # @@protoc_insertion_point(class_scope:wandb_internal.ServerInformStartResponse)
-  })
-_sym_db.RegisterMessage(ServerInformStartResponse)
-
 ServerInformFinishRequest = _reflection.GeneratedProtocolMessageType('ServerInformFinishRequest', (_message.Message,), {
   'DESCRIPTOR' : _SERVERINFORMFINISHREQUEST,
   '__module__' : 'wandb.proto.wandb_server_pb2'
@@ -185,44 +170,40 @@ if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
   DESCRIPTOR._serialized_options = b'Z\031core/pkg/service_go_proto'
-  _SERVERAUTHENTICATEREQUEST._serialized_start=148
-  _SERVERAUTHENTICATEREQUEST._serialized_end=255
-  _SERVERAUTHENTICATERESPONSE._serialized_start=257
-  _SERVERAUTHENTICATERESPONSE._serialized_end=376
-  _SERVERSHUTDOWNREQUEST._serialized_start=378
-  _SERVERSHUTDOWNREQUEST._serialized_end=446
-  _SERVERSHUTDOWNRESPONSE._serialized_start=448
-  _SERVERSHUTDOWNRESPONSE._serialized_end=472
-  _SERVERSTATUSREQUEST._serialized_start=474
-  _SERVERSTATUSREQUEST._serialized_end=540
-  _SERVERSTATUSRESPONSE._serialized_start=542
-  _SERVERSTATUSRESPONSE._serialized_end=564
-  _SERVERINFORMINITREQUEST._serialized_start=566
-  _SERVERINFORMINITREQUEST._serialized_end=680
-  _SERVERINFORMINITRESPONSE._serialized_start=682
-  _SERVERINFORMINITRESPONSE._serialized_end=708
-  _SERVERINFORMSTARTREQUEST._serialized_start=710
-  _SERVERINFORMSTARTREQUEST._serialized_end=825
-  _SERVERINFORMSTARTRESPONSE._serialized_start=827
-  _SERVERINFORMSTARTRESPONSE._serialized_end=854
-  _SERVERINFORMFINISHREQUEST._serialized_start=856
-  _SERVERINFORMFINISHREQUEST._serialized_end=928
-  _SERVERINFORMFINISHRESPONSE._serialized_start=930
-  _SERVERINFORMFINISHRESPONSE._serialized_end=958
-  _SERVERINFORMATTACHREQUEST._serialized_start=960
-  _SERVERINFORMATTACHREQUEST._serialized_end=1032
-  _SERVERINFORMATTACHRESPONSE._serialized_start=1034
-  _SERVERINFORMATTACHRESPONSE._serialized_end=1151
-  _SERVERINFORMDETACHREQUEST._serialized_start=1153
-  _SERVERINFORMDETACHREQUEST._serialized_end=1225
-  _SERVERINFORMDETACHRESPONSE._serialized_start=1227
-  _SERVERINFORMDETACHRESPONSE._serialized_end=1255
-  _SERVERINFORMTEARDOWNREQUEST._serialized_start=1257
-  _SERVERINFORMTEARDOWNREQUEST._serialized_end=1350
-  _SERVERINFORMTEARDOWNRESPONSE._serialized_start=1352
-  _SERVERINFORMTEARDOWNRESPONSE._serialized_end=1382
-  _SERVERREQUEST._serialized_start=1385
-  _SERVERREQUEST._serialized_end=2020
-  _SERVERRESPONSE._serialized_start=2023
-  _SERVERRESPONSE._serialized_end=2680
+  _SERVERAUTHENTICATEREQUEST._serialized_start=178
+  _SERVERAUTHENTICATEREQUEST._serialized_end=285
+  _SERVERAUTHENTICATERESPONSE._serialized_start=287
+  _SERVERAUTHENTICATERESPONSE._serialized_end=406
+  _SERVERSHUTDOWNREQUEST._serialized_start=408
+  _SERVERSHUTDOWNREQUEST._serialized_end=476
+  _SERVERSHUTDOWNRESPONSE._serialized_start=478
+  _SERVERSHUTDOWNRESPONSE._serialized_end=502
+  _SERVERSTATUSREQUEST._serialized_start=504
+  _SERVERSTATUSREQUEST._serialized_end=570
+  _SERVERSTATUSRESPONSE._serialized_start=572
+  _SERVERSTATUSRESPONSE._serialized_end=594
+  _SERVERINFORMINITREQUEST._serialized_start=596
+  _SERVERINFORMINITREQUEST._serialized_end=710
+  _SERVERINFORMINITRESPONSE._serialized_start=712
+  _SERVERINFORMINITRESPONSE._serialized_end=738
+  _SERVERINFORMFINISHREQUEST._serialized_start=740
+  _SERVERINFORMFINISHREQUEST._serialized_end=812
+  _SERVERINFORMFINISHRESPONSE._serialized_start=814
+  _SERVERINFORMFINISHRESPONSE._serialized_end=842
+  _SERVERINFORMATTACHREQUEST._serialized_start=844
+  _SERVERINFORMATTACHREQUEST._serialized_end=916
+  _SERVERINFORMATTACHRESPONSE._serialized_start=918
+  _SERVERINFORMATTACHRESPONSE._serialized_end=1035
+  _SERVERINFORMDETACHREQUEST._serialized_start=1037
+  _SERVERINFORMDETACHREQUEST._serialized_end=1109
+  _SERVERINFORMDETACHRESPONSE._serialized_start=1111
+  _SERVERINFORMDETACHRESPONSE._serialized_end=1139
+  _SERVERINFORMTEARDOWNREQUEST._serialized_start=1141
+  _SERVERINFORMTEARDOWNREQUEST._serialized_end=1234
+  _SERVERINFORMTEARDOWNRESPONSE._serialized_start=1236
+  _SERVERINFORMTEARDOWNRESPONSE._serialized_end=1266
+  _SERVERREQUEST._serialized_start=1269
+  _SERVERREQUEST._serialized_end=2019
+  _SERVERRESPONSE._serialized_start=2022
+  _SERVERRESPONSE._serialized_end=2814
 # @@protoc_insertion_point(module_scope)