mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-12 22:28:03 +00:00
c2a3021bb0
Signed-off-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com> Co-authored-by: Bagatur <22008038+baskaryan@users.noreply.github.com> Co-authored-by: Dan O'Donovan <dan.odonovan@gmail.com> Co-authored-by: Tom Daniel Grande <tomdgrande@gmail.com> Co-authored-by: Grande <Tom.Daniel.Grande@statsbygg.no> Co-authored-by: Bagatur <baskaryan@gmail.com> Co-authored-by: ccurme <chester.curme@gmail.com> Co-authored-by: Harrison Chase <hw.chase.17@gmail.com> Co-authored-by: Tomaz Bratanic <bratanic.tomaz@gmail.com> Co-authored-by: ZhangShenao <15201440436@163.com> Co-authored-by: Friso H. Kingma <fhkingma@gmail.com> Co-authored-by: ChengZi <chen.zhang@zilliz.com> Co-authored-by: Nuno Campos <nuno@langchain.dev> Co-authored-by: Morgante Pell <morgantep@google.com>
63 lines
1.4 KiB
Python
63 lines
1.4 KiB
Python
from abc import abstractmethod
|
|
from typing import List, Tuple
|
|
|
|
from langchain_core.output_parsers import BaseOutputParser
|
|
from pydantic 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):
|
|
"""Container for List of steps."""
|
|
|
|
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."""
|