huggingface-hub 0.35.0rc0__py3-none-any.whl → 1.0.0rc0__py3-none-any.whl

huggingface_hub/inference/_generated/types/automatic_speech_recognition.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List, Literal, Optional, Union
+from typing import Literal, Optional, Union
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -97,7 +97,7 @@ class AutomaticSpeechRecognitionInput(BaseInferenceType):
 class AutomaticSpeechRecognitionOutputChunk(BaseInferenceType):
     text: str
     """A chunk of text identified by the model"""
-    timestamp: List[float]
+    timestamp: list[float]
     """The start and end timestamps corresponding with the text"""
 
 
@@ -107,7 +107,7 @@ class AutomaticSpeechRecognitionOutput(BaseInferenceType):
 
     text: str
     """The recognized text."""
-    chunks: Optional[List[AutomaticSpeechRecognitionOutputChunk]] = None
+    chunks: Optional[list[AutomaticSpeechRecognitionOutputChunk]] = None
     """When returnTimestamps is enabled, chunks contains a list of audio chunks identified by
     the model.
     """

huggingface_hub/inference/_generated/types/base.py

@@ -15,8 +15,9 @@
 
 import inspect
 import json
+import types
 from dataclasses import asdict, dataclass
-from typing import Any, Dict, List, Type, TypeVar, Union, get_args
+from typing import Any, TypeVar, Union, get_args
 
 
 T = TypeVar("T", bound="BaseInferenceType")
@@ -28,7 +29,7 @@ def _repr_with_extra(self):
     return f"{self.__class__.__name__}({', '.join(f'{k}={self.__dict__[k]!r}' for k in fields + other_fields)})"
 
 
-def dataclass_with_extra(cls: Type[T]) -> Type[T]:
+def dataclass_with_extra(cls: type[T]) -> type[T]:
     """Decorator to add a custom __repr__ method to a dataclass, showing all fields, including extra ones.
 
     This decorator only works with dataclasses that inherit from `BaseInferenceType`.
@@ -49,7 +50,7 @@ class BaseInferenceType(dict):
     """
 
     @classmethod
-    def parse_obj_as_list(cls: Type[T], data: Union[bytes, str, List, Dict]) -> List[T]:
+    def parse_obj_as_list(cls: type[T], data: Union[bytes, str, list, dict]) -> list[T]:
         """Alias to parse server response and return a single instance.
 
         See `parse_obj` for more details.
@@ -60,7 +61,7 @@ class BaseInferenceType(dict):
         return output
 
     @classmethod
-    def parse_obj_as_instance(cls: Type[T], data: Union[bytes, str, List, Dict]) -> T:
+    def parse_obj_as_instance(cls: type[T], data: Union[bytes, str, list, dict]) -> T:
         """Alias to parse server response and return a single instance.
 
         See `parse_obj` for more details.
@@ -71,7 +72,7 @@ class BaseInferenceType(dict):
         return output
 
     @classmethod
-    def parse_obj(cls: Type[T], data: Union[bytes, str, List, Dict]) -> Union[List[T], T]:
+    def parse_obj(cls: type[T], data: Union[bytes, str, list, dict]) -> Union[list[T], T]:
         """Parse server response as a dataclass or list of dataclasses.
 
         To enable future-compatibility, we want to handle cases where the server return more fields than expected.
@@ -85,7 +86,7 @@ class BaseInferenceType(dict):
             data = json.loads(data)
 
         # If a list, parse each item individually
-        if isinstance(data, List):
+        if isinstance(data, list):
             return [cls.parse_obj(d) for d in data]  # type: ignore [misc]
 
         # At this point, we expect a dict
@@ -109,7 +110,9 @@ class BaseInferenceType(dict):
                     else:
                         expected_types = get_args(field_type)
                         for expected_type in expected_types:
-                            if getattr(expected_type, "_name", None) == "List":
+                            if (
+                                isinstance(expected_type, types.GenericAlias) and expected_type.__origin__ is list
+                            ) or getattr(expected_type, "_name", None) == "List":
                                 expected_type = get_args(expected_type)[
                                     0
                                 ]  # assume same type for all items in the list

huggingface_hub/inference/_generated/types/chat_completion.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, List, Literal, Optional, Union
+from typing import Any, Literal, Optional, Union
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -40,9 +40,9 @@ class ChatCompletionInputToolCall(BaseInferenceType):
 @dataclass_with_extra
 class ChatCompletionInputMessage(BaseInferenceType):
     role: str
-    content: Optional[Union[List[ChatCompletionInputMessageChunk], str]] = None
+    content: Optional[Union[list[ChatCompletionInputMessageChunk], str]] = None
     name: Optional[str] = None
-    tool_calls: Optional[List[ChatCompletionInputToolCall]] = None
+    tool_calls: Optional[list[ChatCompletionInputToolCall]] = None
 
 
 @dataclass_with_extra
@@ -56,7 +56,7 @@ class ChatCompletionInputJSONSchema(BaseInferenceType):
     A description of what the response format is for, used by the model to determine
     how to respond in the format.
     """
