mirror of
https://github.com/hwchase17/langchain
synced 2024-11-18 09:25:54 +00:00
c0d67420e5
<!-- Thank you for contributing to LangChain! Replace this entire comment with: - Description: a description of the change, - Issue: the issue # it fixes (if applicable), - Dependencies: any dependencies required for this change, - Tag maintainer: for a quicker response, tag the relevant maintainer (see below), - Twitter handle: we announce bigger features on Twitter. If your PR gets announced and you'd like a mention, we'll gladly shout you out! Please make sure your PR is passing linting and testing before submitting. Run `make format`, `make lint` and `make test` to check this locally. See contribution guidelines for more information on how to write/run tests, lint, etc: https://github.com/hwchase17/langchain/blob/master/.github/CONTRIBUTING.md If you're adding a new integration, please include: 1. a test for the integration, preferably unit tests that do not rely on network access, 2. an example notebook showing its use. These live is docs/extras directory. If no one reviews your PR within a few days, please @-mention one of @baskaryan, @eyurtsev, @hwchase17, @rlancemartin. -->
64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
from abc import abstractmethod
|
|
from typing import List, Tuple
|
|
|
|
from langchain.schema import BaseOutputParser
|
|
|
|
from langchain_experimental.pydantic_v1 import BaseModel, Field
|
|
|
|
|
|
class Step(BaseModel):
|
|
"""Step."""
|
|
|
|
value: str
|
|
"""The value."""
|
|
|
|
|
|
class Plan(BaseModel):
|
|
"""Plan."""
|
|
|
|
steps: List[Step]
|
|
"""The steps."""
|
|
|
|
|
|
class StepResponse(BaseModel):
|
|
"""Step response."""
|
|
|
|
response: str
|
|
"""The response."""
|
|
|
|
|
|
class BaseStepContainer(BaseModel):
|
|
"""Base step container."""
|
|
|
|
@abstractmethod
|
|
def add_step(self, step: Step, step_response: StepResponse) -> None:
|
|
"""Add step and step response to the container."""
|
|
|
|
@abstractmethod
|
|
def get_final_response(self) -> str:
|
|
"""Return the final response based on steps taken."""
|
|
|
|
|
|
class ListStepContainer(BaseStepContainer):
|
|
"""List step container."""
|
|
|
|
steps: List[Tuple[Step, StepResponse]] = Field(default_factory=list)
|
|
"""The steps."""
|
|
|
|
def add_step(self, step: Step, step_response: StepResponse) -> None:
|
|
self.steps.append((step, step_response))
|
|
|
|
def get_steps(self) -> List[Tuple[Step, StepResponse]]:
|
|
return self.steps
|
|
|
|
def get_final_response(self) -> str:
|
|
return self.steps[-1][1].response
|
|
|
|
|
|
class PlanOutputParser(BaseOutputParser):
|
|
"""Plan output parser."""
|
|
|
|
@abstractmethod
|
|
def parse(self, text: str) -> Plan:
|
|
"""Parse into a plan."""
|