hyperpocket-langgraph 0.0.1__py3-none-any.whl → 0.1.9__py3-none-any.whl

hyperpocket_langgraph/pocket_langgraph.py

@@ -1,17 +1,14 @@
 import copy
-from typing import List, Any
+from typing import Optional
 
 from langchain_core.runnables import RunnableConfig
 from langgraph.errors import NodeInterrupt
 from pydantic import BaseModel
 
 from hyperpocket.config import pocket_logger
-from hyperpocket.pocket_main import ToolLike
-from hyperpocket.prompts import pocket_extended_tool_description
-from hyperpocket.server.server import PocketServerOperations
 
 try:
-    from langchain_core.messages import AnyMessage, ToolMessage, AIMessage, RemoveMessage, SystemMessage
+    from langchain_core.messages import ToolMessage
     from langchain_core.tools import BaseTool, StructuredTool
 except ImportError:
     raise ImportError(
@@ -19,15 +16,13 @@ except ImportError:
     )
 
 try:
-    from langgraph.constants import START, END
-    from langgraph.graph import add_messages, StateGraph, MessagesState
-    from langgraph.graph.state import CompiledStateGraph
+    from langgraph.graph import MessagesState
 except ImportError:
     raise ImportError(
         "You need to install langgraph to use pocket langgraph"
     )
 
-from hyperpocket import Pocket, PocketAuth
+from hyperpocket import Pocket
 from hyperpocket.tool import Tool as PocketTool
 
 
@@ -36,20 +31,16 @@ class PocketLanggraphBaseState(MessagesState):
 
 
 class PocketLanggraph(Pocket):
-    langgraph_tools: dict[str, BaseTool]
+    def get_tools(self, use_profile: Optional[bool] = None):
+        if use_profile is not None:
+            self.use_profile = use_profile
+        return [self._get_langgraph_tool(tool_impl) for tool_impl in
+                self.core.tools.values()]
 
-    def __init__(self,
-                 tools: List[ToolLike],
-                 auth: PocketAuth = None):
-        super().__init__(tools, auth)
-        self.langgraph_tools = {}
-        for tool_name, tool_impl in self.tools.items():
-            self.langgraph_tools[tool_name] = self._get_langgraph_tool(tool_impl)
+    def get_tool_node(self, should_interrupt: bool = False, use_profile: Optional[bool] = None):
+        if use_profile is not None:
+            self.use_profile = use_profile
 
-    def get_tools(self):
-        return [v for v in self.langgraph_tools.values()]
-
-    def get_tool_node(self, should_interrupt: bool = False):
         async def _tool_node(state: PocketLanggraphBaseState, config: RunnableConfig) -> dict:
             thread_id = config.get("configurable", {}).get("thread_id", "default")
             last_message = state['messages'][-1]
@@ -66,12 +57,19 @@ class PocketLanggraph(Pocket):
                 tool_call_id = _tool_call['id']
                 tool_name = _tool_call['name']
                 tool_args = _tool_call['args']
-                body = tool_args.pop('body')
+
+                if self.use_profile:
+                    body = tool_args.pop("body")
+                    profile = tool_args.pop("profile", "default")
+                else:
+                    body = tool_args
+                    profile = "default"
+
                 if isinstance(body, BaseModel):
                     body = body.model_dump()
 
                 prepare = await self.prepare_in_subprocess(tool_name, body=body, thread_id=thread_id,
-                                                           profile=tool_args.get("profile", "default"))
+                                                           profile=profile)
                 need_prepare |= True if prepare else False
 
                 if prepare is None:
@@ -101,13 +99,19 @@ class PocketLanggraph(Pocket):
                 tool_call_id = _tool_call['id']
                 tool_name = _tool_call['name']
                 tool_args = _tool_call['args']