-    schema: Optional[Dict[str, object]] = None
+    schema: Optional[dict[str, object]] = None
     """
     The schema for the response format, described as a JSON Schema object. Learn how
     to build JSON schemas [here](https://json-schema.org/).
@@ -129,14 +129,14 @@ class ChatCompletionInput(BaseInferenceType):
     https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts.
     """
 
-    messages: List[ChatCompletionInputMessage]
+    messages: list[ChatCompletionInputMessage]
     """A list of messages comprising the conversation so far."""
     frequency_penalty: Optional[float] = None
     """Number between -2.0 and 2.0. Positive values penalize new tokens based on their existing
     frequency in the text so far,
     decreasing the model's likelihood to repeat the same line verbatim.
     """
-    logit_bias: Optional[List[float]] = None
+    logit_bias: Optional[list[float]] = None
     """UNUSED
     Modify the likelihood of specified tokens appearing in the completion. Accepts a JSON
     object that maps tokens
@@ -172,7 +172,7 @@ class ChatCompletionInput(BaseInferenceType):
     """
     response_format: Optional[ChatCompletionInputGrammarType] = None
     seed: Optional[int] = None
-    stop: Optional[List[str]] = None
+    stop: Optional[list[str]] = None
     """Up to 4 sequences where the API will stop generating further tokens."""
     stream: Optional[bool] = None
     stream_options: Optional[ChatCompletionInputStreamOptions] = None
@@ -185,7 +185,7 @@ class ChatCompletionInput(BaseInferenceType):
     tool_choice: Optional[Union[ChatCompletionInputToolChoiceClass, "ChatCompletionInputToolChoiceEnum"]] = None
     tool_prompt: Optional[str] = None
     """A prompt to be appended before the tools"""
-    tools: Optional[List[ChatCompletionInputTool]] = None
+    tools: Optional[list[ChatCompletionInputTool]] = None
     """A list of tools the model may call. Currently, only functions are supported as a tool.
     Use this to provide a list of
     functions the model may generate JSON inputs for.
@@ -213,12 +213,12 @@ class ChatCompletionOutputTopLogprob(BaseInferenceType):
 class ChatCompletionOutputLogprob(BaseInferenceType):
     logprob: float
     token: str
-    top_logprobs: List[ChatCompletionOutputTopLogprob]
+    top_logprobs: list[ChatCompletionOutputTopLogprob]
 
 
 @dataclass_with_extra
 class ChatCompletionOutputLogprobs(BaseInferenceType):
-    content: List[ChatCompletionOutputLogprob]
+    content: list[ChatCompletionOutputLogprob]
 
 
 @dataclass_with_extra
@@ -239,8 +239,9 @@ class ChatCompletionOutputToolCall(BaseInferenceType):
 class ChatCompletionOutputMessage(BaseInferenceType):
     role: str
     content: Optional[str] = None
+    reasoning: Optional[str] = None
     tool_call_id: Optional[str] = None
-    tool_calls: Optional[List[ChatCompletionOutputToolCall]] = None
+    tool_calls: Optional[list[ChatCompletionOutputToolCall]] = None
 
 
 @dataclass_with_extra
@@ -266,7 +267,7 @@ class ChatCompletionOutput(BaseInferenceType):
     https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts.
     """
 
-    choices: List[ChatCompletionOutputComplete]
+    choices: list[ChatCompletionOutputComplete]
     created: int
     id: str
     model: str
@@ -292,8 +293,9 @@ class ChatCompletionStreamOutputDeltaToolCall(BaseInferenceType):
 class ChatCompletionStreamOutputDelta(BaseInferenceType):
     role: str
     content: Optional[str] = None
