hyperpocket 0.4.5__tar.gz → 0.5.0__tar.gz

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket
-Version: 0.4.5
+Version: 0.5.0
 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
@@ -103,7 +103,7 @@ from hyperpocket_langchain import PocketLangchain
 if __name__ == '__main__':
     pocket = PocketLangchain(
         tools=[
-            "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+            "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
             "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
         ],
     )
@@ -176,7 +176,7 @@ from hyperpocket_langchain import PocketLangchain
 
 pklc = PocketLangchain(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
     ]
 )
@@ -235,7 +235,7 @@ from hyperpocket_langgraph import PocketLanggraph
 
 pklg = PocketLanggraph(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
     ],
 )
@@ -272,7 +272,7 @@ from hyperpocket_llamaindex import PocketLlamaindex
 llm = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 pocket = PocketLlamaindex(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/linear/get-issues",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/google/get-calendar-events",

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/README.md

@@ -75,7 +75,7 @@ from hyperpocket_langchain import PocketLangchain
 if __name__ == '__main__':
     pocket = PocketLangchain(
         tools=[
-            "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+            "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
             "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
         ],
     )
@@ -148,7 +148,7 @@ from hyperpocket_langchain import PocketLangchain
 
 pklc = PocketLangchain(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
     ]
 )
@@ -207,7 +207,7 @@ from hyperpocket_langgraph import PocketLanggraph
 
 pklg = PocketLanggraph(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
     ],
 )
