arcade-google-sheets 4.0.0__py3-none-any.whl → 4.2.1__py3-none-any.whl

arcade_google_sheets/tools/__init__.py

@@ -1,6 +1,7 @@
 from arcade_google_sheets.tools.file_picker import generate_google_file_picker_url
 from arcade_google_sheets.tools.read import get_spreadsheet, get_spreadsheet_metadata
 from arcade_google_sheets.tools.search import search_spreadsheets
+from arcade_google_sheets.tools.system_context import who_am_i
 from arcade_google_sheets.tools.write import (
     add_note_to_cell,
     create_spreadsheet,
@@ -17,4 +18,5 @@ __all__ = [
     "add_note_to_cell",
     "write_to_cell",
     "generate_google_file_picker_url",
+    "who_am_i",
 ]

arcade_google_sheets/tools/system_context.py

@@ -0,0 +1,37 @@
+from typing import Annotated, Any
+
+from arcade_tdk import ToolContext, tool
+from arcade_tdk.auth import Google
+
+from arcade_google_sheets.utils import build_sheets_service
+from arcade_google_sheets.who_am_i_util import build_who_am_i_response
+
+
+@tool(
+    requires_auth=Google(
+        scopes=[
+            "https://www.googleapis.com/auth/drive.file",
+            "https://www.googleapis.com/auth/userinfo.profile",
+            "https://www.googleapis.com/auth/userinfo.email",
+        ]
+    )
+)
+async def who_am_i(
+    context: ToolContext,
+) -> Annotated[
+    dict[str, Any],
+    "Get comprehensive user profile and Google Sheets environment information.",
+]:
+    """
+    Get comprehensive user profile and Google Sheets environment information.
+
+    This tool provides detailed information about the authenticated user including
+    their name, email, profile picture, Google Sheets access permissions, and other
+    important profile details from Google services.
+    """
+
+    auth_token = context.get_auth_token_or_empty()
+    sheets_service = build_sheets_service(auth_token)
+    user_info = build_who_am_i_response(context, sheets_service)
+
+    return dict(user_info)

arcade_google_sheets/who_am_i_util.py

@@ -0,0 +1,81 @@
+from typing import Any, TypedDict, cast
+
+from google.oauth2.credentials import Credentials
+from googleapiclient.discovery import build
+
+
+class WhoAmIResponse(TypedDict, total=False):
+    my_email_address: str
+    display_name: str
+    given_name: str
+    family_name: str
+    formatted_name: str
+    profile_picture_url: str
+    google_sheets_access: bool
+
+
+def build_who_am_i_response(context: Any, sheets_service: Any) -> WhoAmIResponse:
+    """Build complete who_am_i response from Google Sheets and People APIs."""
+    credentials = Credentials(
+        context.authorization.token if context.authorization and context.authorization.token else ""
+    )
+    people_service = _build_people_service(credentials)
+    person = _get_people_api_data(people_service)
+
+    user_info = _extract_profile_data(person)
+    user_info.update(_extract_google_sheets_info(sheets_service))
+
+    return cast(WhoAmIResponse, user_info)
+
+
+def _extract_profile_data(person: dict[str, Any]) -> dict[str, Any]:
+    """Extract user profile data from People API response."""
+    profile_data = {}
+
+    names = person.get("names", [])
+    if names:
+        primary_name = names[0]
+        profile_data.update({
+            "display_name": primary_name.get("displayName"),
+            "given_name": primary_name.get("givenName"),
+            "family_name": primary_name.get("familyName"),
+            "formatted_name": primary_name.get("displayNameLastFirst"),
+        })
+
+    photos = person.get("photos", [])
+    if photos:
+        profile_data["profile_picture_url"] = photos[0].get("url")
+
+    email_addresses = person.get("emailAddresses", [])
+    if email_addresses:
+        primary_emails = [
+            email for email in email_addresses if email.get("metadata", {}).get("primary")
+        ]
+        if primary_emails:
+            profile_data["my_email_address"] = primary_emails[0].get("value")
+
+    return profile_data
+
+
+def _extract_google_sheets_info(sheets_service: Any) -> dict[str, Any]:
+    """Extract minimal Google Sheets access information."""
+    sheets_info: dict[str, Any] = {}
+    try:
+        sheets_info["google_sheets_access"] = True
+    except Exception:
+        sheets_info["google_sheets_access"] = False
+    return sheets_info
+
+
+def _build_people_service(credentials: Credentials) -> Any:
+    """Build and return the People API service client."""
+    return build("people", "v1", credentials=credentials)
+
+
+def _get_people_api_data(people_service: Any) -> dict[str, Any]:
+    """Get user profile information from People API."""
+    person_fields = "names,emailAddresses,photos"
+    return cast(
+        dict[str, Any],
+        people_service.people().get(resourceName="people/me", personFields=person_fields).execute(),
+    )

{arcade_google_sheets-4.0.0.dist-info → arcade_google_sheets-4.2.1.dist-info}/METADATA

@@ -1,20 +1,20 @@
 Metadata-Version: 2.4
 Name: arcade_google_sheets
-Version: 4.0.0
+Version: 4.2.1
 Summary: Arcade.dev LLM tools for Google Sheets
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0
 License-File: LICENSE
 Requires-Python: >=3.10
-Requires-Dist: arcade-tdk<3.0.0,>=2.0.0
+Requires-Dist: arcade-tdk<3.0.0,>=2.3.1
 Requires-Dist: google-api-core<3.0.0,>=2.19.1
 Requires-Dist: google-api-python-client<3.0.0,>=2.137.0
 Requires-Dist: google-auth-httplib2<1.0.0,>=0.2.0
 Requires-Dist: google-auth<3.0.0,>=2.32.0
 Requires-Dist: googleapis-common-protos<2.0.0,>=1.63.2
 Provides-Extra: dev
-Requires-Dist: arcade-ai[evals]<3.0.0,>=2.0.0; extra == 'dev'
-Requires-Dist: arcade-serve<3.0.0,>=2.0.0; extra == 'dev'
+Requires-Dist: arcade-ai[evals]<3.0.0,>=2.2.1; extra == 'dev'
+Requires-Dist: arcade-serve<3.0.0,>=2.1.0; extra == 'dev'
 Requires-Dist: mypy<1.6.0,>=1.5.1; extra == 'dev'
 Requires-Dist: pre-commit<3.5.0,>=3.4.0; extra == 'dev'
 Requires-Dist: pytest-asyncio<0.25.0,>=0.24.0; extra == 'dev'

{arcade_google_sheets-4.0.0.dist-info → arcade_google_sheets-4.2.1.dist-info}/RECORD

@@ -6,12 +6,14 @@ arcade_google_sheets/models.py,sha256=xwPdwUis0OHTDbSLy85qWl7zIh2lDTi5HZMNKwdZjz
 arcade_google_sheets/templates.py,sha256=hSvjiErERLPe6RqWY6QNmivOy7Ofdg_zABcLWCI-ByU,189
 arcade_google_sheets/types.py,sha256=R-rCRcyFqDZx3jgl_kWeCliqC8fHuZ8ub_LQ2KoU2AE,37
 arcade_google_sheets/utils.py,sha256=VmDZOzAOEtfSPOra-ieVl_U16RLonQUOnZ4RW4Gf-oA,34895
-arcade_google_sheets/tools/__init__.py,sha256=ZUhznD9qcK0cdaFM05I2eA2SCpY2nD4ou2_zUR8rHek,594
+arcade_google_sheets/who_am_i_util.py,sha256=MfDKsAOMUkH0DZNjMzcmDiuvxMEUJMT9o_a7uixCJF4,2734
+arcade_google_sheets/tools/__init__.py,sha256=GBe1Op-o6QzCy9nA3qS53Ij2Uex5IXdFPUnGQo_TPkg,673
 arcade_google_sheets/tools/file_picker.py,sha256=Dqn-hfMoTsWyHM8QCakVgHr5TKrzL_1Lj-vYHVGtOW4,2342
 arcade_google_sheets/tools/read.py,sha256=qQX_TdcbPbkuaHPuhx7CzWsWJ_a97FSCDJqOlGHF_TM,4003
 arcade_google_sheets/tools/search.py,sha256=olCaUwDqW26yaYqMT5lPmQDFL6M9g7qON7JG3mcgYJM,4841
+arcade_google_sheets/tools/system_context.py,sha256=rVpYeyFsL8mj0Lt6VWOhnfGaSy3ezOHL74EFJHjE9Ww,1176
 arcade_google_sheets/tools/write.py,sha256=4kNx941PQt6VUGTogbepnbfUdcsVze6u5c8QvlNnWCI,7782
-arcade_google_sheets-4.0.0.dist-info/METADATA,sha256=qDIQQqgbsYxUsqvmpNS7Om8xx9yC3HO_LfLA4uAbcYs,1123
-arcade_google_sheets-4.0.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arcade_google_sheets-4.0.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
-arcade_google_sheets-4.0.0.dist-info/RECORD,,
+arcade_google_sheets-4.2.1.dist-info/METADATA,sha256=xM1z7j9tHzv6ncsWfQ2fU0dKY7BkNZ_lBgufskoz68w,1123
+arcade_google_sheets-4.2.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arcade_google_sheets-4.2.1.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_google_sheets-4.2.1.dist-info/RECORD,,