hyperpocket 0.3.5__py3-none-any.whl → 0.3.7__py3-none-any.whl

hyperpocket/auth/provider.py

@@ -70,6 +70,8 @@ class AuthProvider(Enum):
     LINKEDIN = "linkedin"
     WANDB = "wandb"
     APITOKEN = "apitoken"
+    ZINC = "zinc"
+    SEMANTIC_SCHOLAR = "semantic_scholar"
 
     @classmethod
     def get_auth_provider(cls, auth_provider_name: str) -> "AuthProvider":

hyperpocket/auth/semantic_scholar/context.py

@@ -0,0 +1,12 @@
+
+from hyperpocket.auth.context import AuthContext
+class SemanticScholarAuthContext(AuthContext):
+    _ACCESS_TOKEN_KEY: str = "SEMANTIC_SCHOLAR_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/semantic_scholar/token_context.py

@@ -0,0 +1,11 @@
+from hyperpocket.auth.semantic_scholar.context import SemanticScholarAuthContext
+from hyperpocket.auth.semantic_scholar.token_schema import SemanticScholarTokenResponse
+class SemanticScholarTokenAuthContext(SemanticScholarAuthContext):
+    @classmethod
+    def from_semantic_scholar_token_response(cls, response: SemanticScholarTokenResponse):
+        description = f'SemanticScholar Token Context logged in'
+        return cls(
+            access_token=response.access_token,
+            description=description,
+            expires_at=None
+        )

hyperpocket/auth/semantic_scholar/token_handler.py

@@ -0,0 +1,64 @@
+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.semantic_scholar.token_context import SemanticScholarTokenAuthContext
+from hyperpocket.auth.semantic_scholar.token_schema import SemanticScholarTokenResponse, SemanticScholarTokenRequest
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class SemanticScholarTokenAuthHandler(AuthHandlerInterface):
+    name: str = "semantic-scholar-token"
+    description: str = "This handler is used to authenticate users using the SemanticScholar 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.SEMANTIC_SCHOLAR
+
+    @staticmethod
+    def recommended_scopes() -> set[str]:
+        return set()
+
+    def prepare(self, auth_req: SemanticScholarTokenRequest, 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/semantic_scholar/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: SemanticScholarTokenRequest, future_uid: str, *args, **kwargs) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        access_token = await future_data.future
+
+        response = SemanticScholarTokenResponse(access_token=access_token)
+        context = SemanticScholarTokenAuthContext.from_semantic_scholar_token_response(response)
+
+        return context
+
+    async def refresh(self, auth_req: SemanticScholarTokenRequest, context: AuthContext, *args, **kwargs) -> AuthContext:
+        raise Exception("SemanticScholar token doesn't support refresh")
+
+    def _make_auth_url(self, auth_req: SemanticScholarTokenRequest, 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) -> SemanticScholarTokenRequest:
+        return SemanticScholarTokenRequest()

hyperpocket/auth/semantic_scholar/token_schema.py

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

hyperpocket/auth/zinc/context.py

@@ -0,0 +1,12 @@
+
+from hyperpocket.auth.context import AuthContext
+class ZincAuthContext(AuthContext):
+    _ACCESS_TOKEN_KEY: str = "ZINC_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/zinc/token_context.py

@@ -0,0 +1,11 @@
+from hyperpocket.auth.zinc.context import ZincAuthContext
+from hyperpocket.auth.zinc.token_schema import ZincTokenResponse
+class ZincTokenAuthContext(ZincAuthContext):
+    @classmethod
+    def from_zinc_token_response(cls, response: ZincTokenResponse):
+        description = f'Zinc Token Context logged in'
+        return cls(
+            access_token=response.access_token,
+            description=description,
+            expires_at=None
+        )

hyperpocket/auth/zinc/token_handler.py

@@ -0,0 +1,64 @@
+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.zinc.token_context import ZincTokenAuthContext
+from hyperpocket.auth.zinc.token_schema import ZincTokenResponse, ZincTokenRequest
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class ZincTokenAuthHandler(AuthHandlerInterface):
+    name: str = "zinc-token"
+    description: str = "This handler is used to authenticate users using the Zinc 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.ZINC
+
+    @staticmethod
+    def recommended_scopes() -> set[str]:
+        return set()
+
+    def prepare(self, auth_req: ZincTokenRequest, 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/zinc/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: ZincTokenRequest, future_uid: str, *args, **kwargs) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        access_token = await future_data.future
+
+        response = ZincTokenResponse(access_token=access_token)
+        context = ZincTokenAuthContext.from_zinc_token_response(response)
+
+        return context
+
+    async def refresh(self, auth_req: ZincTokenRequest, context: AuthContext, *args, **kwargs) -> AuthContext:
+        raise Exception("Zinc token doesn't support refresh")
+
+    def _make_auth_url(self, auth_req: ZincTokenRequest, 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) -> ZincTokenRequest:
+        return ZincTokenRequest()

