Method _coerce_tool_argument

examples/server/server.py:6997–7074

(
        self,
        raw_value: str,
        schema: Dict[str, Any],
        *,
        tool_name: str,
        argument_name: str,
    )

        return None

    def _coerce_tool_argument(
        self,
        raw_value: str,
        schema: Dict[str, Any],
        *,
        tool_name: str,
        argument_name: str,
    ) -> Any:
        if "oneOf" in schema:
            last_error: Optional[BaseException] = None
            for variant in schema["oneOf"]:
                try:
                    return self._coerce_tool_argument(
                        raw_value,
                        variant,
                        tool_name=tool_name,
                        argument_name=argument_name,
                    )
                except BaseException as exc:
                    last_error = exc
            raise CompletionResponseParsingError(
                f"tool '{tool_name}' argument '{argument_name}' did not match any allowed schema"
            ) from last_error
        schema_type = schema.get("type")
        if isinstance(schema_type, list):
            last_type_error: Optional[BaseException] = None
            for variant_type in schema_type:
                try:
                    return self._coerce_tool_argument(
                        raw_value,
                        {**schema, "type": variant_type},
                        tool_name=tool_name,
                        argument_name=argument_name,
                    )
                except BaseException as exc:
                    last_type_error = exc
            raise CompletionResponseParsingError(
                f"tool '{tool_name}' argument '{argument_name}' did not match any allowed type"
            ) from last_type_error
        if schema_type in {None, "string"}:
            return raw_value
        try:
            decoded = json.loads(raw_value)
        except json.JSONDecodeError as exc:
            raise CompletionResponseParsingError(
                f"tool '{tool_name}' argument '{argument_name}' is not valid JSON for type '{schema_type}'"
            ) from exc
        if schema_type == "integer":
            if isinstance(decoded, bool) or not isinstance(decoded, int):
                raise CompletionResponseParsingError(
                    f"tool '{tool_name}' argument '{argument_name}' must be an integer"
                )
            return decoded
        if schema_type == "number":
            if isinstance(decoded, bool) or not isinstance(decoded, (int, float)):
                raise CompletionResponseParsingError(
                    f"tool '{tool_name}' argument '{argument_name}' must be a number"
                )