github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-24 23:54:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix typing of attribute (#3999)

Browse Source
This commit is contained in:
Zander Chase 2023-05-02 15:11:23 -07:00 committed by GitHub
parent 36ee60c96c
commit 1c68cbdb28
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
langchain/agents/structured_chat/base.py
View File

@ -5,7 +5,6 @@ from pydantic import Field
from langchain.agents.agent import Agent, AgentOutputParser from langchain.agents.agent import Agent, AgentOutputParser
from langchain.agents.structured_chat.output_parser import ( from langchain.agents.structured_chat.output_parser import (
StructuredChatOutputParser,
StructuredChatOutputParserWithRetries, StructuredChatOutputParserWithRetries,
) )
from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS, PREFIX, SUFFIX from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS, PREFIX, SUFFIX
@ -23,7 +22,9 @@ from langchain.tools import BaseTool
class StructuredChatAgent(Agent): class StructuredChatAgent(Agent):
output_parser: AgentOutputParser = Field(default_factory=StructuredChatOutputParser) output_parser: AgentOutputParser = Field(
default_factory=StructuredChatOutputParserWithRetries
)
@property @property
def observation_prefix(self) -> str: def observation_prefix(self) -> str:

5
langchain/agents/structured_chat/output_parser.py
View File

@ -11,7 +11,6 @@ from langchain.agents.agent import AgentOutputParser
from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS from langchain.agents.structured_chat.prompt import FORMAT_INSTRUCTIONS
from langchain.base_language import BaseLanguageModel from langchain.base_language import BaseLanguageModel
from langchain.output_parsers import OutputFixingParser from langchain.output_parsers import OutputFixingParser
from langchain.output_parsers.pydantic import PydanticOutputParser
from langchain.schema import AgentAction, AgentFinish, OutputParserException from langchain.schema import AgentAction, AgentFinish, OutputParserException
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -43,9 +42,7 @@ class StructuredChatOutputParser(AgentOutputParser):
class StructuredChatOutputParserWithRetries(AgentOutputParser): class StructuredChatOutputParserWithRetries(AgentOutputParser):
base_parser: PydanticOutputParser = Field( base_parser: AgentOutputParser = Field(default_factory=StructuredChatOutputParser)
default_factory=StructuredChatOutputParser
)
output_fixing_parser: Optional[OutputFixingParser] = None output_fixing_parser: Optional[OutputFixingParser] = None
def get_format_instructions(self) -> str: def get_format_instructions(self) -> str: