arcade-google-sheets 3.1.1__py3-none-any.whl → 4.1.0__py3-none-any.whl

arcade_google_sheets/__init__.py

@@ -1,7 +1,21 @@
 from arcade_google_sheets.tools import (
+    add_note_to_cell,
     create_spreadsheet,
+    generate_google_file_picker_url,
     get_spreadsheet,
+    get_spreadsheet_metadata,
+    search_spreadsheets,
+    update_cells,
     write_to_cell,
 )
 
-__all__ = ["create_spreadsheet", "get_spreadsheet", "write_to_cell"]
+__all__ = [
+    "create_spreadsheet",
+    "get_spreadsheet",
+    "write_to_cell",
+    "generate_google_file_picker_url",
+    "get_spreadsheet_metadata",
+    "search_spreadsheets",
+    "update_cells",
+    "add_note_to_cell",
+]

arcade_google_sheets/templates.py

@@ -1,8 +1,2 @@
-optional_file_picker_instructions_template = (
-    "Ensure the user knows that they have the option to select and grant access permissions to "
-    "additional files and folders via the Google Drive File Picker. "
-    "The user can pick additional files and folders via the following link: {url}"
-)
-
 spreadsheet_url_template = "https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit"
 sheet_url_template = "https://docs.google.com/spreadsheets/d/{spreadsheet_id}/edit#gid={sheet_id}"

arcade_google_sheets/tools/__init__.py

@@ -1,5 +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,
@@ -15,4 +17,6 @@ __all__ = [
     "update_cells",
     "add_note_to_cell",
     "write_to_cell",
+    "generate_google_file_picker_url",
+    "who_am_i",
 ]

arcade_google_sheets/{file_picker.py → tools/file_picker.py}

@@ -1,14 +1,22 @@
 import base64
 import json
+from typing import Annotated
 
-from arcade_tdk import ToolContext, ToolMetadataKey
+from arcade_tdk import ToolContext, ToolMetadataKey, tool
+from arcade_tdk.auth import Google
 from arcade_tdk.errors import ToolExecutionError
 
 
-def generate_google_file_picker_url(context: ToolContext) -> dict:
+@tool(
+    requires_auth=Google(),
+    requires_metadata=[ToolMetadataKey.CLIENT_ID, ToolMetadataKey.COORDINATOR_URL],
+)
+def generate_google_file_picker_url(
+    context: ToolContext,
+) -> Annotated[dict, "Google File Picker URL for user file selection and permission granting"]:
     """Generate a Google File Picker URL for user-driven file selection and authorization.
 
-    Generates a URL that directs the end-user to a Google File Picker interface where
+    This tool generates a URL that directs the end-user to a Google File Picker interface where
     where they can select or upload Google Drive files. Users can grant permission to access their
     Drive files, providing a secure and authorized way to interact with their files.
 
@@ -17,8 +25,8 @@ def generate_google_file_picker_url(context: ToolContext) -> dict:
     (Requested entity was not found) or permission errors. Once the user completes the file
     picker flow, the prior tool can be retried.
 
-    Returns:
-        A dictionary containing the URL and instructions for the llm to instruct the user.
+    Suggest this tool to users when they are surprised or confused that the file they are
+    searching for or attempting to access cannot be found.
     """
     client_id = context.get_metadata(ToolMetadataKey.CLIENT_ID)
     client_id_parts = client_id.split("-")
@@ -44,6 +52,6 @@ def generate_google_file_picker_url(context: ToolContext) -> dict:
         "url": url,
         "llm_instructions": (
             "Instruct the user to click the following link to open the Google Drive File Picker. "
-            f"This will allow them to select files and grant access permissions: {url}"
+            "This will allow them to select files and grant access permissions: {url}"
         ),
     }

arcade_google_sheets/tools/read.py

@@ -1,9 +1,8 @@
 from typing import Annotated
 