hyperpocket/auth/zinc/token_schema.py

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

hyperpocket/server/auth/semantic_scholar.py

@@ -0,0 +1,27 @@
+from fastapi import APIRouter
+from starlette.responses import HTMLResponse
+from hyperpocket.futures import FutureStore
+
+semantic_scholar_auth_router = APIRouter(
+    prefix="/semantic_scholar"
+)
+
+
+@semantic_scholar_auth_router.get("/oauth2/callback")
+async def semantic_scholar_oauth2_callback(state: str, code: str):
+    try:
+        FutureStore.resolve_future(state, code)
+    except ValueError:
+        return HTMLResponse(content="failed")
+
+    return HTMLResponse(content="success")
+
+
+@semantic_scholar_auth_router.get("/token/callback")
+async def semantic_scholar_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/zinc.py

@@ -0,0 +1,27 @@
+from fastapi import APIRouter
+from starlette.responses import HTMLResponse
+from hyperpocket.futures import FutureStore
+
+zinc_auth_router = APIRouter(
+    prefix="/zinc"
+)
+
+
+@zinc_auth_router.get("/oauth2/callback")
+async def zinc_oauth2_callback(state: str, code: str):
+    try:
+        FutureStore.resolve_future(state, code)
+    except ValueError:
+        return HTMLResponse(content="failed")
+
+    return HTMLResponse(content="success")
+
+
+@zinc_auth_router.get("/token/callback")
+async def zinc_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.5.dist-info → hyperpocket-0.3.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.3.5
+Version: 0.3.7
 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.5.dist-info → hyperpocket-0.3.7.dist-info}/RECORD

@@ -10,7 +10,7 @@ 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=X2NseyRS_48Hwi1yUwezbvHkQZZUleqtF_1Z_XhnzzE,1975
+hyperpocket/auth/provider.py,sha256=bTJn6Atm4E9nm9vf8Ggudf7-XxAWMOIPo6xiZnGqDvM,2035
 hyperpocket/auth/schema.py,sha256=pl4oRTNj8PdqQg6UVPWf8ei2uYQ4DtOmmD58cVFMYQw,537
 hyperpocket/auth/activeloop/README.md,sha256=xp8n0itbY9VNt8XEvCAE4Ealvj4uf_f1uSZyWG4Q5FE,172
 hyperpocket/auth/activeloop/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -360,6 +360,11 @@ hyperpocket/auth/salesforce/context.py,sha256=JuxwHIBfubOi_7VWWy0d9tKN3XOV9bT2FY
 hyperpocket/auth/salesforce/oauth2_context.py,sha256=7STo2n6yIHK-QHeFGf0t7MSxlJwCwvX-tuB34mi3vRU,1088
 hyperpocket/auth/salesforce/oauth2_handler.py,sha256=2zZ58usN5DjxBbrynhN9DPMr2Vzmm8iAN8TpHEKuf-c,4717
 hyperpocket/auth/salesforce/oauth2_schema.py,sha256=XY26_-pxxD8XMvks34-_G-FJb2xitFDSesIiuztw628,477
+hyperpocket/auth/semantic_scholar/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperpocket/auth/semantic_scholar/context.py,sha256=wQzQLW0Hkv4DW1VC1bD6XZ4UsZw07f4rnuPmZFZKlac,442
+hyperpocket/auth/semantic_scholar/token_context.py,sha256=t1F5iF0-Jte2pvFP5C74lL_OLH3CwywerikPmD1D-9I,551
+hyperpocket/auth/semantic_scholar/token_handler.py,sha256=FTUBLEorMPJ8BjdL877SHRwIJ78dJK9ZeM2CpN511Lw,2800
+hyperpocket/auth/semantic_scholar/token_schema.py,sha256=Kw8H6DozQ_NGekODhPvWXr5HX73HbDJCP7YSh2WkKX8,287
 hyperpocket/auth/sendgrid/README.md,sha256=u_5Cg6OGV2EvwvV_jQp0ZmY1wvjuTAcVPaw_1W0k-WI,305
 hyperpocket/auth/sendgrid/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/sendgrid/context.py,sha256=uaJoo7lS2KEtab9fUC1FgECe3kH2EqEfdnXyBWZ6iKU,431
@@ -435,6 +440,11 @@ hyperpocket/auth/x/context.py,sha256=5Ur9GI8og49E-8FToKlqR3TtPn6vWAdu1boLqYxJL8k
 hyperpocket/auth/x/oauth2_context.py,sha256=q2HB14WENz2LU7Qe7KkE7-AtsEOtzXIj7n8alykac8M,955
 hyperpocket/auth/x/oauth2_handler.py,sha256=kEnsbL1YIeIQUlMVPIZYqOswk9_t-sNGiaixj43M1Yc,5634
 hyperpocket/auth/x/oauth2_schema.py,sha256=PKeaHccpxGLJjEd8RxP2wVcrIPCE-W9HEtP8EMzU4To,401
