andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-27 17:18:13 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refactor: propagate the current org thought Header instead of params

Browse Source
This commit is contained in:
Lucas Gomide
2025-06-06 10:53:47 -03:00
parent 3ffe9c2ebe
commit 0953b3f8b5
2 changed files with 52 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
src/crewai/cli/plus_api.py
View File

@@ -26,6 +26,9 @@ class PlusAPI:
"User-Agent": f"CrewAI-CLI/{get_crewai_version()}",
"X-Crewai-Version": get_crewai_version(),
}
settings = Settings()
if settings.org_uuid:
self.headers["X-Crewai-Organization-Id"] = settings.org_uuid
self.base_url = getenv("CREWAI_BASE_URL", "https://app.crewai.com")
def _make_request(self, method: str, endpoint: str, **kwargs) -> requests.Response:
@@ -35,25 +38,13 @@ class PlusAPI:
return session.request(method, url, headers=self.headers, **kwargs)
def login_to_tool_repository(self):
settings = Settings()
payload = {}
if settings.org_uuid:
payload["organization_uuid"] = settings.org_uuid
return self._make_request("POST", f"{self.TOOLS_RESOURCE}/login", json=payload)
return self._make_request("POST", f"{self.TOOLS_RESOURCE}/login")
def get_tool(self, handle: str):
settings = Settings()
params = {}
if settings.org_uuid:
params["organization_uuid"] = settings.org_uuid
return self._make_request("GET", f"{self.TOOLS_RESOURCE}/{handle}", params=params)
return self._make_request("GET", f"{self.TOOLS_RESOURCE}/{handle}")
def get_agent(self, handle: str):
settings = Settings()
params = {}
if settings.org_uuid:
params["organization_uuid"] = settings.org_uuid
return self._make_request("GET", f"{self.AGENTS_RESOURCE}/{handle}", params=params)
return self._make_request("GET", f"{self.AGENTS_RESOURCE}/{handle}")
def publish_tool(
self,
@@ -72,9 +63,6 @@ class PlusAPI:
"description": description,
"available_exports": available_exports,
}
settings = Settings()
if settings.org_uuid:
params["organization_uuid"] = settings.org_uuid
return self._make_request("POST", f"{self.TOOLS_RESOURCE}", json=params)
def deploy_by_name(self, project_name: str) -> requests.Response: