diff --git a/litellm/types/llms/anthropic.py b/litellm/types/llms/anthropic.py index df6580f9b3..23dd661e9a 100644 --- a/litellm/types/llms/anthropic.py +++ b/litellm/types/llms/anthropic.py @@ -1,7 +1,7 @@ from enum import Enum from typing import Any, Dict, Iterable, List, Optional, Union -from pydantic import BaseModel +from pydantic import BaseModel, ConfigDict from typing_extensions import Literal, Required, TypedDict from .openai import ( @@ -535,8 +535,7 @@ class AnthropicResponseContentBlockToolUse(BaseModel): input: dict provider_specific_fields: Optional[Dict[str, Any]] = None - class Config: - extra = "allow" # Allow provider_specific_fields + model_config = ConfigDict(extra="allow") # Allow provider_specific_fields class AnthropicResponseContentBlockThinking(BaseModel): diff --git a/litellm/types/rag.py b/litellm/types/rag.py index 7a964931af..dd724ca217 100644 --- a/litellm/types/rag.py +++ b/litellm/types/rag.py @@ -4,7 +4,7 @@ Type definitions for RAG (Retrieval Augmented Generation) Ingest API. from typing import Any, Dict, List, Literal, Optional, Union -from pydantic import BaseModel +from pydantic import BaseModel, ConfigDict from typing_extensions import TypedDict @@ -185,6 +185,5 @@ class RAGIngestRequest(BaseModel): file_id: Optional[str] = None # Existing file ID ingest_options: Dict[str, Any] # RAGIngestOptions as dict for flexibility - class Config: - extra = "allow" # Allow additional fields + model_config = ConfigDict(extra="allow") # Allow additional fields