-                body = tool_args.pop('body')
+                if self.use_profile:
+                    body = tool_args.pop("body")
+                    profile = tool_args.pop("profile", "default")
+                else:
+                    body = tool_args
+                    profile = "default"
+
                 if isinstance(body, BaseModel):
                     body = body.model_dump()
 
                 try:
                     auth = await self.authenticate_in_subprocess(
-                        tool_name, body=body, thread_id=thread_id, profile=tool_args.get("profile", "default"))
+                        tool_name, body=body, thread_id=thread_id, profile=profile)
                 except Exception as e:
                     pocket_logger.error(f"occur exception during authenticate. error : {e}")
                     tool_messages.append(
@@ -133,79 +137,39 @@ class PocketLanggraph(Pocket):
 
         return _tool_node
 
-    async def prepare_in_subprocess(self,
-                                    tool_name: str,
-                                    body: Any,
-                                    thread_id: str = 'default',
-                                    profile: str = 'default',
-                                    *args, **kwargs):
-        prepare = await self.server.call_in_subprocess(
-            PocketServerOperations.PREPARE_AUTH,
-            args,
-            {
-                'tool_name': tool_name,
-                'body': body,
-                'thread_id': thread_id,
-                'profile': profile,
-                **kwargs,
-            },
-        )
-
-        return prepare
-
-    async def authenticate_in_subprocess(self,
-                                         tool_name: str,
-                                         body: Any,
-                                         thread_id: str = 'default',
-                                         profile: str = 'default',
-                                         *args, **kwargs):
-        credentials = await self.server.call_in_subprocess(
-            PocketServerOperations.AUTHENTICATE,
-            args,
-            {
-                'tool_name': tool_name,
-                'body': body,
-                'thread_id': thread_id,
-                'profile': profile,
-                **kwargs,
-            },
-        )
-
-        return credentials
-
-    async def tool_call_in_subprocess(self,
-                                      tool_name: str,
-                                      body: Any,
-                                      thread_id: str = 'default',
-                                      profile: str = 'default',
-                                      *args, **kwargs):
-        result = await self.server.call_in_subprocess(
-            PocketServerOperations.TOOL_CALL,
-            args,
-            {
-                'tool_name': tool_name,
-                'body': body,
-                'thread_id': thread_id,
-                'profile': profile,
-                **kwargs,
-            },
-        )
-
-        return result
-
     def _get_langgraph_tool(self, pocket_tool: PocketTool) -> BaseTool:
-        def _invoke(body: Any, thread_id: str = 'default', profile: str = 'default', **kwargs) -> str:
+        def _invoke(**kwargs) -> str:
+            if self.use_profile:
+                body = kwargs["body"]
+                thread_id = kwargs.pop("thread_id", "default")
+                profile = kwargs.pop("profile", "default")
+            else:
+                body = kwargs
+                thread_id = "default"
+                profile = "default"
+
             if isinstance(body, BaseModel):
                 body = body.model_dump()
+
             result, interrupted = self.invoke_with_state(pocket_tool.name, body, thread_id, profile, **kwargs)
             say = result
             if interrupted:
                 say = f'{say}\n\nThe tool execution interrupted. Please talk to me to resume.'
             return say
 
-        async def _ainvoke(body: Any, thread_id: str = 'default', profile: str = 'default', **kwargs) -> str:
+        async def _ainvoke(**kwargs) -> str:
+            if self.use_profile:
+                body = kwargs["body"]
+                thread_id = kwargs.pop("thread_id", "default")
+                profile = kwargs.pop("profile", "default")
+            else:
+                body = kwargs
+                thread_id = "default"
+                profile = "default"
+
             if isinstance(body, BaseModel):
                 body = body.model_dump()
+
             result, interrupted = await self.ainvoke_with_state(pocket_tool.name, body, thread_id, profile, **kwargs)
             say = result
             if interrupted:
@@ -216,6 +180,6 @@ class PocketLanggraph(Pocket):
             func=_invoke,
             coroutine=_ainvoke,
             name=pocket_tool.name,
-            description=pocket_extended_tool_description(pocket_tool.description),
-            args_schema=pocket_tool.schema_model(),
+            description=pocket_tool.get_description(use_profile=self.use_profile),
+            args_schema=pocket_tool.schema_model(use_profile=self.use_profile),
         )

{hyperpocket_langgraph-0.0.1.dist-info → hyperpocket_langgraph-0.1.9.dist-info}/METADATA

@@ -1,16 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: hyperpocket-langgraph
-Version: 0.0.1
-Summary: 
-Author: moon
-Author-email: moon@vessl.ai
-Requires-Python: >=3.11,<4.0
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Requires-Dist: hyperpocket (==0.0.3)
-Requires-Dist: langgraph (>=0.2.59,<0.3.0)
+Version: 0.1.9
+Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
+Requires-Python: >=3.10
+Requires-Dist: hyperpocket>=0.0.3
+Requires-Dist: langgraph>=0.2.59
 Description-Content-Type: text/markdown
 
 # Langgraph extensions
@@ -23,12 +17,13 @@ Description-Content-Type: text/markdown
 ## Get Pocket Subgraph
 
 ```python
-import hyperpocket as pk
-from pocket_langgraph import PocketLanggraph
+from hyperpocket.tool import from_git
+
+from hyperpocket_langgraph import PocketLanggraph
 
 pocket = PocketLanggraph(tools=[
-  *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-  *pk.curated_tools.LINEAR,
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/get-message"),
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/post-message"),
   "https://github.com/my-org/some-awesome-tool",
 ])
 
@@ -41,15 +36,14 @@ pocket_node = pocket.get_tool_node()
 ```python
 import os
 
