hyperpocket-anthropic 0.1.10__tar.gz → 0.2.0__tar.gz

{hyperpocket_anthropic-0.1.10 → hyperpocket_anthropic-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hyperpocket-anthropic
-Version: 0.1.10
+Version: 0.2.0
 Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
 Requires-Python: >=3.10
 Requires-Dist: anthropic>=0.40.0

{hyperpocket_anthropic-0.1.10 → hyperpocket_anthropic-0.2.0}/hyperpocket_anthropic/pocket_anthropic.py

@@ -36,7 +36,11 @@ class PocketAnthropic(Pocket):
             body = json.loads(body)
 
         result, interrupted = self.invoke_with_state(
-            tool_use_block.name, body=body, thread_id=thread_id, profile=profile, **kwargs
+            tool_use_block.name,
+            body=body,
+            thread_id=thread_id,
+            profile=profile,
+            **kwargs,
         )
         say = result
         if interrupted:
@@ -51,7 +55,7 @@ class PocketAnthropic(Pocket):
         return tool_result_block
 
     async def ainvoke(
-            self, tool_use_block: ToolUseBlock, **kwargs
+        self, tool_use_block: ToolUseBlock, **kwargs
     ) -> ToolResultBlockParam:
         if isinstance(tool_use_block.input, str):
             arg = json.loads(tool_use_block.input)
@@ -73,7 +77,11 @@ class PocketAnthropic(Pocket):
             body = json.loads(body)
 
         result, interrupted = await self.ainvoke_with_state(
-            tool_use_block.name, body=body, thread_id=thread_id, profile=profile, **kwargs
+            tool_use_block.name,
+            body=body,
+            thread_id=thread_id,
+            profile=profile,
+            **kwargs,
         )
         say = result
 
@@ -88,7 +96,9 @@ class PocketAnthropic(Pocket):
 
         return tool_result_block
 
-    def get_anthropic_tool_specs(self, use_profile: Optional[bool] = None) -> List[dict]:
+    def get_anthropic_tool_specs(
+        self, use_profile: Optional[bool] = None
+    ) -> List[dict]:
         if use_profile is not None:
             self.use_profile = use_profile
 

{hyperpocket_anthropic-0.1.10 → hyperpocket_anthropic-0.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hyperpocket-anthropic"
-version = "0.1.10"
+version = "0.2.0"
 description = ""
 authors = [{ name = "Hyperpocket Team", email = "hyperpocket@vessl.ai" }]
 requires-python = ">=3.10"
@@ -16,3 +16,6 @@ dev = ["pytest>=8.3.4", "ruff>=0.8.6"]
 [build-system]
 requires = ["hatchling"]
 build-backend = "hatchling.build"
+
+[tool.ruff]
+extend = "../../../.ruff.toml"

{hyperpocket_anthropic-0.1.10 → hyperpocket_anthropic-0.2.0}/tests/test_pocket_anthropic_no_profile.py

@@ -2,10 +2,10 @@ import ast
 from unittest.async_case import IsolatedAsyncioTestCase
 
 from anthropic import Anthropic
-from pydantic import BaseModel
-
 from hyperpocket.config import config, secret
 from hyperpocket.tool import from_git
+from pydantic import BaseModel
+
 from hyperpocket_anthropic import PocketAnthropic
 
 
@@ -49,13 +49,18 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
 
         self.pocket = PocketAnthropic(
             tools=[
-                from_git("https://github.com/vessl-ai/hyperawesometools", "main",
-                         "managed-tools/none/simple-echo-tool"),
+                from_git(
+                    "https://github.com/vessl-ai/hyperawesometools",
+                    "main",
+                    "managed-tools/none/simple-echo-tool",
+                ),
                 self.add,
-                self.sub_pydantic_args
+                self.sub_pydantic_args,
             ],
         )
-        self.tool_specs_no_profile = self.pocket.get_anthropic_tool_specs(use_profile=False)
+        self.tool_specs_no_profile = self.pocket.get_anthropic_tool_specs(
+            use_profile=False
+        )
         self.client = Anthropic(api_key=secret["ANTHROPIC_API_KEY"])
 
     async def asyncTearDown(self):
@@ -63,10 +68,20 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
 
     def test_get_tools_from_pocket_no_profile(self):
         # given
-        pocket = PocketAnthropic(tools=[
-            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"),
-        ])
+        pocket = PocketAnthropic(
+            tools=[
+                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",
+                ),
+            ]
+        )
 
         # when
         specs = pocket.get_anthropic_tool_specs()
@@ -74,12 +89,12 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
 
         # then
         self.assertIsInstance(get_tool, dict)
-        self.assertEqual(get_tool["name"], 'slack_get_messages')
+        self.assertEqual(get_tool["name"], "slack_get_messages")
         self.assertTrue("channel" in get_tool["input_schema"]["properties"])
         self.assertTrue("limit" in get_tool["input_schema"]["properties"])
 
         self.assertIsInstance(send_tool, dict)
-        self.assertEqual(send_tool["name"], 'slack_send_messages')
+        self.assertEqual(send_tool["name"], "slack_send_messages")
         self.assertTrue("channel" in send_tool["input_schema"]["properties"])
         self.assertTrue("text" in send_tool["input_schema"]["properties"])
 
@@ -87,10 +102,7 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "add 1, 2"
-            }],
+            messages=[{"role": "user", "content": "add 1, 2"}],
             tools=self.tool_specs_no_profile,
         )
 
