hyperpocket 0.4.4__py3-none-any.whl → 0.4.5__py3-none-any.whl

hyperpocket/auth/kraken/README.md

@@ -0,0 +1,8 @@
+# Valyu Authentication
+
+## Supported Authentication Types
+- Token
+
+## Additional Resources
+
+- [Valyu API Documentation](https://www.valyu.network/) 

hyperpocket/auth/kraken/context.py

@@ -0,0 +1,14 @@
+from hyperpocket.auth.context import AuthContext
+
+
+class KrakenAuthContext(AuthContext):
+    def to_dict(self) -> dict[str, str]:
+        return {
+            'KRAKEN_API_KEY': self.detail["KRAKEN_API_KEY"],
+            'KRAKEN_API_SECRET': self.detail["KRAKEN_API_SECRET"],
+        }
+
+    def to_profiled_dict(self, profile: str) -> dict[str, str]:
+        return {
+            f"{profile.upper()}_{self._ACCESS_TOKEN_KEY}": self.access_token,
+        }

hyperpocket/auth/kraken/keypair_context.py

@@ -0,0 +1,17 @@
+from hyperpocket.auth.kraken.context import KrakenAuthContext
+from hyperpocket.auth.kraken.keypair_schema import KrakenKeypairResponse
+
+
+class KrakenKeypairAuthContext(KrakenAuthContext):
+    @classmethod
+    def from_kraken_keypair_response(cls, response: KrakenKeypairResponse):
+        description = f"Kraken Keypair Context logged in"
+        return cls(
+            access_token="",
+            detail={
+                "KRAKEN_API_KEY": response.kraken_api_key,
+                "KRAKEN_API_SECRET": response.kraken_api_secret,
+            },
+            description=description,
+            expires_at=None,
+        )

hyperpocket/auth/kraken/keypair_handler.py

@@ -0,0 +1,97 @@
+from typing import Optional
+from urllib.parse import urljoin, urlencode
+
+from hyperpocket.auth import AuthProvider
+from hyperpocket.auth.ahrefs.token_schema import AhrefsTokenRequest
+from hyperpocket.auth.context import AuthContext
+from hyperpocket.auth.handler import AuthHandlerInterface
+from hyperpocket.auth.kraken.keypair_context import KrakenKeypairAuthContext
+from hyperpocket.auth.kraken.keypair_schema import KrakenKeypairRequest, KrakenKeypairResponse
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class KrakeyKeypairAuthHandler(AuthHandlerInterface):
+    name: str = "kraken-keypair"
+    description: str = (
+        "This handler is used to authenticate users using the Kraken keypair."
+    )
+    scoped: bool = False
+
+    _TOKEN_URL: str = urljoin(
+        config().public_base_url + "/",
+        f"{config().callback_url_rewrite_prefix}/auth/kraken/keypair",
+    )
+
+    @staticmethod
+    def provider() -> AuthProvider:
+        return AuthProvider.KRAKEN
+
+    @staticmethod
+    def provider_default() -> bool:
+        return True
+
+    @staticmethod
+    def recommended_scopes() -> set[str]:
+        return set()
+
+    def prepare(
+        self,
+        auth_req: AhrefsTokenRequest,
+        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/kraken/keypair/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: KrakenKeypairRequest, future_uid: str, *args, **kwargs
+    ) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        kraken_api_keypair = await future_data.future
+
+        response = KrakenKeypairResponse(
+            kraken_api_key=kraken_api_keypair["kraken_api_key"],
+            kraken_api_secret=kraken_api_keypair["kraken_api_secret"],
+        )
+        context = KrakenKeypairAuthContext.from_kraken_keypair_response(response)
+
+        return context
+
+    async def refresh(
+        self, auth_req: KrakenKeypairRequest, context: AuthContext, *args, **kwargs
+    ) -> AuthContext:
+        raise Exception("Kraken keypair doesn't support refresh")
+
+    def _make_auth_url(
+        self, auth_req: KrakenKeypairRequest, 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
+    ) -> KrakenKeypairRequest:
+        return KrakenKeypairRequest()

hyperpocket/auth/kraken/keypair_schema.py

@@ -0,0 +1,10 @@
+from hyperpocket.auth.schema import AuthenticateRequest, AuthenticateResponse
+
+
+class KrakenKeypairRequest(AuthenticateRequest):
+    pass
+
+
+class KrakenKeypairResponse(AuthenticateResponse):
+    kraken_api_key: str
+    kraken_api_secret: str

hyperpocket/auth/provider.py

@@ -74,6 +74,7 @@ class AuthProvider(Enum):
     SEMANTIC_SCHOLAR = "semantic_scholar"
     WEAVIATE = "weaviate"
     VALYU = "valyu"
+    KRAKEN = "kraken"
 
     @classmethod
     def get_auth_provider(cls, auth_provider_name: str) -> "AuthProvider":

hyperpocket/auth/valyu/token_handler.py

@@ -2,7 +2,6 @@ from typing import Optional
 from urllib.parse import urljoin, urlencode
 
 from hyperpocket.auth import AuthProvider
-from hyperpocket.auth.ahrefs.token_schema import AhrefsTokenRequest
 from hyperpocket.auth.context import AuthContext
 from hyperpocket.auth.handler import AuthHandlerInterface
 from hyperpocket.auth.valyu.token_context import ValyuTokenAuthContext
@@ -37,7 +36,7 @@ class ValyuTokenAuthHandler(AuthHandlerInterface):
 
     def prepare(
         self,
-        auth_req: AhrefsTokenRequest,
+        auth_req: ValyuTokenRequest,
         thread_id: str,
         profile: str,
         future_uid: str,

hyperpocket/pocket_main.py

@@ -30,10 +30,9 @@ class Pocket(object):
                 auth=auth,
             )
         except Exception as e:
-            if hasattr(self, "server"):
-                self.server.refcnt_down(self._uid)
+            self.teardown()
             pocket_logger.error(f"Failed to initialize pocket server. error : {e}")
-            self._teardown_server()
+            # self._teardown_server()
             raise e
         
         try:
@@ -337,15 +336,19 @@ class Pocket(object):
 
     def _teardown_server(self):
         self.server.teardown()
+    
+    def teardown(self):
+        if hasattr(self, 'server'):
+            self.server.refcnt_down(self._uid)
 
     def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
-        self.server.refcnt_down(self._uid)
+        self.teardown()
 
     def __del__(self):
-        self.server.refcnt_down(self._uid)
+        self.teardown()
 
     def __getstate__(self):
         state = self.__dict__.copy()

hyperpocket/server/auth/kraken.py

@@ -0,0 +1,58 @@
+from http import HTTPStatus
+
+from fastapi import APIRouter, Form
+from starlette.responses import HTMLResponse, RedirectResponse
+from hyperpocket.futures import FutureStore
+from hyperpocket.server.auth.token import add_query_params
+
+kraken_auth_router = APIRouter(prefix="/kraken")
+
+@kraken_auth_router.get("/keypair", response_class=HTMLResponse)
+async def keypair_form(redirect_uri: str, state: str = ""):
+    html = f"""    
+    <html>
+        <body>
+            <h2>Enter Token</h2>
+            <form action="submit" method="post">
+                <input type="hidden" name="redirect_uri" value="{redirect_uri}">
+                <input type="hidden" name="state" value="{state}">
+                
+                <label for="kraken_api_key">Kraken API Key:</label>
+                <input type="text" id="kraken_api_key" name="kraken_api_key" required>
+                
+                <label for="kraken_api_secret">Kraken API Secret:</label>
+                <input type="text" id="kraken_api_secret" name="kraken_api_secret" required>
+                
+                <button type="submit">submit</button>
+            </form>
+        </body>
+    </html>
+    """
+    return HTMLResponse(content=html)
+
+@kraken_auth_router.post("/submit", response_class=RedirectResponse)
+async def submit_keypair(
+    kraken_api_key: str = Form(...),
+    kraken_api_secret: str = Form(...),
+    redirect_uri: str = Form(...),
+    state: str = Form(...),
+):
+    new_callback_url = add_query_params(
+        redirect_uri, {
+            "kraken_api_key": kraken_api_key,
+            "kraken_api_secret": kraken_api_secret,
+            "state": state,
+        }
+    )
+    return RedirectResponse(url=new_callback_url, status_code=HTTPStatus.SEE_OTHER)
+
+@kraken_auth_router.get("/keypair/callback")
+async def kraken_keypair_callback(state: str, kraken_api_key: str, kraken_api_secret: str):
+    try:
+        FutureStore.resolve_future(state, {
+            "kraken_api_key": kraken_api_key,
+            "kraken_api_secret": kraken_api_secret,
+        })
+    except ValueError:
+        return HTMLResponse(content="failed")
+    return HTMLResponse(content="success")

hyperpocket/util/json_schema_to_model.py

@@ -1,4 +1,4 @@
-from typing import Type, Union
+from typing import Type, Union, Optional
 
 from pydantic import BaseModel, Field, create_model
 
@@ -42,7 +42,7 @@ def json_schema_to_model(
             fields[property_name] = (field_type, Field(description=field_description))
         else:
             fields[property_name] = (
-                field_type,
+                Optional[field_type],
                 Field(default=field_default, description=field_description),
             )
 

{hyperpocket-0.4.4.dist-info → hyperpocket-0.4.5.dist-info}/METADATA

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

@@ -3,14 +3,14 @@ hyperpocket/builtin.py,sha256=SOrVrNjoKadDMksfB1rt6pKreJFzHG2YGBsLGVsg72c,2385
 hyperpocket/constants.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/pocket_auth.py,sha256=VzpGpHOOQIiEgrk1sYg5DYa5WYV6gRQmFlm8Kb2C9V0,17611
 hyperpocket/pocket_core.py,sha256=WkY7Dp7KcJTORAFNuxDbZ5PDgicRANUtKkpZmI5gg6s,10021
-hyperpocket/pocket_main.py,sha256=XH0wlT1AafZDItgbYBkoo0dSb0z6Tcvl8AC9puXe8mg,10215
+hyperpocket/pocket_main.py,sha256=_YoxA01tFqnU3MavJndiXn1cKYWZ7z9_v0IihCzlHmo,10228
 hyperpocket/prompts.py,sha256=N1bCzCLZvGUVhH1Vn_cgeBPsdY3MdIU7ZGqVgexoj5E,472
 hyperpocket/tool_like.py,sha256=lUk9kUHPUmugqHCLbnWflKtZbCwtgcoYWtqGqUQtV38,187
 hyperpocket/auth/README.md,sha256=zn4QqnFZCA_4X3x8Wb6lE3OP5otYxpByZaCiUkBvaNs,11562
 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=ZKgDYiwI82cfKTcSvlLiYWV1DFwJJEldRj6ZACNmoeQ,2081
+hyperpocket/auth/provider.py,sha256=qZGenhTsL8vqd9zmUMM52TVlmdNNOTsmLmCPmZshUNg,2103
 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
@@ -240,6 +240,12 @@ hyperpocket/auth/klaviyo/context.py,sha256=JLALnthzihgMZkAQhKpzf8X07iHt0Iz2lEYsD
 hyperpocket/auth/klaviyo/token_context.py,sha256=OwckVoIbJaNH65WmIgi7C03TCHR0eCInA20RYmNX9Vc,455
 hyperpocket/auth/klaviyo/token_handler.py,sha256=0IiLFRhMhAm2LAMSGhkOoIoSiBbx0b776YTfB9VlJpw,2938
 hyperpocket/auth/klaviyo/token_schema.py,sha256=-x9e36JdNkMmgjRkXzzmpDBdeLsA_yA1wS2svhqNqlI,276
+hyperpocket/auth/kraken/README.md,sha256=QeSzmWSS86cXG1CmhAnGWLaVAeAo4ZpMWYwJ8OlCOMg,148
+hyperpocket/auth/kraken/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+hyperpocket/auth/kraken/context.py,sha256=LNa6UjosOg880PmBinubg8pHqjQWFqJ5sKBklZrhD38,455
+hyperpocket/auth/kraken/keypair_context.py,sha256=KyyJOlCE2wo3EIFFz5DGmHCPNb0tBOICvma9KrPyJHA,624
+hyperpocket/auth/kraken/keypair_handler.py,sha256=JdYLFV-S9IcCHQuKa2xYLh7oYAmxO29GQrcUvvmik_c,3123
+hyperpocket/auth/kraken/keypair_schema.py,sha256=xpZqmIg743Gf0OOfq9KNOg6Y-q5Yv67xOf9uBL7xq-0,242
 hyperpocket/auth/lever/README.md,sha256=00OuO_5cJ7RwZ4ZuWzYonSaQwnO4BYYycH3IvbBNgPM,166
 hyperpocket/auth/lever/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/lever/context.py,sha256=rNqjJyM4AToXEKT9DDKMdux2xk5vhwy6B6Fg84YCPIg,425
@@ -426,7 +432,7 @@ hyperpocket/auth/valyu/README.md,sha256=QeSzmWSS86cXG1CmhAnGWLaVAeAo4ZpMWYwJ8OlC
 hyperpocket/auth/valyu/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 hyperpocket/auth/valyu/context.py,sha256=He1n_Yo9CNOoOG_2ESKEJi8F5OphLbMruxK20zgUDHA,427
 hyperpocket/auth/valyu/token_context.py,sha256=0FKxYQUu4Zvu2gywcNdvRIEto4k_Iff9qQt_hE3-gd4,437
-hyperpocket/auth/valyu/token_handler.py,sha256=s5KkRezfEEXjVa7qtAz8OvGh8-iVTWxy980zYeyNHkU,2933
+hyperpocket/auth/valyu/token_handler.py,sha256=iDLEkB3dG5Ag8jm7xP7bkLL3wElfU6Xdyxjwr7P7YL8,2864
 hyperpocket/auth/valyu/token_schema.py,sha256=1yaf4fSREnTpvwmzKDrVh1oBxrwPSRnRPbKPOub-iPM,207
 hyperpocket/auth/wandb/README.md,sha256=Zsohbn6yELf8pfhBddDRm_iqs4NgQBZZhnC00yP5mVE,252
 hyperpocket/auth/wandb/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -533,6 +539,7 @@ hyperpocket/server/auth/heygen.py,sha256=dtHq_CkBkDFsj2QfebpLRuFIByVE9PXX3fe3z8P
 hyperpocket/server/auth/hubspot.py,sha256=6ApquQXemvjuFattNDALrz3932w20rFEOZaz_71bvBg,722
 hyperpocket/server/auth/jira.py,sha256=xYzQ7Ve-z90wvChoVF_7rWuez9vcjHXB3FdZEVshSXo,703
 hyperpocket/server/auth/klaviyo.py,sha256=5EShL4tD5JpeYh7lserpv_PFEC-fyn4QpKNNP8RUVJo,445
+hyperpocket/server/auth/kraken.py,sha256=vhKhEI8NQiaMOBWbw1dROBBOCia91AYXYRozX8TyDds,2146
 hyperpocket/server/auth/lever.py,sha256=AJjd852iUgfbudd-ILknC7HykgaKBKlMWfEUIWZ81ok,437
 hyperpocket/server/auth/lever_sandbox.py,sha256=RrhBvEZUAfj26Nj7Ic5MQAauyMmhT9rE__tkMhmAT6Y,469
 hyperpocket/server/auth/linear.py,sha256=vUW4TEqdOYqAyx9YLJC4-dVldPTDuIPXRvoo-a89ai8,988
@@ -593,8 +600,8 @@ hyperpocket/util/flatten_json_schema.py,sha256=iuNBEmMSKFtPi-uqo6fb3RWN0koHOAihW
 hyperpocket/util/function_to_model.py,sha256=TXUs-qPbzL8C9-qqpz4Ad4D9MOPP61n_p0iPU6SoBeM,2318
 hyperpocket/util/generate_slug.py,sha256=gwwf9gfTlqribrvybMQQj7VIHX5FspVoTm5-EYH-X74,150
 hyperpocket/util/get_objects_from_subpackage.py,sha256=4mR_S8eaJSdU68YfCkiXeIcXxb6q7LjFGsY_IHeNIZw,929
-hyperpocket/util/json_schema_to_model.py,sha256=BNTFYzBK-yZr2VkO6lAdZkXdZXfGSabzWa3DM6it4KI,3731
-hyperpocket-0.4.4.dist-info/METADATA,sha256=CItp4XP56pSM5F32k5YsOIuaWAjJHzWVORyCbnz5vLM,13074
-hyperpocket-0.4.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket-0.4.4.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
-hyperpocket-0.4.4.dist-info/RECORD,,
+hyperpocket/util/json_schema_to_model.py,sha256=nc5AmnqkrdeFLELu-7_O9sEAaUaD8_KGlvIMDRobt-4,3751
+hyperpocket-0.4.5.dist-info/METADATA,sha256=vFdzt3Dz_z0naddvIBneuu-KAqGYi1lwtnUaYnAKUmk,13074
+hyperpocket-0.4.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket-0.4.5.dist-info/entry_points.txt,sha256=KpBleaYr0SaENXOa-dFvJ_cvFCHYFEQ4LMl11ShAcBI,61
+hyperpocket-0.4.5.dist-info/RECORD,,