hyperpocket 0.3.5__tar.gz → 0.3.6__tar.gz

{hyperpocket-0.3.5 → hyperpocket-0.3.6}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.3.5
+Version: 0.3.6
 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 → hyperpocket-0.3.6}/hyperpocket/auth/provider.py

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

hyperpocket-0.3.6/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-0.3.6/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-0.3.6/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-0.3.6/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-0.3.6/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-0.3.5 → hyperpocket-0.3.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyperpocket"
-version = "0.3.5"
+version = "0.3.6"
 description = "Building AI agent with hyperpocket tool in a flash"
 authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
 requires-python = ">=3.10"