@@ -244,7 +244,7 @@ from hyperpocket_llamaindex import PocketLlamaindex
 llm = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 pocket = PocketLlamaindex(
     tools=[
-        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-message",
+        "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/get-messages",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/slack/post-message",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/linear/get-issues",
         "https://github.com/vessl-ai/hyperpocket/tree/main/tools/google/get-calendar-events",

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/hyperpocket/builtin.py

@@ -12,7 +12,7 @@ def get_builtin_tools(pocket_auth: PocketAuth) -> List[Tool]:
     Builtin Tool can access to Pocket Core.
     """
 
-    def __get_current_thread_session_state(thread_id: str = "default") -> str:
+    async def __get_current_thread_session_state(thread_id: str = "default") -> str:
         """
         This tool retrieves the current session state list for the specified thread.
 
@@ -31,7 +31,7 @@ def get_builtin_tools(pocket_auth: PocketAuth) -> List[Tool]:
 
         This tool ensures transparency about the current session but must respect user-driven intent and should never be called automatically or without a specific user request.
         """
-        session_list = pocket_auth.list_session_state(thread_id)
+        session_list = await pocket_auth.list_session_state(thread_id)
         return str(session_list)
 
     def __delete_session(
@@ -58,7 +58,7 @@ def get_builtin_tools(pocket_auth: PocketAuth) -> List[Tool]:
         return str(is_deleted)
 
     builtin_tools = [
-        from_func(func=__get_current_thread_session_state),
+        from_func(func=__get_current_thread_session_state, afunc=__get_current_thread_session_state),
         from_func(func=__delete_session),
     ]
 

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/hyperpocket/cli/__main__.py

@@ -2,7 +2,6 @@ import click
 
 from hyperpocket.cli.eject import eject
 from hyperpocket.cli.pull import pull
-from hyperpocket.cli.sync import sync
 from hyperpocket.cli.eject import eject
 from hyperpocket.cli.auth_token import create_token_auth_template
 from hyperpocket.cli.auth_oauth2 import create_oauth2_auth_template
@@ -28,7 +27,6 @@ devtool.add_command(build_tool)
 devtool.add_command(export_tool)
 
 cli.add_command(pull)
-cli.add_command(sync)
 cli.add_command(eject)
 
 cli()

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/hyperpocket/config/logger.py

@@ -1,5 +1,6 @@
 import logging
 import os
+import pathlib
 from logging.handlers import RotatingFileHandler
 from pathlib import Path
 
@@ -27,9 +28,7 @@ class ColorFormatter(logging.Formatter):
 
 
 def get_logger():
-    # init log file
-    package_path = Path(os.path.dirname(hyperpocket.__file__))
-    log_dir = package_path / ".log"
+    log_dir = pathlib.Path(os.getcwd()) / ".log"
     os.makedirs(log_dir, exist_ok=True)
     log_file = log_dir / "pocket.log"
     if not log_file.exists():

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/hyperpocket/futures/futurestore.py

@@ -25,7 +25,6 @@ class FutureStore(object):
                 f"the future already exists. the existing future is returned. uid: {uid}"
             )
             return future
-
         loop = asyncio.get_running_loop()
         future = loop.create_future()
         future_data = FutureData(future=future, data=data)
@@ -42,7 +41,13 @@ class FutureStore(object):
         if not future_data:
             raise ValueError(f"Future not found for uid={uid}")
         if not future_data.future.done():
-            future_data.future.set_result(value)
+            # if the future loop is running, it should be executed in same event loop
+            loop = future_data.future.get_loop()
+            if loop.is_running():
+                loop.call_soon_threadsafe(future_data.future.set_result, value)
+            # if the future loop is not running, it can be executed from anywhere.
+            else:
+                future_data.future.set_result(value)
 
     def delete_future(self, uid: str):
         self.futures.pop(uid, None)

{hyperpocket-0.4.5 → hyperpocket-0.5.0}/hyperpocket/pocket_auth.py

@@ -78,7 +78,7 @@ class PocketAuth(object):
         handler = self.find_handler_instance(auth_handler_name, auth_provider)
         return handler.make_request(auth_scopes, **kwargs)
 
-    def check(
+    async def check(
         self,
         auth_req: AuthenticateRequest,
         auth_handler_name: Optional[str] = None,
@@ -112,12 +112,12 @@ class PocketAuth(object):
         """
         handler = self.find_handler_instance(auth_handler_name, auth_provider)
         session = self.session_storage.get(handler.provider(), thread_id, profile)
-        auth_state = self.get_session_state(session=session, auth_req=auth_req)
+        auth_state = await self.get_session_state(session=session, auth_req=auth_req)
 
         return auth_state
 
     @staticmethod
-    def get_session_state(
+    async def get_session_state(
         session: Optional[BaseSessionValue], auth_req: Optional[AuthenticateRequest]
     ) -> AuthState:
         if not session:
@@ -125,6 +125,8 @@ class PocketAuth(object):
 
         if session.auth_resolve_uid:
             future_data = FutureStore.get_future(session.auth_resolve_uid)
+            # it yields before checking future's state, because the future is being resolved on another thread's event loop.
+            await asyncio.sleep(0)
             if future_data is not None and future_data.future.done():
                 return AuthState.RESOLVED
 
@@ -140,7 +142,7 @@ class PocketAuth(object):
 
         return AuthState.SKIP_AUTH
 
-    def prepare(
+    async def prepare(
         self,
         auth_req: AuthenticateRequest,
         auth_handler_name: Optional[str] = None,
@@ -166,7 +168,7 @@ class PocketAuth(object):
         Returns:
             Optional[str]: authentication URL
         """
-        auth_state = self.check(
+        auth_state = await self.check(
             auth_req=auth_req,
             auth_handler_name=auth_handler_name,
             auth_provider=auth_provider,
@@ -263,7 +265,7 @@ class PocketAuth(object):
         Returns:
             AuthContext: authentication context
         """
-        auth_state = self.check(
+        auth_state = await self.check(
             auth_req=auth_req,
             auth_handler_name=auth_handler_name,
             auth_provider=auth_provider,
@@ -353,7 +355,7 @@ class PocketAuth(object):
 
         return session.auth_context
 
-    def list_session_state(
+    async def list_session_state(
         self, thread_id: str, auth_provider: Optional[AuthProvider] = None
     ):
         session_list = self.session_storage.get_by_thread_id(
@@ -361,7 +363,7 @@ class PocketAuth(object):
         )
         session_state_list = []
         for session in session_list:
-            state = self.get_session_state(session=session, auth_req=None)
+            state = await self.get_session_state(session=session, auth_req=None)
 
             session_state_list.append(
                 {