hyperpocket-langgraph 0.0.1__tar.gz → 0.1.9__tar.gz

hyperpocket_langgraph-0.1.9/.gitignore

@@ -0,0 +1,157 @@
+.vs/
+.vscode/
+.idea/
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+pip-wheel-metadata/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# Google GitHub Actions credentials files created by:
+# https://github.com/google-github-actions/auth
+#
+# That action recommends adding this gitignore to prevent accidentally committing keys.
+gha-creds-*.json
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+
+# Jupyter Notebook
+.ipynb_checkpoints
+notebooks/
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+.python-version
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.envrc
+.venv*
+venv*
+env/
+ENV/
+env.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.mypy_cache_test/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# macOS display setting files
+.DS_Store
+
+# Wandb directory
+wandb/
+
+# asdf tool versions
+.tool-versions
+/.ruff_cache/
+
+*.pkl
+*.bin
+
+# integration test artifacts
+data_map*
+\[('_type', 'fake'), ('stop', None)]
+
+# Replit files
+*replit*
+
+
+prof
+virtualenv/
+
+# Ignore dynaconf secret files
+**/.secrets.toml
+*.xml
+*.iml
+pocket.lock
+.log/
+.pocket/
+_build/

{hyperpocket_langgraph-0.0.1 → hyperpocket_langgraph-0.1.9}/PKG-INFO

@@ -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.0.1 → hyperpocket_langgraph-0.1.9}/README.md

@@ -8,12 +8,13 @@
 ## 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",
 ])
 
@@ -26,15 +27,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
@@ -51,6 +51,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
@@ -59,14 +60,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
@@ -123,14 +122,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
@@ -140,14 +139,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

{hyperpocket_langgraph-0.0.1 → hyperpocket_langgraph-0.1.9}/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.1.9/pyproject.toml

@@ -0,0 +1,42 @@
+
+[project]
+name = "hyperpocket-langgraph"
+version = "0.1.9"
+description = ""
+authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
+requires-python = ">=3.10"
+readme = "README.md"
+dependencies = ["langgraph>=0.2.59", "hyperpocket>=0.0.3"]
+
+[tool.uv.sources]
+hyperpocket = { path = "../../hyperpocket", editable = true }
+
+[dependency-groups]
+dev = [
+    "pytest>=8.3.4",
+    "ruff>=0.8.6",
+]
+test = [
+    "langchain-openai>=0.3.1",
+]
+
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[tool.ruff.lint]
+select = [
+    "E", # pycodestyle errors,
+    "F", # pyflakes errors,
+    "I", # isort errors,
+]
+ignore = [
+    "E501", # line too long, handled by formatting
+]
+
+[tool.ruff]
+line-length = 88
+target-version = "py310"
+
+[tool.ruff.lint.per-file-ignores]
+"__init__.py" = ["F401"]