huggingface-hub 0.33.1__tar.gz → 0.33.2__tar.gz

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface_hub
-Version: 0.33.1
+Version: 0.33.2
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub/__init__.py

@@ -46,7 +46,7 @@ import sys
 from typing import TYPE_CHECKING
 
 
-__version__ = "0.33.1"
+__version__ = "0.33.2"
 
 # Alphabetical order of definitions is ensured in tests
 # WARNING: any comment added in this dictionary definition will be lost when

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub/inference/_client.py

@@ -947,8 +947,8 @@ class InferenceClient:
         Answer questions on document images.
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The input image for the context. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The input image for the context. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             question (`str`):
                 Question to be answered.
             model (`str`, *optional*):
@@ -1156,8 +1156,8 @@ class InferenceClient:
         Perform image classification on the given image using the specified model.
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The image to classify. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The image to classify. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             model (`str`, *optional*):
                 The model to use for image classification. Can be a model ID hosted on the Hugging Face Hub or a URL to a
                 deployed Inference Endpoint. If not provided, the default recommended model for image classification will be used.
@@ -1214,8 +1214,8 @@ class InferenceClient:
         </Tip>
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The image to segment. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The image to segment. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             model (`str`, *optional*):
                 The model to use for image segmentation. Can be a model ID hosted on the Hugging Face Hub or a URL to a
                 deployed Inference Endpoint. If not provided, the default recommended model for image segmentation will be used.
@@ -1286,8 +1286,8 @@ class InferenceClient:
         </Tip>
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The input image for translation. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The input image for translation. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             prompt (`str`, *optional*):
                 The text prompt to guide the image generation.
             negative_prompt (`str`, *optional*):
@@ -1348,8 +1348,8 @@ class InferenceClient:
         (OCR), Pix2Struct, etc). Please have a look to the model card to learn more about a model's specificities.
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The input image to caption. It can be raw bytes, an image file, or a URL to an online image..
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The input image to caption. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             model (`str`, *optional*):
                 The model to use for inference. Can be a model ID hosted on the Hugging Face Hub or a URL to a deployed
                 Inference Endpoint. This parameter overrides the model defined at the instance level. Defaults to None.
@@ -1399,8 +1399,8 @@ class InferenceClient:
         </Tip>
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The image to detect objects on. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The image to detect objects on. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             model (`str`, *optional*):
                 The model to use for object detection. Can be a model ID hosted on the Hugging Face Hub or a URL to a
                 deployed Inference Endpoint. If not provided, the default recommended model for object detection (DETR) will be used.
@@ -2974,8 +2974,8 @@ class InferenceClient:
         Answering open-ended questions based on an image.
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The input image for the context. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The input image for the context. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             question (`str`):
                 Question to be answered.
             model (`str`, *optional*):
@@ -3141,8 +3141,8 @@ class InferenceClient:
         Provide input image and text labels to predict text labels for the image.
 
         Args:
-            image (`Union[str, Path, bytes, BinaryIO]`):
-                The input image to caption. It can be raw bytes, an image file, or a URL to an online image.
+            image (`Union[str, Path, bytes, BinaryIO, PIL.Image.Image]`):
+                The input image to caption. It can be raw bytes, an image file, a URL to an online image, or a PIL Image.
             candidate_labels (`List[str]`):
                 The candidate labels for this image
             labels (`List[str]`, *optional*):

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub/inference/_common.py

@@ -61,7 +61,7 @@ if TYPE_CHECKING:
 UrlT = str
 PathT = Union[str, Path]
 BinaryT = Union[bytes, BinaryIO]
-ContentT = Union[BinaryT, PathT, UrlT]
+ContentT = Union[BinaryT, PathT, UrlT, "Image"]
 
 # Use to set a Accept: image/png header
 TASKS_EXPECTING_IMAGES = {"text-to-image", "image-to-image"}
