hyperpocket 0.3.3__py3-none-any.whl → 0.3.4__py3-none-any.whl

hyperpocket/auth/apitoken/README.md

@@ -0,0 +1,28 @@
+### API Token Auth
+
+This module provides authentication using simple API tokens.
+
+1. To use this authentication in your tool, include the following in your `config.toml`:
+
+```toml
+[auth]
+auth_provider = "apitoken"
+auth_handler = "apitoken"
+scopes = []
+```
+
+2. To use it with `function_tool`, you can define your function as follows:
+
+```python
+from hyperpocket.tool import function_tool
+from hyperpocket.auth import AuthProvider
+
+
+@function_tool(
+    auth_provider=AuthProvider.API_TOKEN
+)
+def my_function(**kwargs):
+    token = kwargs["API_TOKEN"]
+
+    # ...
+```

hyperpocket/auth/apitoken/context.py

@@ -0,0 +1,15 @@
+from hyperpocket.auth.context import AuthContext
+
+
+class ApiTokenAuthContext(AuthContext):
+    _ACCESS_TOKEN_KEY: str = "API_TOKEN"
+
+    def to_dict(self) -> dict[str, str]:
+        return {
+            self._ACCESS_TOKEN_KEY: self.access_token,
+        }
+
+    def to_profiled_dict(self, profile: str) -> dict[str, str]:
+        return {
+            f"{profile.upper()}_{self._ACCESS_TOKEN_KEY}": self.access_token,
+        }

hyperpocket/auth/apitoken/token_context.py

@@ -0,0 +1,12 @@
+from hyperpocket.auth.apitoken.context import ApiTokenAuthContext
+from hyperpocket.auth.apitoken.token_schema import ApiTokenResponse
+
+
+class ApiTokenAuthContext(ApiTokenAuthContext):
+    @classmethod
+    def from_api_token_response(cls, response: ApiTokenResponse):
+        description = "Api Token Context logged in"
+
+        return cls(
+            access_token=response.access_token, description=description, expires_at=None
+        )

hyperpocket/auth/apitoken/token_handler.py

@@ -0,0 +1,89 @@
+from typing import Optional
+from urllib.parse import urlencode, urljoin
+
+from hyperpocket.auth import AuthProvider
+from hyperpocket.auth.context import AuthContext
+from hyperpocket.auth.handler import AuthHandlerInterface
+from hyperpocket.auth.apitoken.token_context import ApiTokenAuthContext
+from hyperpocket.auth.apitoken.token_schema import ApiTokenRequest, ApiTokenResponse
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class ApiTokenAuthHandler(AuthHandlerInterface):
+    name: str = "apitoken"
+    description: str = "This handler is used to authenticate users using the Api token."
+    scoped: bool = False
+
+    _TOKEN_URL: str = urljoin(
+        config().public_base_url + "/",
+        f"{config().callback_url_rewrite_prefix}/auth/token",
+    )
+
+    @staticmethod
+    def provider() -> AuthProvider:
+        return AuthProvider.APITOKEN
+
+    @staticmethod
+    def recommended_scopes() -> set[str]:
+        return set()
+
+    @staticmethod
+    def provider_default() -> bool:
+        return True
+
+    def prepare(
+        self,
+        auth_req: ApiTokenRequest,
+        thread_id: str,
+        profile: str,
+        future_uid: str,
+        *args,
+        **kwargs,
+    ) -> str:
+        redirect_uri = urljoin(
+            config().public_base_url + "/",
+            f"{config().callback_url_rewrite_prefix}/auth/apitoken/token/callback",
+        )
+        url = self._make_auth_url(
+            auth_req=auth_req, redirect_uri=redirect_uri, state=future_uid
+        )
+        FutureStore.create_future(
+            future_uid,
+            data={
+                "redirect_uri": redirect_uri,
+                "thread_id": thread_id,
+                "profile": profile,
+            },
+        )
+
+        return f"User needs to authenticate using the following URL: {url}"
+
+    async def authenticate(
+        self, auth_req: ApiTokenRequest, future_uid: str, *args, **kwargs
+    ) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        access_token = await future_data.future
+
+        response = ApiTokenResponse(access_token=access_token)
+        context = ApiTokenAuthContext.from_api_token_response(response)
+
+        return context
+
+    async def refresh(
+        self, auth_req: ApiTokenRequest, context: AuthContext, *args, **kwargs
+    ) -> AuthContext:
+        raise Exception("Api token doesn't support refresh")
+
+    def _make_auth_url(self, auth_req: ApiTokenRequest, redirect_uri: str, state: str):
+        params = {
+            "redirect_uri": redirect_uri,
+            "state": state,
+        }
+        auth_url = f"{self._TOKEN_URL}?{urlencode(params)}"
+        return auth_url
+
+    def make_request(
+        self, auth_scopes: Optional[list[str]] = None, **kwargs
+    ) -> ApiTokenRequest:
+        return ApiTokenRequest()

