huggingface[patch], pinecone[patch], fireworks[patch], mistralai[patch], voyageai[patch], togetherai[path]: convert Pydantic extras to literals (#25384)

Backwards compatible change that converts pydantic extras to literals
which is consistent with pydantic 2 usage.

- fireworks
- voyage ai
- mistralai
- mistral ai
- together ai
- huggigng face
- pinecone
This commit is contained in:
Eugene Yurtsev
2024-08-14 09:55:30 -04:00
committed by GitHub
parent d00176e523
commit 5f5e8c9a60
10 changed files with 15 additions and 20 deletions

View File

@@ -7,7 +7,6 @@ from typing import Any, Iterator, List, Mapping, Optional
from langchain_core.callbacks import CallbackManagerForLLMRun
from langchain_core.language_models.llms import BaseLLM
from langchain_core.outputs import Generation, GenerationChunk, LLMResult
from langchain_core.pydantic_v1 import Extra
DEFAULT_MODEL_ID = "gpt2"
DEFAULT_TASK = "text-generation"
@@ -67,7 +66,7 @@ class HuggingFacePipeline(BaseLLM):
class Config:
"""Configuration for this pydantic object."""
extra = Extra.forbid
extra = "forbid"
@classmethod
def from_model_id(