@@ -108,10 +120,7 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "sub 1, 2"
-            }],
+            messages=[{"role": "user", "content": "sub 1, 2"}],
             tools=self.tool_specs_no_profile,
         )
 
@@ -129,10 +138,7 @@ class TestPocketAnthropicNoProfile(IsolatedAsyncioTestCase):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "echo 'hello world'"
-            }],
+            messages=[{"role": "user", "content": "echo 'hello world'"}],
             tools=self.tool_specs_no_profile,
         )
 

{hyperpocket_anthropic-0.1.10 → hyperpocket_anthropic-0.2.0}/tests/test_pocket_anthropic_use_profile.py

@@ -2,10 +2,10 @@ import ast
 from unittest.async_case import IsolatedAsyncioTestCase
 
 from anthropic import Anthropic
-from pydantic import BaseModel
-
 from hyperpocket.config import config, secret
 from hyperpocket.tool import from_git
+from pydantic import BaseModel
+
 from hyperpocket_anthropic import PocketAnthropic
 
 
@@ -49,13 +49,18 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
 
         self.pocket = PocketAnthropic(
             tools=[
-                from_git("https://github.com/vessl-ai/hyperawesometools", "main",
-                         "managed-tools/none/simple-echo-tool"),
+                from_git(
+                    "https://github.com/vessl-ai/hyperawesometools",
+                    "main",
+                    "managed-tools/none/simple-echo-tool",
+                ),
                 self.add,
-                self.sub_pydantic_args
+                self.sub_pydantic_args,
             ],
         )
-        self.tool_specs_use_profile = self.pocket.get_anthropic_tool_specs(use_profile=True)
+        self.tool_specs_use_profile = self.pocket.get_anthropic_tool_specs(
+            use_profile=True
+        )
         self.client = Anthropic(api_key=secret["ANTHROPIC_API_KEY"])
 
     async def asyncTearDown(self):
@@ -63,10 +68,20 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
 
     def test_get_tools_from_pocket_use_profile(self):
         # given
-        pocket = PocketAnthropic(tools=[
-            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"),
-        ])
+        pocket = PocketAnthropic(
+            tools=[
+                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",
+                ),
+            ]
+        )
 
         # when
         specs = pocket.get_anthropic_tool_specs(use_profile=True)
@@ -74,26 +89,30 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
 
         # then
         self.assertIsInstance(get_tool, dict)
-        self.assertEqual(get_tool["name"], 'slack_get_messages')
+        self.assertEqual(get_tool["name"], "slack_get_messages")
         self.assertTrue("body" in get_tool["input_schema"]["properties"])
-        self.assertTrue("channel" in get_tool["input_schema"]["properties"]["body"]["properties"])
-        self.assertTrue("limit" in get_tool["input_schema"]["properties"]["body"]["properties"])
+        self.assertTrue(
+            "channel" in get_tool["input_schema"]["properties"]["body"]["properties"]
+        )
+        self.assertTrue(
+            "limit" in get_tool["input_schema"]["properties"]["body"]["properties"]
+        )
 
         self.assertIsInstance(send_tool, dict)
-        self.assertEqual(send_tool["name"], 'slack_send_messages')
+        self.assertEqual(send_tool["name"], "slack_send_messages")
         self.assertTrue("body" in send_tool["input_schema"]["properties"])
-        self.assertTrue("channel" in send_tool["input_schema"]["properties"]["body"]["properties"])
-        self.assertTrue("text" in send_tool["input_schema"]["properties"]["body"]["properties"])
-
+        self.assertTrue(
+            "channel" in send_tool["input_schema"]["properties"]["body"]["properties"]
+        )
+        self.assertTrue(
+            "text" in send_tool["input_schema"]["properties"]["body"]["properties"]
+        )
 
     async def test_function_tool_use_profile(self):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "add 1, 2"
-            }],
+            messages=[{"role": "user", "content": "add 1, 2"}],
             tools=self.tool_specs_use_profile,
         )
 
@@ -107,15 +126,11 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
         self.assertIsNotNone(tool_result_block)
         self.assertEqual(tool_result_block["content"], "3")
 
-
     async def test_pydantic_function_tool_use_profile(self):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "sub 1, 2"
-            }],
+            messages=[{"role": "user", "content": "sub 1, 2"}],
             tools=self.tool_specs_use_profile,
         )
 
@@ -129,15 +144,11 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
         self.assertIsNotNone(tool_result_block)
         self.assertEqual(tool_result_block["content"], "-1")
 
-
     async def test_wasm_tool_use_profile(self):
         response = self.client.messages.create(
             model="claude-3-5-haiku-latest",
             max_tokens=500,
-            messages=[{
-                "role": "user",
-                "content": "echo 'hello world'"
-            }],
+            messages=[{"role": "user", "content": "echo 'hello world'"}],
             tools=self.tool_specs_use_profile,
         )
 
@@ -151,4 +162,3 @@ class TestPocketAnthropicUseProfile(IsolatedAsyncioTestCase):
         self.assertEqual(response.stop_reason, "tool_use")
         self.assertIsNotNone(tool_result_block)
         self.assertTrue(output["stdout"].startswith("echo message : hello world"))
-