jiarongqiu commited on
Commit
5273e65
·
1 Parent(s): b63d6c4

add scheduler

Browse files
Files changed (1) hide show
  1. main.py +5 -5
main.py CHANGED
@@ -1,8 +1,9 @@
1
  from fastapi import FastAPI,Request
2
- from service.vector_store import vector_store
3
  from fastapi.responses import StreamingResponse
4
- from service.api import api
5
  from fastapi.middleware.cors import CORSMiddleware
 
 
 
6
  from pydantic import BaseModel
7
 
8
  app = FastAPI()
@@ -14,9 +15,8 @@ app.add_middleware(
14
  allow_headers=["*"], # 允许所有头
15
  )
16
 
17
- class Query(BaseModel):
18
- endpoint: str
19
- query: str
20
 
21
 
22
 
 
1
  from fastapi import FastAPI,Request
 
2
  from fastapi.responses import StreamingResponse
 
3
  from fastapi.middleware.cors import CORSMiddleware
4
+ from service.vector_store import vector_store
5
+ from service.api import api
6
+ from service.scheduler import Scheduler
7
  from pydantic import BaseModel
8
 
9
  app = FastAPI()
 
15
  allow_headers=["*"], # 允许所有头
16
  )
17
 
18
+ scheduler = Scheduler()
19
+ scheduler.start()
 
20
 
21
 
22