-from arcade_tdk import ToolContext, ToolMetadataKey, tool
+from arcade_tdk import ToolContext, tool
 from arcade_tdk.auth import Google
 
-from arcade_google_sheets.decorators import with_filepicker_fallback
 from arcade_google_sheets.templates import sheet_url_template
 from arcade_google_sheets.utils import (
     build_sheets_service,
@@ -18,9 +17,7 @@ from arcade_google_sheets.utils import (
     requires_auth=Google(
         scopes=["https://www.googleapis.com/auth/drive.file"],
     ),
-    requires_metadata=[ToolMetadataKey.CLIENT_ID, ToolMetadataKey.COORDINATOR_URL],
 )
-@with_filepicker_fallback
 async def get_spreadsheet(
     context: ToolContext,
     spreadsheet_id: Annotated[str, "The id of the spreadsheet to get"],
@@ -90,9 +87,7 @@ async def get_spreadsheet(
     requires_auth=Google(
         scopes=["https://www.googleapis.com/auth/drive.file"],
     ),
-    requires_metadata=[ToolMetadataKey.CLIENT_ID, ToolMetadataKey.COORDINATOR_URL],
 )
-@with_filepicker_fallback
 async def get_spreadsheet_metadata(
     context: ToolContext,
     spreadsheet_id: Annotated[str, "The id of the spreadsheet to get metadata for"],

arcade_google_sheets/tools/search.py

@@ -1,12 +1,10 @@
 from typing import Annotated, Any
 
-from arcade_tdk import ToolContext, ToolMetadataKey, tool
+from arcade_tdk import ToolContext, tool
 from arcade_tdk.auth import Google
 
 from arcade_google_sheets.enums import OrderBy
-from arcade_google_sheets.file_picker import generate_google_file_picker_url
 from arcade_google_sheets.templates import (
-    optional_file_picker_instructions_template,
     spreadsheet_url_template,
 )
 from arcade_google_sheets.utils import (
@@ -20,7 +18,6 @@ from arcade_google_sheets.utils import (
     requires_auth=Google(
         scopes=["https://www.googleapis.com/auth/drive.file"],
     ),
-    requires_metadata=[ToolMetadataKey.CLIENT_ID, ToolMetadataKey.COORDINATOR_URL],
 )
 async def search_spreadsheets(
     context: ToolContext,
@@ -121,20 +118,10 @@ async def search_spreadsheets(
     for spreadsheet in spreadsheets:
         spreadsheet["url"] = spreadsheet_url_template.format(spreadsheet_id=spreadsheet["id"])
 
-    file_picker_response = generate_google_file_picker_url(
-        context,
-    )
-
     tool_response = {
         "pagination_token": pagination_token,
         "spreadsheets_count": len(spreadsheets),
         "spreadsheets": spreadsheets,
-        "file_picker": {
-            "url": file_picker_response["url"],
-            "llm_instructions": optional_file_picker_instructions_template.format(
-                url=file_picker_response["url"]
-            ),
-        },
     }
     tool_response = remove_none_values(tool_response)
 

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-3.1.1.dist-info → arcade_google_sheets-4.1.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: arcade_google_sheets
-Version: 3.1.1
+Version: 4.1.0
 Summary: Arcade.dev LLM tools for Google Sheets
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0

arcade_google_sheets-4.1.0.dist-info/RECORD

@@ -0,0 +1,19 @@
+arcade_google_sheets/__init__.py,sha256=asCX32sPK-bi2HRpWZSlLd7dRKy7Uztqxwu-eoXTY1g,466
+arcade_google_sheets/constants.py,sha256=4tQOrQ1YagSklJSSw5Eq21XCcFCJdO7lso5SqWIdrPI,63
+arcade_google_sheets/converters.py,sha256=1muE_28Jk0eyFUzoVo_lN0_qy3z-uv7f5XoZLH9mitQ,3924
+arcade_google_sheets/enums.py,sha256=30OMPu2l_aUjz8kQzC1mR5w0lLbs0fZn8T6PDtoXpL4,4608
+arcade_google_sheets/models.py,sha256=xwPdwUis0OHTDbSLy85qWl7zIh2lDTi5HZMNKwdZjzo,8627
+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/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.1.0.dist-info/METADATA,sha256=QUcLQJJbVePzOX49jkTOVDOgsN_nKzX605HcboCi3sM,1123
+arcade_google_sheets-4.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arcade_google_sheets-4.1.0.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
+arcade_google_sheets-4.1.0.dist-info/RECORD,,

arcade_google_sheets/decorators.py

@@ -1,24 +0,0 @@
-import functools
-from collections.abc import Callable
-from typing import Any
-
-from arcade_tdk import ToolContext
-from googleapiclient.errors import HttpError
-
-from arcade_google_sheets.file_picker import generate_google_file_picker_url
-
-
-def with_filepicker_fallback(func: Callable[..., Any]) -> Callable[..., Any]:
-    """ """
-
-    @functools.wraps(func)
-    async def async_wrapper(context: ToolContext, *args: Any, **kwargs: Any) -> Any:
-        try:
-            return await func(context, *args, **kwargs)
-        except HttpError as e:
-            if e.status_code in [403, 404]:
-                file_picker_response = generate_google_file_picker_url(context)
-                return file_picker_response
-            raise
-
-    return async_wrapper

arcade_google_sheets-3.1.1.dist-info/RECORD

@@ -1,18 +0,0 @@
-arcade_google_sheets/__init__.py,sha256=FB9h_cws_gu3UJp32GWlqvBQyAOb77JfAJNSSBqz-Jk,177
-arcade_google_sheets/constants.py,sha256=4tQOrQ1YagSklJSSw5Eq21XCcFCJdO7lso5SqWIdrPI,63
-arcade_google_sheets/converters.py,sha256=1muE_28Jk0eyFUzoVo_lN0_qy3z-uv7f5XoZLH9mitQ,3924
-arcade_google_sheets/decorators.py,sha256=QMqfvSXaFBoxYJrz69EGeMdAxF0V7JPReVXfp73Nf3Y,753
-arcade_google_sheets/enums.py,sha256=30OMPu2l_aUjz8kQzC1mR5w0lLbs0fZn8T6PDtoXpL4,4608
-arcade_google_sheets/file_picker.py,sha256=kGfUVfH5QVlIW1sL-_gAwPokt7TwVEcPk3Vnk53GKUE,2005
-arcade_google_sheets/models.py,sha256=xwPdwUis0OHTDbSLy85qWl7zIh2lDTi5HZMNKwdZjzo,8627
-arcade_google_sheets/templates.py,sha256=p3ty6Kwo7l73EEsZTaRfdG4jzQ9XvlqqeXMLKB1ydEw,489
-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=IiSd-0Q4_uxeKI8nVMT3JVRoF5G5GB8e5Z9SkGpSnt8,472
-arcade_google_sheets/tools/read.py,sha256=g7uGhyhFNlDPHKT3xhg7NdMaSYuvzzefSzrb1XqiI80,4309
-arcade_google_sheets/tools/search.py,sha256=EnO739RmBryDerYnzV7eD_drchNNVqS50DgdySYh9Ac,5381
-arcade_google_sheets/tools/write.py,sha256=4kNx941PQt6VUGTogbepnbfUdcsVze6u5c8QvlNnWCI,7782
-arcade_google_sheets-3.1.1.dist-info/METADATA,sha256=M6BDKH6yb3xVzPC9ByD0BjJePVy0D5g5x7IxyiouuZ8,1123
-arcade_google_sheets-3.1.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arcade_google_sheets-3.1.1.dist-info/licenses/LICENSE,sha256=ixeE7aL9b2B-_ZYHTY1vQcJB4NufKeo-LWwKNObGDN0,1960
-arcade_google_sheets-3.1.1.dist-info/RECORD,,