From 477af9e6c47b9f0f660e332c6a181df7f5725f3e Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Mon, 27 Nov 2023 16:52:03 -0500 Subject: [PATCH] Update widget (#261) Update the widget to use "output" key --- examples/conversational_retrieval_chain/server.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/examples/conversational_retrieval_chain/server.py b/examples/conversational_retrieval_chain/server.py index 66c20a21..d676498d 100755 --- a/examples/conversational_retrieval_chain/server.py +++ b/examples/conversational_retrieval_chain/server.py @@ -87,12 +87,14 @@ class ChatHistory(BaseModel): chat_history: List[Tuple[str, str]] = Field( ..., - extra={"widget": {"type": "chat", "input": "question", "output": "answer"}}, + extra={"widget": {"type": "chat", "input": "question", "output": "output"}}, ) question: str -conversational_qa_chain = _inputs | _context | ANSWER_PROMPT | ChatOpenAI() +conversational_qa_chain = ( + _inputs | _context | ANSWER_PROMPT | ChatOpenAI() | StrOutputParser() +) chain = conversational_qa_chain.with_types(input_type=ChatHistory) app = FastAPI(