experimental[major]: upgrade pydantic (#26228)

This commit is contained in:
ccurme
2024-09-09 14:27:24 -04:00
committed by GitHub
parent 109ba548bd
commit 6c8d626d70
45 changed files with 183 additions and 170 deletions

View File

@@ -4,6 +4,7 @@ from langchain.chains.base import Chain
from langchain_core.callbacks import CallbackManagerForChainRun
from langchain_core.language_models import BaseLanguageModel
from langchain_core.prompts import PromptTemplate
from pydantic import ConfigDict
from langchain_experimental.video_captioning.services.audio_service import (
AudioProcessor,
@@ -36,9 +37,10 @@ class VideoCaptioningChain(Chain):
closed_caption_similarity_threshold: int = 80
use_unclustered_video_models: bool = False
class Config:
arbitrary_types_allowed = True
extra = "allow"
model_config = ConfigDict(
arbitrary_types_allowed=True,
extra="allow",
)
@property
def input_keys(self) -> List[str]: