Update widget (#261)

Update the widget to use "output" key
langchain-ai · Nov 27, 2023 · 477af9e · 477af9e
1 parent af904f5
commit 477af9e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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(