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

hyperpocket_langgraph/pocket_langgraph.py

@@ -8,10 +8,9 @@ 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,9 +18,7 @@ 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"
@@ -43,7 +40,7 @@ class PocketLanggraph(Pocket):
                  auth: PocketAuth = None):
         super().__init__(tools, auth)
         self.langgraph_tools = {}
-        for tool_name, tool_impl in self.tools.items():
+        for tool_name, tool_impl in self.core.tools.items():
             self.langgraph_tools[tool_name] = self._get_langgraph_tool(tool_impl)
 
     def get_tools(self):
@@ -133,66 +130,6 @@ 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:
             if isinstance(body, BaseModel):

{hyperpocket_langgraph-0.0.1.dist-info → hyperpocket_langgraph-0.1.8.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.8
+Author-email: moon <moon@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.8.dist-info/RECORD

@@ -0,0 +1,5 @@
+hyperpocket_langgraph/__init__.py,sha256=0Az6FKsAvohOk8G4kUE99g6kyWz-Nf7dUuCuN3_pViY,97
+hyperpocket_langgraph/pocket_langgraph.py,sha256=6_KfStimtuvW7W7TzyUWN4WJ5veybDgJv_bGCAtxOt4,6852
+hyperpocket_langgraph-0.1.8.dist-info/METADATA,sha256=umosqoSEHq85vwsT7JiLfCMxAasNxLsvLFinxQ_sjeI,5827
+hyperpocket_langgraph-0.1.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket_langgraph-0.1.8.dist-info/RECORD,,

{hyperpocket_langgraph-0.0.1.dist-info → hyperpocket_langgraph-0.1.8.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,,