hyperpocket/auth/apitoken/token_schema.py

@@ -0,0 +1,9 @@
+from hyperpocket.auth.schema import AuthenticateRequest, AuthenticateResponse
+
+
+class ApiTokenRequest(AuthenticateRequest):
+    pass
+
+
+class ApiTokenResponse(AuthenticateResponse):
+    access_token: str

hyperpocket/auth/provider.py

@@ -10,7 +10,9 @@ class AuthProvider(Enum):
     NOTION = "notion"
     REDDIT = "reddit"
     GUMLOOP = "gumloop"
+    SERPAPI = "serpapi"
     X = "x"
+    APITOKEN = "apitoken"
 
     @classmethod
     def get_auth_provider(cls, auth_provider_name: str) -> "AuthProvider":

hyperpocket/auth/serpapi/context.py

@@ -0,0 +1,15 @@
+from hyperpocket.auth.context import AuthContext
+
+
+class SerpapiAuthContext(AuthContext):
+    _ACCESS_TOKEN_KEY: str = "SERPAPI_TOKEN"
+
+    def to_dict(self) -> dict[str, str]:
+        return {
+            self._ACCESS_TOKEN_KEY: self.access_token,
+        }
+
+    def to_profiled_dict(self, profile: str) -> dict[str, str]:
+        return {
+            f"{profile.upper()}_{self._ACCESS_TOKEN_KEY}": self.access_token,
+        }

hyperpocket/auth/serpapi/token_context.py

@@ -0,0 +1,13 @@
+from hyperpocket.auth.serpapi.context import SerpapiAuthContext
+from hyperpocket.auth.serpapi.token_schema import SerpapiTokenResponse
+
+
+class SerpapiTokenAuthContext(SerpapiAuthContext):
+    @classmethod
+    def from_serpapi_token_response(cls, response: SerpapiTokenResponse):
+        description = f'Serpapi Token Context logged in'
+        return cls(
+            access_token=response.access_token,
+            description=description,
+            expires_at=None
+        )

hyperpocket/auth/serpapi/token_handler.py