+from hyperpocket.tool import from_git
 from langchain_openai import ChatOpenAI
 
-import hyperpocket as pk
-from pocket_langgraph import PocketLanggraph
+from hyperpocket_langgraph import PocketLanggraph
 
 pocket = PocketLanggraph(tools=[
-  *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-  *pk.curated_tools.LINEAR,
-  "https://github.com/my-org/some-awesome-tool",
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/get-message"),
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/post-message"),
 ])
 
 # get tools from pocket to bind llm
@@ -66,6 +60,7 @@ llm_with_tools = llm.bind_tools(tools)
 import os
 from typing import Annotated
 
+from hyperpocket.tool import from_git
 from langchain_core.runnables import RunnableConfig
 from langchain_openai import ChatOpenAI
 from langgraph.checkpoint.memory import MemorySaver
@@ -74,14 +69,12 @@ from langgraph.graph.message import add_messages
 from langgraph.prebuilt import tools_condition
 from typing_extensions import TypedDict
 
-import hyperpocket as pk
-from pocket_langgraph import PocketLanggraph
+from hyperpocket_langgraph import PocketLanggraph
 
 # Define pocket tools
 pocket = PocketLanggraph(tools=[
-  *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-  *pk.curated_tools.LINEAR,
-  "https://github.com/my-org/some-awesome-tool",
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/get-message"),
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/post-message"),
 ])
 
 # Get Pocket ToolNode
@@ -138,14 +131,14 @@ by setting the `should_interrupt` flag when calling `get_tool_node`
 Perform authentication in a multi-turn way
 
 ```python
-import hyperpocket as pk
-from pocket_langgraph import PocketLanggraph
+from hyperpocket.tool import from_git
+
+from hyperpocket_langgraph import PocketLanggraph
 
 # Define pocket tools
 pocket = PocketLanggraph(tools=[
-  *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-  *pk.curated_tools.LINEAR,
-  "https://github.com/my-org/some-awesome-tool",
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/get-message"),
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/post-message"),
 ])
 
 # Get Pocket ToolNode
@@ -155,14 +148,14 @@ pocket_node = pocket.get_tool_node(should_interrupt=False)  # multi turn
 ### Human-in-the-loop Auth
 
 ```python
-import hyperpocket as pk
-from pocket_langgraph import PocketLanggraph
+from hyperpocket.tool import from_git
+
+from hyperpocket_langgraph import PocketLanggraph
 
 # Define pocket tools
 pocket = PocketLanggraph(tools=[
-  *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-  *pk.curated_tools.LINEAR,
-  "https://github.com/my-org/some-awesome-tool",
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/get-message"),
+  from_git("https://github.com/vessl-ai/hyperawesometools", "main", "managed-tools/slack/post-message"),
 ])
 
 # Get Pocket ToolNode
@@ -198,4 +191,3 @@ flow, there is a risk of incorrect decisions being made.
 Conversely, the human-in-the-loop approach eliminates this risk by always continuing the previous flow as it is,
 offering more predictable and controlled behavior for managing the auth flow.
 
-

hyperpocket_langgraph-0.1.9.dist-info/RECORD

@@ -0,0 +1,5 @@
+hyperpocket_langgraph/__init__.py,sha256=0Az6FKsAvohOk8G4kUE99g6kyWz-Nf7dUuCuN3_pViY,97
+hyperpocket_langgraph/pocket_langgraph.py,sha256=y8GT64UrcriVr36gAf5NbTSFfnefl3NrcGnzFKqz6qY,7540
+hyperpocket_langgraph-0.1.9.dist-info/METADATA,sha256=mYL-Xfo7WLAqXv6tGz4C0tr5-FrGFR076Wzvckhqzk0,5846
+hyperpocket_langgraph-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket_langgraph-0.1.9.dist-info/RECORD,,

{hyperpocket_langgraph-0.0.1.dist-info → hyperpocket_langgraph-0.1.9.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: hatchling 1.27.0
 Root-Is-Purelib: true
 Tag: py3-none-any

hyperpocket_langgraph-0.0.1.dist-info/RECORD

@@ -1,5 +0,0 @@
-hyperpocket_langgraph/__init__.py,sha256=0Az6FKsAvohOk8G4kUE99g6kyWz-Nf7dUuCuN3_pViY,97
-hyperpocket_langgraph/pocket_langgraph.py,sha256=SqXZDdqvOKX66f0rRI8Ol-rNkL66yIXLFlToyorMvMs,9176
-hyperpocket_langgraph-0.0.1.dist-info/METADATA,sha256=9zG8UMulVdv3cVcSMHiOXzauJov9Jldx44myu1PRQeg,5671
-hyperpocket_langgraph-0.0.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-hyperpocket_langgraph-0.0.1.dist-info/RECORD,,