hyperpocket 0.4.3__tar.gz → 0.4.4__tar.gz

{hyperpocket-0.4.3 → hyperpocket-0.4.4}/PKG-INFO

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

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

hyperpocket-0.4.4/hyperpocket/auth/valyu/README.md

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

hyperpocket-0.4.4/hyperpocket/auth/valyu/context.py

@@ -0,0 +1,15 @@
+from hyperpocket.auth.context import AuthContext
+
+
+class ValyuAuthContext(AuthContext):
+    _ACCESS_TOKEN_KEY: str = "VALYU_API_KEY"
+
+    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.4.4/hyperpocket/auth/valyu/token_context.py

@@ -0,0 +1,11 @@
+from hyperpocket.auth.valyu.context import ValyuAuthContext
+from hyperpocket.auth.valyu.token_schema import ValyuTokenResponse
+
+
+class ValyuTokenAuthContext(ValyuAuthContext):
+    @classmethod
+    def from_valyu_token_response(cls, response: ValyuTokenResponse):
+        description = f"Valyu Token Context logged in"
+        return cls(
+            access_token=response.access_token, description=description, expires_at=None
+        )

hyperpocket-0.4.4/hyperpocket/auth/valyu/token_handler.py

@@ -0,0 +1,94 @@
+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
+from hyperpocket.auth.valyu.token_schema import ValyuTokenRequest, ValyuTokenResponse
+from hyperpocket.config import config
+from hyperpocket.futures import FutureStore
+
+
+class ValyuTokenAuthHandler(AuthHandlerInterface):
+    name: str = "valyu-token"
+    description: str = (
+        "This handler is used to authenticate users using the Valyu 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.VALYU
+
+    @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/valyu/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: ValyuTokenRequest, future_uid: str, *args, **kwargs
+    ) -> AuthContext:
+        future_data = FutureStore.get_future(future_uid)
+        access_token = await future_data.future
+
+        response = ValyuTokenResponse(access_token=access_token)
+        context = ValyuTokenAuthContext.from_valyu_token_response(response)
+
+        return context
+
+    async def refresh(
+        self, auth_req: ValyuTokenRequest, context: AuthContext, *args, **kwargs
+    ) -> AuthContext:
+        raise Exception("Valyu token doesn't support refresh")
+
+    def _make_auth_url(
+        self, auth_req: ValyuTokenRequest, 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
+    ) -> ValyuTokenRequest:
+        return ValyuTokenRequest()

hyperpocket-0.4.4/hyperpocket/auth/valyu/token_schema.py

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

hyperpocket-0.4.4/hyperpocket/server/auth/valyu.py

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

{hyperpocket-0.4.3 → hyperpocket-0.4.4}/hyperpocket/tool/function/annotation.py

@@ -1,3 +1,4 @@
+import inspect
 from typing import Callable, List, Optional
 
 from hyperpocket.auth import AuthProvider
@@ -24,6 +25,9 @@ def function_tool(
                 auth_handler=auth_handler,
             )
 
+        if inspect.iscoroutinefunction(inner_func):
+            return FunctionTool.from_func(func=inner_func, afunc=inner_func, auth=auth, tool_vars=tool_vars)
+
         return FunctionTool.from_func(func=inner_func, auth=auth, tool_vars=tool_vars)
 
     if func is not None:

{hyperpocket-0.4.3 → hyperpocket-0.4.4}/pyproject.toml

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