@@ -0,0 +1,68 @@
+from typing import Optional
+from urllib.parse import urljoin, urlencode
+
+from hyperpocket.auth import AuthProvider
+from hyperpocket.auth.context import AuthContext
+from hyperpocket.auth.handler import AuthHandlerInterface, AuthenticateRequest
+from hyperpocket.auth.serpapi.token_context import SerpapiTokenAuthContext
+from hyperpocket.auth.serpapi.token_schema import SerpapiTokenResponse, SerpapiTokenRequest
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class SerpapiTokenAuthHandler(AuthHandlerInterface):
+    name: str = "serpapi-token"
+    description: str = "This handler is used to authenticate users using the Serpapi token."
+    scoped: bool = False
+
+    _TOKEN_URL: str = urljoin(config().public_base_url + "/", f"{config().callback_url_rewrite_prefix}/auth/token")
+
+    @staticmethod
+    def provider_default() -> bool:
+        return True
+
+    @staticmethod
+    def provider() -> AuthProvider:
+        return AuthProvider.SERPAPI
+
+    @staticmethod
+    def recommended_scopes() -> set[str]:
+        return set()
+
+    def prepare(self, auth_req: SerpapiTokenRequest, thread_id: str, profile: str,
+                future_uid: str, *args, **kwargs) -> str:
+        redirect_uri = urljoin(
+            config().public_base_url + "/",
+            f"{config().callback_url_rewrite_prefix}/auth/serpapi/token/callback",
+        )
+        url = self._make_auth_url(auth_req=auth_req, redirect_uri=redirect_uri, state=future_uid)
+        FutureStore.create_future(future_uid, data={
+            "redirect_uri": redirect_uri,
+            "thread_id": thread_id,
+            "profile": profile,
+        })
+
+        return f'User needs to authenticate using the following URL: {url}'
+
+    async def authenticate(self, auth_req: SerpapiTokenRequest, future_uid: str, *args, **kwargs) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        access_token = await future_data.future
+
+        response = SerpapiTokenResponse(access_token=access_token)
+        context = SerpapiTokenAuthContext.from_serpapi_token_response(response)
+
+        return context
+
+    async def refresh(self, auth_req: SerpapiTokenRequest, context: AuthContext, *args, **kwargs) -> AuthContext:
+        raise Exception("Serpapi token doesn't support refresh")
+
+    def _make_auth_url(self, auth_req: SerpapiTokenRequest, redirect_uri: str, state: str):
+        params = {
+            "redirect_uri": redirect_uri,
+            "state": state,
+        }
+        auth_url = f"{self._TOKEN_URL}?{urlencode(params)}"
+        return auth_url
+
+    def make_request(self, auth_scopes: Optional[list[str]] = None, **kwargs) -> SerpapiTokenRequest:
+        return SerpapiTokenRequest()

hyperpocket/auth/serpapi/token_schema.py

@@ -0,0 +1,11 @@
+from typing import List, Optional
+from pydantic import BaseModel
+from hyperpocket.auth.schema import AuthenticateRequest, AuthenticateResponse
+
+
+class SerpapiTokenRequest(AuthenticateRequest):
+    pass
+
+
+class SerpapiTokenResponse(AuthenticateResponse):
+    access_token: str

hyperpocket/server/auth/apitoken.py

@@ -0,0 +1,16 @@
+from fastapi import APIRouter
+from starlette.responses import HTMLResponse
+
+from hyperpocket.futures import FutureStore
+
+api_token_auth_router = APIRouter(prefix="/apitoken")
+
+
+@api_token_auth_router.get("/token/callback")
+async def notion_token_callback(state: str, token: str):
+    try:
+        FutureStore.resolve_future(state, token)
+    except ValueError:
+        return HTMLResponse(content="failed")
+
+    return HTMLResponse(content="success")

hyperpocket/server/auth/serpapi.py

@@ -0,0 +1,14 @@
+
+from fastapi import APIRouter
+from starlette.responses import HTMLResponse
+from hyperpocket.futures import FutureStore
+serpapi_auth_router = APIRouter(
+    prefix="/serpapi"
+)
+@serpapi_auth_router.get("/token/callback")
+async def serpapi_token_callback(state: str, token: str):
+    try:
+        FutureStore.resolve_future(state, token)
+    except ValueError:
+        return HTMLResponse(content="failed")
+    return HTMLResponse(content="success")

