arcade-google-docs 4.1.0__py3-none-any.whl → 4.3.1__py3-none-any.whl

arcade_google_docs/tools/__init__.py

@@ -12,6 +12,7 @@ from arcade_google_docs.tools.search import (
     search_and_retrieve_documents,
     search_documents,
 )
+from arcade_google_docs.tools.system_context import who_am_i
 from arcade_google_docs.tools.update import insert_text_at_end_of_document
 
 __all__ = [
@@ -24,4 +25,5 @@ __all__ = [
     "search_and_retrieve_documents",
     "search_documents",
     "generate_google_file_picker_url",
+    "who_am_i",
 ]

arcade_google_docs/tools/system_context.py

@@ -0,0 +1,36 @@
+from typing import Annotated, Any
+
+from arcade_tdk import ToolContext, tool
+from arcade_tdk.auth import Google
+
+from arcade_google_docs.utils import build_docs_service
+from arcade_google_docs.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 Docs environment information.",
+]:
+    """
+    Get comprehensive user profile and Google Docs environment information.
+
+    This tool provides detailed information about the authenticated user including
+    their name, email, profile picture, Google Docs access permissions, and other
+    important profile details from Google services.
+    """
+
+    docs_service = build_docs_service(context.get_auth_token_or_empty())
+    user_info = build_who_am_i_response(context, docs_service)
+
+    return dict(user_info)

arcade_google_docs/who_am_i_util.py

@@ -0,0 +1,86 @@
+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_docs_access: bool
+
+
+def build_who_am_i_response(context: Any, docs_service: Any) -> WhoAmIResponse:
+    """Build complete who_am_i response from Google Docs 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_docs_info(docs_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_docs_info(docs_service: Any) -> dict[str, Any]:
+    """Extract Google Docs specific information."""
+    docs_info = {}
+
+    try:
+        # Test Google Docs access by checking if we can access the service
+        # Since there's no direct way to test docs access, we'll assume it's available
+        # if we have the service (the scope validation happens at auth time)
+        docs_info["google_docs_access"] = True
+    except Exception:
+        docs_info["google_docs_access"] = False
+
+    return docs_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_docs-4.1.0.dist-info → arcade_google_docs-4.3.1.dist-info}/METADATA

@@ -1,11 +1,11 @@
 Metadata-Version: 2.4
 Name: arcade_google_docs
-Version: 4.1.0
+Version: 4.3.1
 Summary: Arcade.dev LLM tools for Google Docs
 Author-email: Arcade <dev@arcade.dev>
 License: Proprietary - Arcade Software License Agreement v1.0
 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
@@ -13,8 +13,8 @@ Requires-Dist: google-auth<3.0.0,>=2.32.0
 Requires-Dist: googleapis-common-protos<2.0.0,>=1.63.2
 Requires-Dist: openai==1.82.1
 Provides-Extra: dev
-Requires-Dist: arcade-ai[evals]<3.0.0,>=2.0.4; 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_docs-4.1.0.dist-info → arcade_google_docs-4.3.1.dist-info}/RECORD

@@ -5,15 +5,17 @@ arcade_google_docs/docmd.py,sha256=rgxUt0k2Nes0GWLlPR49qU0nnM8e6HCiqd61Z7zNfTk,2
 arcade_google_docs/enum.py,sha256=kuXlsHcMYbN28Qg-Dwp4viz-CZ8z85_WVjQVZj2EsEY,3441
 arcade_google_docs/templates.py,sha256=pxbdMj57eV3-ImW3CixDWscpVKS94Z8nTNyTxDhUfGY,283
 arcade_google_docs/utils.py,sha256=XMgKcWPWKy8SbH3y2eCil01hssHhUgYxVrwXyvHEU4A,3748
+arcade_google_docs/who_am_i_util.py,sha256=NGQpQ0CuE2N86mGpkIdujXBRjcPBJ5SJ9BYzlb3mGFI,2925
 arcade_google_docs/models/document.py,sha256=0RvZ2_dfpz6ZoF1aUucYWOkRYWy_K_hiChSzoQtwhTc,30419
 arcade_google_docs/models/document_writables.py,sha256=DMBT5A05y7o7_PYlBB6O3KThma6-pm6hd5nxKGydT5Q,27575
 arcade_google_docs/models/requests.py,sha256=8Cga7QECmQWNFhM2QiGudvnQcgA_THi7ThNsUb7uavg,52176
-arcade_google_docs/tools/__init__.py,sha256=s0FS3z-uXE4lQH06YhnUKgIOT-Kq-v6rQMqXUGdPDnw,827
+arcade_google_docs/tools/__init__.py,sha256=kQChYFQ6y1LJg7q6fngzCOPtaDH1-CeWUOzqxfR_oGY,904
 arcade_google_docs/tools/comment.py,sha256=Qm5NHdNHONs3j4gqbZ7Fw9NrTVBb_mZ-th1X-z2IoLM,2836
 arcade_google_docs/tools/create.py,sha256=AuYy8yMGscrxAdLJQX0WiisGHCTufSlaRu_QGMMKQmM,2764
 arcade_google_docs/tools/file_picker.py,sha256=Dqn-hfMoTsWyHM8QCakVgHr5TKrzL_1Lj-vYHVGtOW4,2342
 arcade_google_docs/tools/get.py,sha256=Aby3iI1EzjfZTovGH4OdwpIhuazfV4SQinpXDOl87cg,2076
 arcade_google_docs/tools/search.py,sha256=1k1cI-fKOo4SZU4ufCwZ4DysK3l9MhPScZggPfNCC5Y,8261
+arcade_google_docs/tools/system_context.py,sha256=19HPSpNkLsb-MDWc-9CFgK_ha-rRzwaJr7hV6Us_1LI,1130
 arcade_google_docs/tools/update.py,sha256=_dReYit0s7ykn2bYQEUwohl3D_63U5leF87egO4eEiQ,1836
 arcade_google_docs/tools/edit_agent/edit_agent.py,sha256=1LIgKrQ70pDWzWNoaoy1st659perqa-ZW_ALmbVRbW0,2439
 arcade_google_docs/tools/edit_agent/executor.py,sha256=YbJqNXF0fw4N-u59gTbLpOBwKfoL7N6KZIOzV0pTllc,3718
@@ -23,6 +25,6 @@ arcade_google_docs/tools/edit_agent/prompts.py,sha256=M_f-HsPJppd3FQPhRAw7pKpaAr
 arcade_google_docs/tools/edit_agent/request_generator.py,sha256=eVDmzJDsOmJ-S8yANmJATt_G51rgCcCga6ArX4DTShM,5399
 arcade_google_docs/tools/edit_agent/utils.py,sha256=6lYmdQfX4CHfemP5PfxcXKSYpbdqFyEad997-gzCD48,639
 arcade_google_docs/tools/edit_agent/models/planning.py,sha256=RWQFB_KHl3Pq-snv1rHzoRxVvTnHVLZEGRpdohSX7wc,2962
-arcade_google_docs-4.1.0.dist-info/METADATA,sha256=ufO055P83TTGSEMN81EVwVhp6KusftVC20JpJwu0BRE,1127
-arcade_google_docs-4.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-arcade_google_docs-4.1.0.dist-info/RECORD,,
+arcade_google_docs-4.3.1.dist-info/METADATA,sha256=GOuskDQCsYU78EkuJJTXzGswlz9hf9wuoUzG-oQR0Oo,1127
+arcade_google_docs-4.3.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+arcade_google_docs-4.3.1.dist-info/RECORD,,