+    reasoning: Optional[str] = None
     tool_call_id: Optional[str] = None
-    tool_calls: Optional[List[ChatCompletionStreamOutputDeltaToolCall]] = None
+    tool_calls: Optional[list[ChatCompletionStreamOutputDeltaToolCall]] = None
 
 
 @dataclass_with_extra
@@ -306,12 +308,12 @@ class ChatCompletionStreamOutputTopLogprob(BaseInferenceType):
 class ChatCompletionStreamOutputLogprob(BaseInferenceType):
     logprob: float
     token: str
-    top_logprobs: List[ChatCompletionStreamOutputTopLogprob]
+    top_logprobs: list[ChatCompletionStreamOutputTopLogprob]
 
 
 @dataclass_with_extra
 class ChatCompletionStreamOutputLogprobs(BaseInferenceType):
-    content: List[ChatCompletionStreamOutputLogprob]
+    content: list[ChatCompletionStreamOutputLogprob]
 
 
 @dataclass_with_extra
@@ -337,7 +339,7 @@ class ChatCompletionStreamOutput(BaseInferenceType):
     https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tgi-import.ts.
     """
 
-    choices: List[ChatCompletionStreamOutputChoice]
+    choices: list[ChatCompletionStreamOutputChoice]
     created: int
     id: str
     model: str

huggingface_hub/inference/_generated/types/depth_estimation.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, Optional
+from typing import Any, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -14,7 +14,7 @@ class DepthEstimationInput(BaseInferenceType):
 
     inputs: Any
     """The input image data"""
-    parameters: Optional[Dict[str, Any]] = None
+    parameters: Optional[dict[str, Any]] = None
     """Additional inference parameters for Depth Estimation"""
 
 

huggingface_hub/inference/_generated/types/document_question_answering.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, List, Optional, Union
+from typing import Any, Optional, Union
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -46,7 +46,7 @@ class DocumentQuestionAnsweringParameters(BaseInferenceType):
     """The number of answers to return (will be chosen by order of likelihood). Can return less
     than top_k answers if there are not enough options available within the context.
     """
-    word_boxes: Optional[List[Union[List[float], str]]] = None
+    word_boxes: Optional[list[Union[list[float], str]]] = None
     """A list of words and bounding boxes (normalized 0->1000). If provided, the inference will
     skip the OCR step and use the provided bounding boxes instead.
     """

huggingface_hub/inference/_generated/types/feature_extraction.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List, Literal, Optional, Union
+from typing import Literal, Optional, Union
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -19,7 +19,7 @@ class FeatureExtractionInput(BaseInferenceType):
     https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-tei-import.ts.
     """
 
-    inputs: Union[List[str], str]
+    inputs: Union[list[str], str]
     """The text or list of texts to embed."""
     normalize: Optional[bool] = None
     prompt_name: Optional[str] = None

huggingface_hub/inference/_generated/types/fill_mask.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, List, Optional
+from typing import Any, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -12,7 +12,7 @@ from .base import BaseInferenceType, dataclass_with_extra
 class FillMaskParameters(BaseInferenceType):
     """Additional inference parameters for Fill Mask"""
 
-    targets: Optional[List[str]] = None
+    targets: Optional[list[str]] = None
     """When passed, the model will limit the scores to the passed targets instead of looking up
     in the whole vocabulary. If the provided targets are not in the model vocab, they will be
     tokenized and the first resulting token will be used (with a warning, and that might be

huggingface_hub/inference/_generated/types/sentence_similarity.py

@@ -3,14 +3,14 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, List, Optional
+from typing import Any, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
 
 @dataclass_with_extra
 class SentenceSimilarityInputData(BaseInferenceType):
-    sentences: List[str]
+    sentences: list[str]
     """A list of strings which will be compared against the source_sentence."""
     source_sentence: str
     """The string that you wish to compare the other strings with. This can be a phrase,
@@ -23,5 +23,5 @@ class SentenceSimilarityInput(BaseInferenceType):
     """Inputs for Sentence similarity inference"""
 
     inputs: SentenceSimilarityInputData
-    parameters: Optional[Dict[str, Any]] = None
+    parameters: Optional[dict[str, Any]] = None
     """Additional inference parameters for Sentence Similarity"""

