Add LangServe example (#10)

Author: ssowonny

examples/proofreading-bot-langchain/main.py

@@ -5,9 +5,9 @@
 from collections.abc import AsyncGenerator
 
 from fastapi import FastAPI
-from langchain.chat_models import ChatOpenAI
 from langchain.prompts import ChatPromptTemplate
 from langchain_core.output_parsers import StrOutputParser
+from langchain_openai import ChatOpenAI
 
 import plugbear.fastapi
 

examples/proofreading-bot-langchain/poetry.lock

@@ -10,8 +10,9 @@ python = "^3.11"
 fastapi = "^0.105.0"
 plugbear = {version = "^0.1.0", extras = ["fastapi"]}
 uvicorn = "^0.24.0.post1"
-langchain = "^0.0.353"
+langchain = "0.1.6"
 openai = "^1.6.1"
+langchain-openai = "^0.0.5"
 
 
 [tool.poetry.group.dev.dependencies]

examples/proofreading-bot-langchain/pyproject.toml

@@ -0,0 +1,41 @@
+# PlugBear Python SDK Example (LangChain)
+
+This project introduces an example of integrating a LangServe application with
+communication channels, such as Slack, using PlugBear. Check the [PlugBear Docs: LangServe](https://docs.plugbear.io/integrations/langchain/langserve) page to learn more.
+
+## Prerequisites
+
+- [Poetry](https://python-poetry.org)
+
+## Development
+
+### Installing Dependencies
+
+Use [Poetry](https://python-poetry.org/) to install dependencies.
+
+```bash
+poetry install
+```
+
+### Running Server
+
+Run the command below to run the server:
+
+```bash
+OPENAI_API_KEY="YOUR_OPENAI_API_KEY" \
+  PLUGBEAR_API_KEY="YOUR_PLUGBEAR_API_KEY" \
+  poetry run python main.py
+```
+
+You can obtain your `OPENAI_API_KEY` from the
+[OpenAI API Keys](https://platform.openai.com/api-keys) page and your
+`PLUGBEAR_API_KEY` from the
+[PlugBear API Keys](https://auth.plugbear.io/org/api_keys) page.
+
+### Testing Integration
+
+Follow [PlugBear Documentation](https://docs.plugbear.io) to connect your app
+to communication channels and test it.
+
+Ask any math question to `@PlugBear` after connecting it. e.g.,
+``@PlugBear This is an example sentence.``.

examples/proofreading-bot-langserve/README.md

@@ -0,0 +1,36 @@
+from __future__ import annotations
+
+import os
+
+from fastapi import FastAPI
+from langchain.prompts import ChatPromptTemplate, MessagesPlaceholder
+from langchain_core.messages import convert_to_messages
+from langchain_core.output_parsers import StrOutputParser
+from langchain_openai import ChatOpenAI
+from langserve import add_routes
+
+OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
+PLUGBEAR_API_KEY = os.environ["PLUGBEAR_API_KEY"]
+
+app = FastAPI()
+llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY)
+output_parser = StrOutputParser()
+prompt = ChatPromptTemplate.from_messages(
+    [
+        ("system", "You are the Proofreading Bot, an editor bot designed to proofread technical manuals with the precision and style of a professional technical writer. Your primary function is to make the text clear, concise, and professional. You avoid jargon, ambiguous expressions, and emotional language, aiming for straightforward, easy-to-understand, yet professional sentences. You specialize in improving the readability and accuracy of technical manuals, adhering to high standards of technical writing. While maintaining professionalism, your interaction style is helpful, providing guidance and suggestions to enhance the user's text. Answer the revised version of the text only. Do not add any other descriptions."),
+        MessagesPlaceholder("conversation"),
+    ]
+)
+runnable = {"conversation": convert_to_messages} | prompt | llm | output_parser
+
+add_routes(
+    app,
+    runnable,
+    path="/proofreading-bot-langserve",
+)
+
+
+if __name__ == "__main__":
+    import uvicorn
+
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", default=8000)))