Add echo endpoint

main.py

@@ -17,6 +17,9 @@ class Item(BaseModel):
     top_p: float = 0.15
     repetition_penalty: float = 1.0
 
+class EchoMessage(BaseModel):
+    message: str
+
 def format_prompt(message, history):
     prompt = "<s>"
     for user_prompt, bot_response in history:
@@ -55,3 +58,7 @@ async def generate_text(item: Item):
 @app.get("/")
 async def home():
     return {"msg":"hey"}
+
+@app.post("/echo/")
+async def echo(echo_msg: EchoMessage):
+    return {"msg":echo_msg.message}