huggingface_hub/inference/_generated/types/summarization.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, Literal, Optional
+from typing import Any, Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -17,7 +17,7 @@ class SummarizationParameters(BaseInferenceType):
 
     clean_up_tokenization_spaces: Optional[bool] = None
     """Whether to clean up the potential extra spaces in the text output."""
-    generate_parameters: Optional[Dict[str, Any]] = None
+    generate_parameters: Optional[dict[str, Any]] = None
     """Additional parametrization of the text generation algorithm."""
     truncation: Optional["SummarizationTruncationStrategy"] = None
     """The truncation strategy to use."""

huggingface_hub/inference/_generated/types/table_question_answering.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Dict, List, Literal, Optional
+from typing import Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -14,7 +14,7 @@ class TableQuestionAnsweringInputData(BaseInferenceType):
 
     question: str
     """The question to be answered about the table"""
-    table: Dict[str, List[str]]
+    table: dict[str, list[str]]
     """The table to serve as context for the questions"""
 
 
@@ -54,9 +54,9 @@ class TableQuestionAnsweringOutputElement(BaseInferenceType):
     """The answer of the question given the table. If there is an aggregator, the answer will be
     preceded by `AGGREGATOR >`.
     """
-    cells: List[str]
+    cells: list[str]
     """List of strings made up of the answer cell values."""
-    coordinates: List[List[int]]
+    coordinates: list[list[int]]
     """Coordinates of the cells of the answers."""
     aggregator: Optional[str] = None
     """If the model has an aggregator, this returns the aggregator."""

huggingface_hub/inference/_generated/types/text2text_generation.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, Literal, Optional
+from typing import Any, Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -17,7 +17,7 @@ class Text2TextGenerationParameters(BaseInferenceType):
 
     clean_up_tokenization_spaces: Optional[bool] = None
     """Whether to clean up the potential extra spaces in the text output."""
-    generate_parameters: Optional[Dict[str, Any]] = None
+    generate_parameters: Optional[dict[str, Any]] = None
     """Additional parametrization of the text generation algorithm"""
     truncation: Optional["Text2TextGenerationTruncationStrategy"] = None
     """The truncation strategy to use"""

huggingface_hub/inference/_generated/types/text_generation.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, List, Literal, Optional
+from typing import Any, Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -49,7 +49,7 @@ class TextGenerationInputGenerateParameters(BaseInferenceType):
     """Whether to prepend the prompt to the generated text"""
     seed: Optional[int] = None
     """Random sampling seed."""
-    stop: Optional[List[str]] = None
+    stop: Optional[list[str]] = None
     """Stop generating tokens if a member of `stop` is generated."""
     temperature: Optional[float] = None
     """The value used to module the logits distribution."""
@@ -108,21 +108,21 @@ class TextGenerationOutputBestOfSequence(BaseInferenceType):
     finish_reason: "TextGenerationOutputFinishReason"
     generated_text: str
     generated_tokens: int
-    prefill: List[TextGenerationOutputPrefillToken]
-    tokens: List[TextGenerationOutputToken]
+    prefill: list[TextGenerationOutputPrefillToken]
+    tokens: list[TextGenerationOutputToken]
     seed: Optional[int] = None
-    top_tokens: Optional[List[List[TextGenerationOutputToken]]] = None
+    top_tokens: Optional[list[list[TextGenerationOutputToken]]] = None
 
 
 @dataclass_with_extra
 class TextGenerationOutputDetails(BaseInferenceType):
     finish_reason: "TextGenerationOutputFinishReason"
     generated_tokens: int
-    prefill: List[TextGenerationOutputPrefillToken]
-    tokens: List[TextGenerationOutputToken]
-    best_of_sequences: Optional[List[TextGenerationOutputBestOfSequence]] = None
+    prefill: list[TextGenerationOutputPrefillToken]
+    tokens: list[TextGenerationOutputToken]
+    best_of_sequences: Optional[list[TextGenerationOutputBestOfSequence]] = None
     seed: Optional[int] = None
-    top_tokens: Optional[List[List[TextGenerationOutputToken]]] = None
+    top_tokens: Optional[list[list[TextGenerationOutputToken]]] = None
 
 
 @dataclass_with_extra
@@ -165,4 +165,4 @@ class TextGenerationStreamOutput(BaseInferenceType):
     token: TextGenerationStreamOutputToken
     details: Optional[TextGenerationStreamOutputStreamDetails] = None
     generated_text: Optional[str] = None
