llamaindex-retrieval-api
llamaindex-retrieval-api copied to clipboard
Add an HTTPException for when the engine fails in query.py
Add an HTTPException for when the engine fails in query.py #29 #31
{"tag": "INITIAL_PLANNING", "workflow": {"tasks": [{"step": "Step 1", "name": "Import HTTPException", "action": "modify", "file": "routers/retrievers/query.py", "start_line": 1, "end_line": 1, "instructions": "At the top of the file, import HTTPException from fastapi.exceptions by adding the line 'from fastapi.exceptions import HTTPException'.", "async_score": 2, "assigned_agent": "AI", "dependencies": []}, {"step": "Step 2", "name": "Add HTTPException", "action": "modify", "file": "routers/retrievers/query.py", "start_line": 10, "end_line": 20, "instructions": "Identify the function where the engine is called. After the engine call, add a try-except block. In the except block, raise an HTTPException with status code 500 and a detail message indicating that the engine has failed. The code should look like this: 'try: [engine call] except: raise HTTPException(status_code=500, detail="Engine has failed.")'.", "async_score": 8, "assigned_agent": "AI", "dependencies": ["Step 1"]}, {"step": "Step 3", "name": "Test HTTPException", "action": "create", "file": "tests/test_query.py", "start_line": 1, "end_line": 50, "instructions": "Create a new file in the tests directory named test_query.py. In this file, write tests to ensure that the HTTPException is raised when the engine fails. Import the necessary modules and functions, including the function you modified in Step 2. Create test data that will cause the engine to fail. Call the function with this test data and assert that an HTTPException is raised with the correct status code and detail message.", "async_score": 10, "assigned_agent": "User", "dependencies": ["Step 2"]}], "snippets": []}}
Step 1: Import HTTPException
Action: modify
File: routers/retrievers/query.py
Start Line: 1
End Line: 1
Instructions: At the top of the file, import HTTPException from fastapi.exceptions by adding the line 'from fastapi.exceptions import HTTPException'.
Async Score: 2
Assigned Agent: AI
Dependencies: []
Step 2: Add HTTPException
Action: modify
File: routers/retrievers/query.py
Start Line: 10
End Line: 20
Instructions: Identify the function where the engine is called. After the engine call, add a try-except block. In the except block, raise an HTTPException with status code 500 and a detail message indicating that the engine has failed. The code should look like this: 'try: [engine call] except: raise HTTPException(status_code=500, detail="Engine has failed.")'.
Async Score: 8
Assigned Agent: AI
Dependencies: ['Step 1']
Step 3: Test HTTPException
Action: create
File: tests/test_query.py
Start Line: 1
End Line: 50
Instructions: Create a new file in the tests directory named test_query.py. In this file, write tests to ensure that the HTTPException is raised when the engine fails. Import the necessary modules and functions, including the function you modified in Step 2. Create test data that will cause the engine to fail. Call the function with this test data and assert that an HTTPException is raised with the correct status code and detail message.
Async Score: 10
Assigned Agent: User
Dependencies: ['Step 2']
Looks good to me!