{hyperpocket-0.3.3.dist-info → hyperpocket-0.3.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.3.3
+Version: 0.3.4
 Summary: Building AI agent with hyperpocket tool in a flash
 Project-URL: Homepage, https://vessl-ai.github.io/hyperpocket
 Project-URL: Repository, https://github.com/vessl-ai/hyperpocket

{hyperpocket-0.3.3.dist-info → hyperpocket-0.3.4.dist-info}/RECORD

@@ -10,8 +10,14 @@ hyperpocket/auth/README.md,sha256=zn4QqnFZCA_4X3x8Wb6lE3OP5otYxpByZaCiUkBvaNs,11
 hyperpocket/auth/__init__.py,sha256=pO8M6SAuq0EPqi848_Iy650wqaLekx98e3RRnEAM_r0,607
 hyperpocket/auth/context.py,sha256=m-j2gDYUKBMsiakLHsu9thhM4dYyFiXP0Wp0S_iC0bU,1303
 hyperpocket/auth/handler.py,sha256=5cusl9ANEyG3gORVFjqh709txC0alw6eKtxgV6wjf6k,6683
-hyperpocket/auth/provider.py,sha256=Fe5kYX093sd9K0fakPVXYqXL6oIhdPU530W8pkVzZHU,515
+hyperpocket/auth/provider.py,sha256=oy00TuxaQk51nsXnOimLvY3gQzn-jOFgIywSlGuWa1Q,565
 hyperpocket/auth/schema.py,sha256=pl4oRTNj8PdqQg6UVPWf8ei2uYQ4DtOmmD58cVFMYQw,537
+hyperpocket/auth/apitoken/README.md,sha256=Sq5w_v3_JVp2k3Jaywmx_y_-ytw2Cs_oEeYveoXZVA0,564
+hyperpocket/auth/apitoken/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperpocket/auth/apitoken/context.py,sha256=uIXQsR7iiJEETOMcAy7s74Fob8N55FqWfUMQPfWGKY4,426
+hyperpocket/auth/apitoken/token_context.py,sha256=3Uet_XfJBi7p8fkgwkT63uFJozq84TzFRpziUEoQVac,439
+hyperpocket/auth/apitoken/token_handler.py,sha256=-BrBgkrPegNaA-03I1I7pBtHHfImbqM6UtlHVR-2zmY,2813
+hyperpocket/auth/apitoken/token_schema.py,sha256=MGaOcRzuobpAPyMBjbwvy32RHhjOOr7D0lXIlkpX7jM,203
 hyperpocket/auth/calendly/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/calendly/context.py,sha256=oUQzLy_w5NPvbXjFPOhLD-ggONVSy9VVpFabvX4ZpsY,408
 hyperpocket/auth/calendly/oauth2_context.py,sha256=SiMeZaoWh3aneZVH_MXsT_cZ8qCSVeL-pWHVbfYUS6g,848
@@ -51,6 +57,11 @@ hyperpocket/auth/reddit/context.py,sha256=WlOvIb4u1PvfrbkYbEq5cSRnXxmIEFuHa4LlKj
 hyperpocket/auth/reddit/oauth2_context.py,sha256=v1u_uy6O52REp2anteTPNgvb_nX-hI6bM7jNngBZXq4,1015
 hyperpocket/auth/reddit/oauth2_handler.py,sha256=X_lzoyhYnjEFriBuorGdn977uD5GjphPiXi8W9SYdy0,5166
 hyperpocket/auth/reddit/oauth2_schema.py,sha256=MOcA8VTvjjwewxhyIn_-DRNf6H_5GjsN6C7gV8sDyiI,436
+hyperpocket/auth/serpapi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperpocket/auth/serpapi/context.py,sha256=c1CuL53q22kyewrdc4Ae3zNclR87i_WWZ3n46aY3uoA,429
+hyperpocket/auth/serpapi/token_context.py,sha256=0sK9FSAA4takB02EdHEiYvLs46aN_Es8QZ7e4nQo3ec,479
+hyperpocket/auth/serpapi/token_handler.py,sha256=JFXm99eQUmMWfOhJcIqiJ4DIQEMKePou-Pft17PRkFA,2710
+hyperpocket/auth/serpapi/token_schema.py,sha256=gJ2ONKEwVI057U2lyFD7RQJs3HVSoCwN2lMxiU4mEOU,276
 hyperpocket/auth/slack/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/slack/context.py,sha256=f8dMs7QtSSyissOlPGw8rCZXsVUckFnZWjZ_LAmNXpA,429
 hyperpocket/auth/slack/oauth2_context.py,sha256=kwahsgxtDo5Wg_nrF8mFv16ijIu6biiGe_qz2JlEKjM,1398
@@ -100,6 +111,7 @@ hyperpocket/server/__init__.py,sha256=8laNl098-IkFkarqwOZs_gm6yG0q3NAObq4cCUxK_T
 hyperpocket/server/proxy.py,sha256=ItM1qfuUBs6QiPqIpErBWUZWdtsEf0twPJY5r2UUFBU,2042
 hyperpocket/server/server.py,sha256=KQxspdmw7SQtJpXZYbwaE5z9QwQRAQgXGaixYi6JlI0,8756
 hyperpocket/server/auth/__init__.py,sha256=IMjz9PCzD7qh4QIf2g-gWIdkDeU36jt-9F55vaHvLoM,286
+hyperpocket/server/auth/apitoken.py,sha256=I-ZDEehzwh7WsoHBnNTW1MZDRcmzNucNVBvGNHFjaao,451
 hyperpocket/server/auth/calendly.py,sha256=mi9_ysn0SffhnEgaoNa2jcHWCcD_yzqkS0rvzpJG2Qg,478
 hyperpocket/server/auth/github.py,sha256=5ZdKYoq_1Bh6aLUbSB5RWvr3o5DihZ5Gz5AfRWd4aCg,761
 hyperpocket/server/auth/google.py,sha256=_C33Bzow7yzjnvf0WUObu8p-dSU5rXqBgSPrENORAmY,470
@@ -107,6 +119,7 @@ hyperpocket/server/auth/gumloop.py,sha256=t_BUQ7RR8_x-ogqXnQ3iNHTVthzJyUTN03iqtR
 hyperpocket/server/auth/linear.py,sha256=Awd7GC5dIrwKnVthjW_MxutYIUz38DgC9L7BUcVfB04,442
 hyperpocket/server/auth/notion.py,sha256=NUKwCK61UADhO4FmFpNYjh1vsDzFZoFRDHdcMIBGb6M,443
 hyperpocket/server/auth/reddit.py,sha256=UloBQNEdhU1_eXaF8qVSQ5ivq5xtWINsHvBDGwgFSLM,443
+hyperpocket/server/auth/serpapi.py,sha256=YCyUAYXNEoXQ3nkPiyrDs-WguUo2lfO-pEZTpHoWwr8,448
 hyperpocket/server/auth/slack.py,sha256=frMNtBEfe3fbeNzsQ8udapeau45NZmS8guATmS46qyA,710
 hyperpocket/server/auth/token.py,sha256=Yq5Ym-uEO_3cBpQOsmCBuqtFIdImrNcVUgF5ozs5NHk,1763
 hyperpocket/server/auth/x.py,sha256=CYCD_ajBY6Jt04E2bSEBZFRRIUZmNjF2gn6F0ZV5XuA,450
@@ -144,7 +157,7 @@ hyperpocket/util/flatten_json_schema.py,sha256=iuNBEmMSKFtPi-uqo6fb3RWN0koHOAihW
 hyperpocket/util/function_to_model.py,sha256=TXUs-qPbzL8C9-qqpz4Ad4D9MOPP61n_p0iPU6SoBeM,2318
 hyperpocket/util/get_objects_from_subpackage.py,sha256=4mR_S8eaJSdU68YfCkiXeIcXxb6q7LjFGsY_IHeNIZw,929
 hyperpocket/util/json_schema_to_model.py,sha256=PqI87pU5dWwcrQWB8eQxRdfgAEvvC1x_DKZnhcsRV-o,3586
-hyperpocket-0.3.3.dist-info/METADATA,sha256=SyPQKiQ-LsCfP9vHGwHw-PQOz_wlwjh2o2lSSngYn8w,12958
-hyperpocket-0.3.3.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket-0.3.3.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
-hyperpocket-0.3.3.dist-info/RECORD,,
+hyperpocket-0.3.4.dist-info/METADATA,sha256=x-XIo1chvunVMxpg7jiSok5x6W1cWj2QAFjMlo_g0zU,12958
+hyperpocket-0.3.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket-0.3.4.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
+hyperpocket-0.3.4.dist-info/RECORD,,