-    top_tokens: Optional[List[TextGenerationStreamOutputToken]] = None
+    top_tokens: Optional[list[TextGenerationStreamOutputToken]] = None

huggingface_hub/inference/_generated/types/text_to_video.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, List, Optional
+from typing import Any, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -16,7 +16,7 @@ class TextToVideoParameters(BaseInferenceType):
     """A higher guidance scale value encourages the model to generate videos closely linked to
     the text prompt, but values too high may cause saturation and other artifacts.
     """
-    negative_prompt: Optional[List[str]] = None
+    negative_prompt: Optional[list[str]] = None
     """One or several prompt to guide what NOT to include in video generation."""
     num_frames: Optional[float] = None
     """The num_frames parameter determines how many video frames are generated."""

huggingface_hub/inference/_generated/types/token_classification.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List, Literal, Optional
+from typing import Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -17,7 +17,7 @@ class TokenClassificationParameters(BaseInferenceType):
 
     aggregation_strategy: Optional["TokenClassificationAggregationStrategy"] = None
     """The strategy used to fuse tokens based on model predictions"""
-    ignore_labels: Optional[List[str]] = None
+    ignore_labels: Optional[list[str]] = None
     """A list of labels to ignore"""
     stride: Optional[int] = None
     """The number of overlapping tokens between chunks when splitting the input text."""

huggingface_hub/inference/_generated/types/translation.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import Any, Dict, Literal, Optional
+from typing import Any, Literal, Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -17,7 +17,7 @@ class TranslationParameters(BaseInferenceType):
 
     clean_up_tokenization_spaces: Optional[bool] = None
     """Whether to clean up the potential extra spaces in the text output."""
-    generate_parameters: Optional[Dict[str, Any]] = None
+    generate_parameters: Optional[dict[str, Any]] = None
     """Additional parametrization of the text generation algorithm."""
     src_lang: Optional[str] = None
     """The source language of the text. Required for models that can translate from multiple

huggingface_hub/inference/_generated/types/zero_shot_classification.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List, Optional
+from typing import Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -12,7 +12,7 @@ from .base import BaseInferenceType, dataclass_with_extra
 class ZeroShotClassificationParameters(BaseInferenceType):
     """Additional inference parameters for Zero Shot Classification"""
 
-    candidate_labels: List[str]
+    candidate_labels: list[str]
     """The set of possible class labels to classify the text into."""
     hypothesis_template: Optional[str] = None
     """The sentence used in conjunction with `candidate_labels` to attempt the text

huggingface_hub/inference/_generated/types/zero_shot_image_classification.py

@@ -3,7 +3,7 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List, Optional
+from typing import Optional
 
 from .base import BaseInferenceType, dataclass_with_extra
 
@@ -12,7 +12,7 @@ from .base import BaseInferenceType, dataclass_with_extra
 class ZeroShotImageClassificationParameters(BaseInferenceType):
     """Additional inference parameters for Zero Shot Image Classification"""
 
