CarGDev/avante.nvim
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: remove resource schema (#1512)

Browse Source
This commit is contained in:
yetone
2025-03-07 00:43:57 +08:00
committed by GitHub
parent 25dee44de9
commit a13e5dca1d
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
py/rag-service/src/main.py
View File

@@ -376,11 +376,16 @@ except (OSError, ValueError) as e:
index = VectorStoreIndex([], storage_context=storage_context) index = VectorStoreIndex([], storage_context=storage_context)
class ResourceRequest(BaseModel): class ResourceURIRequest(BaseModel):
"""Request model for resource operations."""
uri: str = Field(..., description="URI of the resource to watch and index")
class ResourceRequest(ResourceURIRequest):
"""Request model for resource operations.""" """Request model for resource operations."""
name: str = Field(..., description="Name of the resource to watch and index") name: str = Field(..., description="Name of the resource to watch and index")
uri: str = Field(..., description="URI of the resource to watch and index")
class SourceDocument(BaseModel): class SourceDocument(BaseModel):
@@ -978,7 +983,7 @@ async def add_resource(request: ResourceRequest, background_tasks: BackgroundTas
404: {"description": "Resource not found in watch list"}, 404: {"description": "Resource not found in watch list"},
}, },
) )
async def remove_resource(request: ResourceRequest): # noqa: D103, ANN201 async def remove_resource(request: ResourceURIRequest): # noqa: D103, ANN201
resource = resource_service.get_resource(request.uri) resource = resource_service.get_resource(request.uri)
if not resource or resource.status != "active": if not resource or resource.status != "active":
raise HTTPException(status_code=404, detail="Resource not being watched") raise HTTPException(status_code=404, detail="Resource not being watched")