From a01f5476947b857fd0577b680ef32480beacef91 Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 16 Nov 2023 11:16:46 -0500 Subject: [PATCH] Update examples (#239) * Additional example updates -- remove configurable key (no longer required) * Use pydantic v1 compatibility layer correctly --- examples/configurable_chain/server.py | 5 +---- examples/conversational_retrieval_chain/server.py | 2 +- 2 files changed, 2 insertions(+), 5 deletions(-) diff --git a/examples/configurable_chain/server.py b/examples/configurable_chain/server.py index 89d0516b..1050fff1 100755 --- a/examples/configurable_chain/server.py +++ b/examples/configurable_chain/server.py @@ -59,9 +59,7 @@ ) chain = prompt | model | StrOutputParser() -# Add routes requires you to specify which config keys are accepted -# specifically, you must accept `configurable` as a config key. -add_routes(app, chain, path="/configurable_temp", config_keys=["configurable"]) +add_routes(app, chain, path="/configurable_temp") ############################################################################### @@ -96,7 +94,6 @@ def fetch_api_key_from_header(config: Dict[str, Any], req: Request) -> Dict[str, app, dynamic_auth_chain, path="/auth_from_header", - config_keys=["configurable"], per_req_config_modifier=fetch_api_key_from_header, ) diff --git a/examples/conversational_retrieval_chain/server.py b/examples/conversational_retrieval_chain/server.py index 57428ba9..66c20a21 100755 --- a/examples/conversational_retrieval_chain/server.py +++ b/examples/conversational_retrieval_chain/server.py @@ -21,9 +21,9 @@ from langchain.schema.output_parser import StrOutputParser from langchain.schema.runnable import RunnableMap, RunnablePassthrough from langchain.vectorstores import FAISS -from pydantic.v1 import BaseModel, Field from langserve import add_routes +from langserve.pydantic_v1 import BaseModel, Field _TEMPLATE = """Given the following conversation and a follow up question, rephrase the follow up question to be a standalone question, in its original language.