refactor: Move permission router inside routers folder, add cognify router
Moved permissions router in it's routers folder, added cognify router Refactor #COG-334
This commit is contained in:
parent
f09c28aada
commit
f8f2746603
5 changed files with 37 additions and 17 deletions
|
|
@ -64,8 +64,9 @@ app.add_middleware(
|
||||||
|
|
||||||
from cognee.api.v1.users.routers import get_auth_router, get_register_router,\
|
from cognee.api.v1.users.routers import get_auth_router, get_register_router,\
|
||||||
get_reset_password_router, get_verify_router, get_users_router
|
get_reset_password_router, get_verify_router, get_users_router
|
||||||
from cognee.api.v1.permissions.get_permissions_router import get_permissions_router
|
from cognee.api.v1.permissions.routers import get_permissions_router
|
||||||
from cognee.api.v1.settings.routers.get_settings_router import get_settings_router
|
from cognee.api.v1.settings.routers import get_settings_router
|
||||||
|
from cognee.api.v1.cognify.routers import get_cognify_router
|
||||||
from cognee.api.v1.search.routers import get_search_router
|
from cognee.api.v1.search.routers import get_search_router
|
||||||
|
|
||||||
from fastapi import Request
|
from fastapi import Request
|
||||||
|
|
@ -313,21 +314,11 @@ async def add(
|
||||||
content = {"error": str(error)}
|
content = {"error": str(error)}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
app.include_router(
|
||||||
class CognifyPayloadDTO(BaseModel):
|
get_cognify_router(),
|
||||||
datasets: List[str]
|
prefix="/api/v1/cognify",
|
||||||
|
tags=["cognify"]
|
||||||
@app.post("/api/v1/cognify", response_model = None)
|
)
|
||||||
async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)):
|
|
||||||
""" This endpoint is responsible for the cognitive processing of the content."""
|
|
||||||
from cognee.api.v1.cognify.cognify_v2 import cognify as cognee_cognify
|
|
||||||
try:
|
|
||||||
await cognee_cognify(payload.datasets, user)
|
|
||||||
except Exception as error:
|
|
||||||
return JSONResponse(
|
|
||||||
status_code = 409,
|
|
||||||
content = {"error": str(error)}
|
|
||||||
)
|
|
||||||
|
|
||||||
app.include_router(
|
app.include_router(
|
||||||
get_search_router(),
|
get_search_router(),
|
||||||
|
|
|
||||||
1
cognee/api/v1/cognify/routers/__init__.py
Normal file
1
cognee/api/v1/cognify/routers/__init__.py
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
from .get_cognify_router import get_cognify_router
|
||||||
27
cognee/api/v1/cognify/routers/get_cognify_router.py
Normal file
27
cognee/api/v1/cognify/routers/get_cognify_router.py
Normal file
|
|
@ -0,0 +1,27 @@
|
||||||
|
from fastapi import APIRouter
|
||||||
|
from typing import List
|
||||||
|
from pydantic import BaseModel
|
||||||
|
from cognee.modules.users.models import User
|
||||||
|
from fastapi.responses import JSONResponse
|
||||||
|
from cognee.modules.users.methods import get_authenticated_user
|
||||||
|
from fastapi import Depends
|
||||||
|
|
||||||
|
def get_cognify_router():
|
||||||
|
router = APIRouter()
|
||||||
|
|
||||||
|
class CognifyPayloadDTO(BaseModel):
|
||||||
|
datasets: List[str]
|
||||||
|
|
||||||
|
@router.post("/", response_model=None)
|
||||||
|
async def cognify(payload: CognifyPayloadDTO, user: User = Depends(get_authenticated_user)):
|
||||||
|
""" This endpoint is responsible for the cognitive processing of the content."""
|
||||||
|
from cognee.api.v1.cognify.cognify_v2 import cognify as cognee_cognify
|
||||||
|
try:
|
||||||
|
await cognee_cognify(payload.datasets, user)
|
||||||
|
except Exception as error:
|
||||||
|
return JSONResponse(
|
||||||
|
status_code=409,
|
||||||
|
content={"error": str(error)}
|
||||||
|
)
|
||||||
|
|
||||||
|
return router
|
||||||
1
cognee/api/v1/permissions/routers/__init__.py
Normal file
1
cognee/api/v1/permissions/routers/__init__.py
Normal file
|
|
@ -0,0 +1 @@
|
||||||
|
from .get_permissions_router import get_permissions_router
|
||||||
Loading…
Add table
Reference in a new issue