@@ -164,7 +164,6 @@ def _open_as_binary(content: Optional[ContentT]) -> Generator[Optional[BinaryT],
 
     Do nothing if `content` is None,
 
-    TODO: handle a PIL.Image as input
     TODO: handle base64 as input
     """
     # If content is a string => must be either a URL or a path
@@ -185,6 +184,12 @@ def _open_as_binary(content: Optional[ContentT]) -> Generator[Optional[BinaryT],
         logger.debug(f"Opening content from {content}")
         with content.open("rb") as f:
             yield f
+    elif hasattr(content, "save"):  # PIL Image
+        logger.debug("Converting PIL Image to bytes")
+        buffer = io.BytesIO()
+        content.save(buffer, format="PNG")
+        buffer.seek(0)
+        yield buffer
     else:
         # Otherwise: already a file-like object or None
         yield content

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub/inference/_mcp/agent.py

@@ -7,6 +7,7 @@ from huggingface_hub import ChatCompletionInputMessage, ChatCompletionStreamOutp
 
 from .._providers import PROVIDER_OR_POLICY_T
 from .constants import DEFAULT_SYSTEM_PROMPT, EXIT_LOOP_TOOLS, MAX_NUM_TURNS
+from .types import ServerConfig
 
 
 class Agent(MCPClient):
@@ -40,7 +41,7 @@ class Agent(MCPClient):
         self,
         *,
         model: Optional[str] = None,
-        servers: Iterable[Dict],
+        servers: Iterable[ServerConfig],
         provider: Optional[PROVIDER_OR_POLICY_T] = None,
         base_url: Optional[str] = None,
         api_key: Optional[str] = None,
@@ -54,7 +55,7 @@ class Agent(MCPClient):
 
     async def load_tools(self) -> None:
         for cfg in self._servers_cfg:
-            await self.add_mcp_server(cfg["type"], **cfg["config"])
+            await self.add_mcp_server(**cfg)
 
     async def run(
         self,

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub/inference/_mcp/cli.py

@@ -85,11 +85,7 @@ async def run_agent(
                 input_vars = set()
                 for server in servers:
                     # Check stdio's "env" and http/sse's "headers" mappings
-                    env_or_headers = (
-                        server["config"].get("env", {})
-                        if server["type"] == "stdio"
-                        else server["config"].get("options", {}).get("requestInit", {}).get("headers", {})
-                    )
+                    env_or_headers = server.get("env", {}) if server["type"] == "stdio" else server.get("headers", {})
                     for key, value in env_or_headers.items():
                         if env_special_value in value:
                             input_vars.add(key)
@@ -99,8 +95,9 @@ async def run_agent(
                     continue
 
                 # Prompt user for input
+                env_variable_key = input_id.replace("-", "_").upper()
                 print(
-                    f"[blue] • {input_id}[/blue]: {description}. (default: load from {', '.join(sorted(input_vars))}).",
+                    f"[blue] • {input_id}[/blue]: {description}. (default: load from {env_variable_key}).",
                     end=" ",
                 )
                 user_input = (await _async_prompt(exit_event=exit_event)).strip()
@@ -109,23 +106,19 @@ async def run_agent(
 
                 # Inject user input (or env variable) into stdio's env or http/sse's headers
                 for server in servers:
-                    env_or_headers = (
-                        server["config"].get("env", {})
-                        if server["type"] == "stdio"
-                        else server["config"].get("options", {}).get("requestInit", {}).get("headers", {})
-                    )
+                    env_or_headers = server.get("env", {}) if server["type"] == "stdio" else server.get("headers", {})
                     for key, value in env_or_headers.items():
                         if env_special_value in value:
                             if user_input:
                                 env_or_headers[key] = env_or_headers[key].replace(env_special_value, user_input)
                             else:
-                                value_from_env = os.getenv(key, "")
+                                value_from_env = os.getenv(env_variable_key, "")
                                 env_or_headers[key] = env_or_headers[key].replace(env_special_value, value_from_env)
                                 if value_from_env:
-                                    print(f"[green]Value successfully loaded from '{key}'[/green]")
+                                    print(f"[green]Value successfully loaded from '{env_variable_key}'[/green]")
                                 else:
                                     print(
-                                        f"[yellow]No value found for '{key}' in environment variables. Continuing.[/yellow]"
+                                        f"[yellow]No value found for '{env_variable_key}' in environment variables. Continuing.[/yellow]"
                                     )
 
             print()

huggingface_hub-0.33.2/src/huggingface_hub/inference/_mcp/types.py

@@ -0,0 +1,39 @@
+from typing import Dict, List, Literal, TypedDict, Union
+
+
+class InputConfig(TypedDict, total=False):
+    id: str
+    description: str
+    type: str
+    password: bool
+
+
+class StdioServerConfig(TypedDict):
+    type: Literal["stdio"]
+    command: str
+    args: List[str]
+    env: Dict[str, str]
+    cwd: str
+
+
+class HTTPServerConfig(TypedDict):
+    type: Literal["http"]
+    url: str
+    headers: Dict[str, str]
+
+
+class SSEServerConfig(TypedDict):
+    type: Literal["sse"]
+    url: str
+    headers: Dict[str, str]
+
+
+ServerConfig = Union[StdioServerConfig, HTTPServerConfig, SSEServerConfig]
+
+
+# AgentConfig root object
+class AgentConfig(TypedDict):
+    model: str
+    provider: str
+    inputs: List[InputConfig]
+    servers: List[ServerConfig]

{huggingface_hub-0.33.1 → huggingface_hub-0.33.2}/src/huggingface_hub.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: huggingface-hub
-Version: 0.33.1
+Version: 0.33.2
 Summary: Client library to download and publish models, datasets and other repos on the huggingface.co hub
 Home-page: https://github.com/huggingface/huggingface_hub
 Author: Hugging Face, Inc.

huggingface_hub-0.33.1/src/huggingface_hub/inference/_mcp/types.py

@@ -1,65 +0,0 @@
-from typing import Dict, List, Literal, TypedDict, Union
-
-
-# Input config
-class InputConfig(TypedDict, total=False):
-    id: str
-    description: str
-    type: str
-    password: bool
-
-
-# stdio server config
-class StdioServerConfig(TypedDict, total=False):
-    command: str
-    args: List[str]
-    env: Dict[str, str]
-    cwd: str
-
-
-class StdioServer(TypedDict):
-    type: Literal["stdio"]
-    config: StdioServerConfig
-
-
-# http server config
-class HTTPRequestInit(TypedDict, total=False):
-    headers: Dict[str, str]
-
-
-class HTTPServerOptions(TypedDict, total=False):
-    requestInit: HTTPRequestInit
-    sessionId: str
-
-
-class HTTPServerConfig(TypedDict, total=False):
-    url: str
-    options: HTTPServerOptions
-
-
-class HTTPServer(TypedDict):
-    type: Literal["http"]
-    config: HTTPServerConfig
-
-
-# sse server config
-class SSEServerOptions(TypedDict, total=False):
-    requestInit: HTTPRequestInit
-
-
-class SSEServerConfig(TypedDict):
-    url: str
-    options: SSEServerOptions
-
-
-class SSEServer(TypedDict):
-    type: Literal["sse"]
-    config: SSEServerConfig
-
-
-# AgentConfig root object
-class AgentConfig(TypedDict):
-    model: str
-    provider: str
-    inputs: List[InputConfig]
-    servers: List[Union[StdioServer, HTTPServer, SSEServer]]