Add HTTPMessangerInterface

chatsky/messengers/__init__.py

@@ -6,3 +6,4 @@
 )
 from chatsky.messengers.telegram import LongpollingInterface as TelegramInterface
 from chatsky.messengers.console import CLIMessengerInterface
+from chatsky.messengers.http_interface import HTTPMessengerInterface

chatsky/messengers/http_interface.py

@@ -0,0 +1,52 @@
+import os
+import time
+from typing import Optional
+
+import dotenv
+import uvicorn
+from fastapi import FastAPI
+from pydantic import BaseModel
+
+from chatsky.core import Message
+from chatsky.messengers.common import MessengerInterface
+
+dotenv.load_dotenv()
+
+HTTP_INTERFACE_PORT = int(os.getenv("HTTP_INTERFACE_PORT", 8020))
+
+
+class HealthStatus(BaseModel):
+    status: str
+    uptime: Optional[float]
+
+
+class HTTPMessengerInterface(MessengerInterface):
+    async def connect(self, pipeline_runner):
+        app = FastAPI()
+
+        class Output(BaseModel):
+            user_id: str
+            response: Message
+
+        @app.post("/chat", response_model=Output)
+        async def respond(
+            user_id: str,
+            user_message: Message,
+        ):
+            message = Message(text=user_message)
+            context = await pipeline_runner(message, user_id)
+            return {"user_id": user_id, "response": context.last_response}
+
+        @app.get("/health", response_model=HealthStatus)
+        async def health_check():
+            return {
+                "status": "ok",
+                "uptime": str(time.time() - self.start_time),
+            }
+
+        self.start_time = time.time()
+        uvicorn.run(
+            app,
+            host="0.0.0.0",
+            port=HTTP_INTERFACE_PORT,
+        )

pyproject.toml

@@ -75,6 +75,8 @@ opentelemetry-instrumentation = { version = "*", optional = true }
 sqlalchemy = { version = "*", extras = ["asyncio"], optional = true }
 opentelemetry-exporter-otlp = { version = ">=1.20.0", optional = true }  # log body serialization is required
 pyyaml = { version = "*", optional = true }
+fastapi = { version = "*", optional = true }
+uvicorn = { version = "*", optional = true }
 
 [tool.poetry.extras]
 json = ["aiofiles"]
@@ -89,6 +91,7 @@ telegram = ["python-telegram-bot"]
 stats = ["opentelemetry-exporter-otlp", "opentelemetry-instrumentation", "requests", "tqdm", "omegaconf"]
 benchmark = ["pympler", "humanize", "pandas", "altair", "tqdm"]
 yaml = ["pyyaml"]
+web-api = ["fastapi", "uvicorn"]
 
 
 [tool.poetry.group.lint]