Skip to content

Commit d48b84a

Browse files
authored
Feature/organization set ellicitation (#72)
* add elicitation to orgs request * fix tests * release: bump version to 0.4.0 (minor release)
1 parent 90d6fa3 commit d48b84a

File tree

9 files changed

+68
-789
lines changed

9 files changed

+68
-789
lines changed

src/api/common.py

Lines changed: 0 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -399,21 +399,3 @@ def get_access_token() -> str:
399399
raise HTTPException(401, "Unauthorized: No access token provided")
400400

401401
return access_token
402-
403-
404-
def get_current_organization():
405-
"""
406-
Get the current organization details from the management API.
407-
408-
Returns:
409-
dict: Organization details including orgID and name
410-
"""
411-
try:
412-
organization = build_request("GET", "organizations/current")
413-
logger.debug(f"Current organization response: {organization}")
414-
return organization
415-
except Exception as e:
416-
logger.error(f"Failed to get current organization: {str(e)}")
417-
raise ValueError(
418-
f"Could not retrieve current organization from the API: {str(e)}"
419-
)

src/api/tools/__init__.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
workspaces_info,
77
run_sql,
88
get_organizations,
9-
set_organization,
109
organization_info,
1110
list_virtual_workspaces,
1211
)
@@ -21,7 +20,6 @@
2120
"workspaces_info",
2221
"run_sql",
2322
"get_organizations",
24-
"set_organization",
2523
"organization_info",
2624
"list_virtual_workspaces",
2725
]

src/api/tools/registery.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ def register_tools(mcp: FastMCP, **filter_flags) -> None:
5656
)
5757

5858
# List of tools to exclude when using API key authentication
59-
api_key_excluded_tools = ["get_organizations", "set_organization"]
59+
api_key_excluded_tools = ["get_organizations"]
6060

6161
for tool in filtered_tools:
6262
func = tool.func

0 commit comments

Comments
 (0)