+hyperpocket/auth/zinc/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperpocket/auth/zinc/context.py,sha256=nsjUnNcOVUB2C3tjN2jT94dWPI9wRagtpyrV452pieY,419
+hyperpocket/auth/zinc/token_context.py,sha256=CG0aVRO-xEIk39zHLu2-IF46VMrKnKlxg4mqTtv2H-s,449
+hyperpocket/auth/zinc/token_handler.py,sha256=S0if3Zb8TFRP52XwHpanm_AbthkojLmsnKaeJU8RyxA,2574
+hyperpocket/auth/zinc/token_schema.py,sha256=18C50Wnr9Vq-YJBS4qzjUotoEJy5zLWqFKK2G7nmcMg,265
 hyperpocket/auth/zoom/README.md,sha256=xzfcoAiWyFxUdHfOgnXwMXygt_RkaAsaZuCH3ylTcVQ,401
 hyperpocket/auth/zoom/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/zoom/context.py,sha256=26S_aMbvtdzXYoXdDNpnHGJ-tmMcOSt0F15cMYwQZaM,423
@@ -533,6 +543,7 @@ hyperpocket/server/auth/posthog.py,sha256=Cdi9OeNn1icCeokrTbLc02oA2UMk-kx6nFTlYV
 hyperpocket/server/auth/ravenseotools.py,sha256=StkE9c4nrm-9ftzbaqQ8gNb8lN_N2iIYw7Hn-3o7Cb4,469
 hyperpocket/server/auth/reddit.py,sha256=UloBQNEdhU1_eXaF8qVSQ5ivq5xtWINsHvBDGwgFSLM,443
 hyperpocket/server/auth/salesforce.py,sha256=MxwrS_2XkMJXA1UuHayHBHKVGgdBA-zmOrkSJit0XJY,739
+hyperpocket/server/auth/semantic_scholar.py,sha256=H4FSPjHiZqLPI4eZU-zfjBt4nY3o4OoFtOhGZpP-9Ik,780
 hyperpocket/server/auth/sendgrid.py,sha256=rOMHV76lVArwsFLuVrJ-nbC4oR-Fq3Xk-Yu11tDl8nA,449
 hyperpocket/server/auth/serpapi.py,sha256=gHqEC93Sw3JI4-s9zxlL3FrvPEiGPLq5U7IaOKuXVWQ,445
 hyperpocket/server/auth/slack.py,sha256=frMNtBEfe3fbeNzsQ8udapeau45NZmS8guATmS46qyA,710
@@ -546,6 +557,7 @@ hyperpocket/server/auth/trello.py,sha256=1T3ez62h8-KMvu2Zj1Tyv2afgqLUbtG2LM4QbIS
 hyperpocket/server/auth/wandb.py,sha256=jfwg-lQVrr5TgEL7QETnaFucDvi057SJvVZlBe7cOqQ,709
 hyperpocket/server/auth/workiom.py,sha256=0apFB5-3rmcJaQeEl1_XZ_QlsZORdq8kixJBGkTpZ3Y,445
 hyperpocket/server/auth/x.py,sha256=CYCD_ajBY6Jt04E2bSEBZFRRIUZmNjF2gn6F0ZV5XuA,450
+hyperpocket/server/auth/zinc.py,sha256=xgVV5lWP6YSEHTCmH0gipG5lh4Wvf716tHTnOvTTj3w,708
 hyperpocket/server/auth/zoom.py,sha256=kWp4MB4i9FKMA4ZYcj4g4_90Y1ZMEoAQTTjbb0rsMRs,703
 hyperpocket/server/tool/__init__.py,sha256=khNLe3H2W7WXKQlHjXuuvd9R87eHOAZhDsQmjDcbYsg,210
 hyperpocket/server/tool/wasm.py,sha256=VJyp6RGsq8llKT_sY6DhV52wsETu-W9bzJ7C9wC17Oo,1698
@@ -581,7 +593,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.5.dist-info/METADATA,sha256=0tAI6rXugUdATKDBk6qvdwDIz1t23YsZN5fYkRfHsSQ,12958
-hyperpocket-0.3.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket-0.3.5.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
-hyperpocket-0.3.5.dist-info/RECORD,,
+hyperpocket-0.3.7.dist-info/METADATA,sha256=oZuYOmr2Tx0PdrjFsKiptOTJP5eUffygor8VdhNITOU,12958
+hyperpocket-0.3.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket-0.3.7.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
+hyperpocket-0.3.7.dist-info/RECORD,,