github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-14 00:47:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Don't type chains in output_parsers (#11092)

Browse Source 
Can't use TYPE_CHECKING style imports for pydantic params because it will try to instantiate the typed object by default.
This commit is contained in:
Bagatur 2023-09-26 17:49:35 -07:00 committed by GitHub
parent 64385c4eae
commit 5514ebe859
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
libs/langchain/langchain/output_parsers/fix.py
View File

@ -1,14 +1,11 @@
from __future__ import annotations
from typing import TYPE_CHECKING, TypeVar
from typing import Any, TypeVar
from langchain.output_parsers.prompts import NAIVE_FIX_PROMPT
from langchain.schema import BaseOutputParser, BasePromptTemplate, OutputParserException
from langchain.schema.language_model import BaseLanguageModel
if TYPE_CHECKING:
from langchain.chains.llm import LLMChain
T = TypeVar("T")
@ -20,7 +17,8 @@ class OutputFixingParser(BaseOutputParser[T]):
return True
parser: BaseOutputParser[T]
retry_chain: LLMChain
# Should be an LLMChain but we want to avoid top-level imports from langchain.chains
retry_chain: Any
@classmethod
def from_llm(

11
libs/langchain/langchain/output_parsers/retry.py
View File

@ -1,6 +1,6 @@
from __future__ import annotations
from typing import TYPE_CHECKING, TypeVar
from typing import Any, TypeVar
from langchain.prompts.prompt import PromptTemplate
from langchain.schema import (
@ -11,9 +11,6 @@ from langchain.schema import (
)
from langchain.schema.language_model import BaseLanguageModel
if TYPE_CHECKING:
from langchain.chains.llm import LLMChain
NAIVE_COMPLETION_RETRY = """Prompt:
{prompt}
Completion:
@ -48,7 +45,8 @@ class RetryOutputParser(BaseOutputParser[T]):
parser: BaseOutputParser[T]
"""The parser to use to parse the output."""
retry_chain: LLMChain
# Should be an LLMChain but we want to avoid top-level imports from langchain.chains
retry_chain: Any
"""The LLMChain to use to retry the completion."""
@classmethod
@ -127,7 +125,8 @@ class RetryWithErrorOutputParser(BaseOutputParser[T]):
"""
parser: BaseOutputParser[T]
retry_chain: LLMChain
# Should be an LLMChain but we want to avoid top-level imports from langchain.chains
retry_chain: Any
@classmethod
def from_llm(