-    candidate_labels: List[str]
+    candidate_labels: list[str]
     """The candidate labels for this image"""
     hypothesis_template: Optional[str] = None
     """The sentence used in conjunction with `candidate_labels` to attempt the image

huggingface_hub/inference/_generated/types/zero_shot_object_detection.py

@@ -3,8 +3,6 @@
 # See:
 #   - script: https://github.com/huggingface/huggingface.js/blob/main/packages/tasks/scripts/inference-codegen.ts
 #   - specs:  https://github.com/huggingface/huggingface.js/tree/main/packages/tasks/src/tasks.
-from typing import List
-
 from .base import BaseInferenceType, dataclass_with_extra
 
 
@@ -12,7 +10,7 @@ from .base import BaseInferenceType, dataclass_with_extra
 class ZeroShotObjectDetectionParameters(BaseInferenceType):
     """Additional inference parameters for Zero Shot Object Detection"""
 
-    candidate_labels: List[str]
+    candidate_labels: list[str]
     """The candidate labels for this image"""
 
 

huggingface_hub/inference/_mcp/_cli_hacks.py

@@ -17,7 +17,7 @@ def _patch_anyio_open_process():
 
     if getattr(anyio, "_tiny_agents_patched", False):
         return
-    anyio._tiny_agents_patched = True
+    anyio._tiny_agents_patched = True  # ty: ignore[invalid-assignment]
 
     original_open_process = anyio.open_process
 
@@ -32,7 +32,7 @@ def _patch_anyio_open_process():
             kwargs.setdefault("creationflags", subprocess.CREATE_NEW_PROCESS_GROUP)
             return await original_open_process(*args, **kwargs)
 
-        anyio.open_process = open_process_in_new_group
+        anyio.open_process = open_process_in_new_group  # ty: ignore[invalid-assignment]
     else:
         # For Unix-like systems, we can use setsid to create a new session
         async def open_process_in_new_group(*args, **kwargs):
@@ -42,7 +42,7 @@ def _patch_anyio_open_process():
             kwargs.setdefault("start_new_session", True)
             return await original_open_process(*args, **kwargs)
 
-        anyio.open_process = open_process_in_new_group
+        anyio.open_process = open_process_in_new_group  # ty: ignore[invalid-assignment]
 
 
 async def _async_prompt(exit_event: asyncio.Event, prompt: str = "» ") -> str:

huggingface_hub/inference/_mcp/agent.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 import asyncio
-from typing import AsyncGenerator, Dict, Iterable, List, Optional, Union
+from typing import AsyncGenerator, Iterable, Optional, Union
 
 from huggingface_hub import ChatCompletionInputMessage, ChatCompletionStreamOutput, MCPClient
 
@@ -24,7 +24,7 @@ class Agent(MCPClient):
         model (`str`, *optional*):
             The model to run inference with. Can be a model id hosted on the Hugging Face Hub, e.g. `meta-llama/Meta-Llama-3-8B-Instruct`
             or a URL to a deployed Inference Endpoint or other local or remote endpoint.
-        servers (`Iterable[Dict]`):
+        servers (`Iterable[dict]`):
             MCP servers to connect to. Each server is a dictionary containing a `type` key and a `config` key. The `type` key can be `"stdio"` or `"sse"`, and the `config` key is a dictionary of arguments for the server.
         provider (`str`, *optional*):
             Name of the provider to use for inference. Defaults to "auto" i.e. the first of the providers available for the model, sorted by the user's order in https://hf.co/settings/inference-providers.
@@ -49,7 +49,7 @@ class Agent(MCPClient):
     ):
         super().__init__(model=model, provider=provider, base_url=base_url, api_key=api_key)
         self._servers_cfg = list(servers)
-        self.messages: List[Union[Dict, ChatCompletionInputMessage]] = [
+        self.messages: list[Union[dict, ChatCompletionInputMessage]] = [
             {"role": "system", "content": prompt or DEFAULT_SYSTEM_PROMPT}
         ]
 

huggingface_hub/inference/_mcp/cli.py

@@ -33,7 +33,7 @@ async def run_agent(
 
     Args:
         agent_path (`str`, *optional*):
-            Path to a local folder containing an `agent.json` and optionally a custom `PROMPT.md` file or a built-in agent stored in a Hugging Face dataset.
+            Path to a local folder containing an `agent.json` and optionally a custom `PROMPT.md` or `AGENTS.md` file or a built-in agent stored in a Hugging Face dataset.
 
     """
     _patch_anyio_open_process()  # Hacky way to prevent stdio connections to be stopped by Ctrl+C

huggingface_hub/inference/_mcp/constants.py

@@ -2,13 +2,12 @@ from __future__ import annotations
 
 import sys
 from pathlib import Path
-from typing import List
 
 from huggingface_hub import ChatCompletionInputTool
 
 
 FILENAME_CONFIG = "agent.json"
-FILENAME_PROMPT = "PROMPT.md"
+PROMPT_FILENAMES = ("PROMPT.md", "AGENTS.md")
 
 DEFAULT_AGENT = {
     "model": "Qwen/Qwen2.5-72B-Instruct",
@@ -76,7 +75,7 @@ ASK_QUESTION_TOOL: ChatCompletionInputTool = ChatCompletionInputTool.parse_obj(
     }
 )
 
-EXIT_LOOP_TOOLS: List[ChatCompletionInputTool] = [TASK_COMPLETE_TOOL, ASK_QUESTION_TOOL]
+EXIT_LOOP_TOOLS: list[ChatCompletionInputTool] = [TASK_COMPLETE_TOOL, ASK_QUESTION_TOOL]
 
 
 DEFAULT_REPO_ID = "tiny-agents/tiny-agents"