mirror of
https://github.com/tiennm99/DocsGPT.git
synced 2026-10-12 20:15:26 +00:00
patch: available tools
This commit is contained in:
1 parent
6b6495f48c
commit
02e93d993d
2 files changed
+11
No files matched your search
@@ -130,6 +130,8 @@ tools_ns = Namespace("tools", description="Tool management operations", path="/a
|
||||
class AvailableTools(Resource):
|
||||
@api.doc(description="Get available tools for a user")
|
||||
def get(self):
|
||||
if not request.decoded_token:
|
||||
return make_response(jsonify({"success": False}), 401)
|
||||
try:
|
||||
tools_metadata = []
|
||||
for tool_name, tool_instance in tool_manager.tools.items():
|
||||
|
||||
@@ -395,6 +395,9 @@ class TestAvailableTools:
|
||||
"application.api.user.tools.routes.tool_manager", mock_manager
|
||||
):
|
||||
with app.test_request_context("/api/available_tools"):
|
||||
from flask import request
|
||||
|
||||
request.decoded_token = {"sub": "user1"}
|
||||
response = AvailableTools().get()
|
||||
|
||||
assert response.status_code == 200
|
||||
@@ -419,6 +422,9 @@ class TestAvailableTools:
|
||||
"application.api.user.tools.routes.tool_manager", mock_manager
|
||||
):
|
||||
with app.test_request_context("/api/available_tools"):
|
||||
from flask import request
|
||||
|
||||
request.decoded_token = {"sub": "user1"}
|
||||
response = AvailableTools().get()
|
||||
|
||||
assert response.status_code == 400
|
||||
@@ -438,6 +444,9 @@ class TestAvailableTools:
|
||||
"application.api.user.tools.routes.tool_manager", mock_manager
|
||||
):
|
||||
with app.test_request_context("/api/available_tools"):
|
||||
from flask import request
|
||||
|
||||
request.decoded_token = {"sub": "user1"}
|
||||
response = AvailableTools().get()
|
||||
|
||||
assert response.status_code == 200
|
||||
|
||||
Reference in new issue
Block a user