RAG_SyS / main.py
jwphantom
fix bug on main.py
61cc288
raw
history blame
No virus
709 Bytes
from fastapi import FastAPI
from app.api import chat
from fastapi.middleware.cors import CORSMiddleware
from app.schema.question import Question as SchemaQuestion
import os
from dotenv import load_dotenv
load_dotenv()
origins = [
"http://localhost:3000",
"http://localhost:4000",
]
app = FastAPI()
app.include_router(chat.router, prefix="/api/chat", tags=["chat"])
app.add_middleware(
CORSMiddleware,
allow_origins=origins,
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
@app.get("/")
async def read_root():
return {"message": "Welcome to the API"}
@app.post("/")
async def home(question: SchemaQuestion):
return await chat.home(question)