52807bfd0e
Updating the auth provider's `id` to simply `salesforce`. We don't want it to be confused with a future well-known `provider_id` of `arcade-salesforce`. Also, what the toolkit was referring to as `org_domain` is actually the organization's SUBdomain. We changed all references to "subdomain" to be more precise. The documentation has been updated accordingly.
64 lines
2.3 KiB
Python
64 lines
2.3 KiB
Python
from typing import Annotated
|
|
|
|
from arcade.sdk import ToolContext, tool
|
|
from arcade.sdk.auth import OAuth2
|
|
from arcade.sdk.errors import ToolExecutionError
|
|
|
|
from arcade_salesforce.exceptions import SalesforceToolExecutionError
|
|
from arcade_salesforce.models import SalesforceClient
|
|
|
|
|
|
# TODO: Add support for referencing an account by keywords (name, website, etc). We can use the
|
|
# get_account_data_by_keywords tool to search for accounts. If more than one is returned, we can
|
|
# raise a RetryableToolError providing the matches for the user to choose.
|
|
@tool(
|
|
requires_auth=OAuth2(
|
|
id="salesforce",
|
|
scopes=["write_contact"],
|
|
)
|
|
)
|
|
async def create_contact(
|
|
context: ToolContext,
|
|
account_id: Annotated[str, "The ID of the account to create the contact for."],
|
|
last_name: Annotated[str, "The last name of the contact."],
|
|
first_name: Annotated[str | None, "The first name of the contact."] = None,
|
|
email: Annotated[str | None, "The email of the contact."] = None,
|
|
phone: Annotated[str | None, "The phone number of the contact."] = None,
|
|
mobile_phone: Annotated[str | None, "The mobile phone number of the contact."] = None,
|
|
title: Annotated[
|
|
str | None,
|
|
"The title of the contact. E.g. 'CEO', 'Sales Director', 'CTO', etc.",
|
|
] = None,
|
|
department: Annotated[
|
|
str | None,
|
|
"The department of the contact. E.g. 'Marketing', 'Sales', 'IT', etc.",
|
|
] = None,
|
|
description: Annotated[str | None, "The description of the contact."] = None,
|
|
) -> Annotated[
|
|
dict,
|
|
"The created contact.",
|
|
]:
|
|
"""Creates a contact in Salesforce."""
|
|
if not last_name:
|
|
raise ToolExecutionError("Last name is required by Salesforce to create a contact.")
|
|
|
|
client = SalesforceClient(context.get_auth_token_or_empty())
|
|
contact = await client.create_contact(
|
|
account_id=account_id,
|
|
first_name=first_name,
|
|
last_name=last_name,
|
|
email=email,
|
|
phone=phone,
|
|
mobile_phone=mobile_phone,
|
|
title=title,
|
|
department=department,
|
|
description=description,
|
|
)
|
|
|
|
if not contact.get("success"):
|
|
raise SalesforceToolExecutionError(
|
|
message="Failed to create contact",
|
|
errors=contact.get("errors", []),
|
|
)
|
|
|
|
return {"success": True, "contactId": contact.get("id")}
|