hyperpocket-anthropic 0.1.7__py3-none-any.whl → 0.1.9__py3-none-any.whl

hyperpocket_anthropic/pocket_anthropic.py

@@ -1,5 +1,7 @@
 import json
-from typing import List
+from typing import List, Optional
+
+from pydantic import BaseModel
 
 try:
     from anthropic.types import ToolResultBlockParam, ToolUseBlock
@@ -14,9 +16,27 @@ from hyperpocket_anthropic.util import tool_to_anthropic_spec
 
 class PocketAnthropic(Pocket):
     def invoke(self, tool_use_block: ToolUseBlock, **kwargs) -> ToolResultBlockParam:
-        body = self.parse_input(tool_use_block)
+        if isinstance(tool_use_block.input, str):
+            arg = json.loads(tool_use_block.input)
+        else:
+            arg = tool_use_block.input
+
+        if self.use_profile:
+            body = arg.pop("body")
+            thread_id = arg.pop("thread_id", "default")
+            profile = arg.pop("profile", "default")
+        else:
+            body = arg
+            thread_id = "default"
+            profile = "default"
+
+        if isinstance(body, BaseModel):
+            body = body.model_dump()
+        elif isinstance(body, str):
+            body = json.loads(body)
+
         result, interrupted = self.invoke_with_state(
-            tool_use_block.name, body=body, **kwargs
+            tool_use_block.name, body=body, thread_id=thread_id, profile=profile, **kwargs
         )
         say = result
         if interrupted:
@@ -33,9 +53,27 @@ class PocketAnthropic(Pocket):
     async def ainvoke(
             self, tool_use_block: ToolUseBlock, **kwargs
     ) -> ToolResultBlockParam:
-        body = self.parse_input(tool_use_block)
+        if isinstance(tool_use_block.input, str):
+            arg = json.loads(tool_use_block.input)
+        else:
+            arg = tool_use_block.input
+
+        if self.use_profile:
+            body = arg.pop("body")
+            thread_id = arg.pop("thread_id", "default")
+            profile = arg.pop("profile", "default")
+        else:
+            body = arg
+            thread_id = "default"
+            profile = "default"
+
+        if isinstance(body, BaseModel):
+            body = body.model_dump()
+        elif isinstance(body, str):
+            body = json.loads(body)
+
         result, interrupted = await self.ainvoke_with_state(
-            tool_use_block.name, body=body, **kwargs
+            tool_use_block.name, body=body, thread_id=thread_id, profile=profile, **kwargs
         )
         say = result
 
@@ -50,28 +88,16 @@ class PocketAnthropic(Pocket):
 
         return tool_result_block
 
-    @staticmethod
-    def parse_input(tool_use_block):
-        if isinstance(tool_use_block.input, str):
-            arg = json.loads(tool_use_block.input)
-            body = arg["body"]
-        else:
-            arg = tool_use_block.input
-            body = arg["body"]
-
-        if isinstance(body, str):
-            body = json.loads(body)
-
-        return body
+    def get_anthropic_tool_specs(self, use_profile: Optional[bool] = None) -> List[dict]:
+        if use_profile is not None:
+            self.use_profile = use_profile
 
-    def get_anthropic_tool_specs(self) -> List[dict]:
         specs = []
-        for tool in self.tools.values():
+        for tool in self.core.tools.values():
             spec = self.get_anthropic_tool_spec(tool)
             specs.append(spec)
         return specs
 
-    @staticmethod
-    def get_anthropic_tool_spec(tool: Tool) -> dict:
-        spec = tool_to_anthropic_spec(tool)
+    def get_anthropic_tool_spec(self, tool: Tool) -> dict:
+        spec = tool_to_anthropic_spec(tool, use_profile=self.use_profile)
         return spec

hyperpocket_anthropic/util/tool_to_anthropic_spec.py

@@ -2,10 +2,10 @@ from hyperpocket.tool import Tool
 from hyperpocket.util.flatten_json_schema import flatten_json_schema
 
 
-def tool_to_anthropic_spec(tool: Tool) -> dict:
+def tool_to_anthropic_spec(tool: Tool, use_profile: bool) -> dict:
     name = tool.name
-    description = tool.description
-    arg_schema = tool.schema_model()
+    description = tool.get_description(use_profile=use_profile)
+    arg_schema = tool.schema_model(use_profile=use_profile)
     json_schema = flatten_json_schema(arg_schema.model_json_schema())
 
     anthropic_spec = {

{hyperpocket_anthropic-0.1.7.dist-info → hyperpocket_anthropic-0.1.9.dist-info}/METADATA

@@ -1,16 +1,10 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: hyperpocket-anthropic
-Version: 0.1.7
-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: anthropic (>=0.40.0,<0.41.0)
-Requires-Dist: hyperpocket (>=0.0.3,<0.0.4)
+Version: 0.1.9
+Author-email: Hyperpocket Team <hyperpocket@vessl.ai>
+Requires-Python: >=3.10
+Requires-Dist: anthropic>=0.40.0
+Requires-Dist: hyperpocket>=0.0.3
 Description-Content-Type: text/markdown
 
 ## Anthropic extensions
@@ -18,13 +12,15 @@ Description-Content-Type: text/markdown
 ### Get Pocket Anthropic Tool Spec
 
 ```python
-import hyperpocket as pk
-from pocket_anthropic import PocketAnthropic
+from hyperpocket.tool import from_git
+
+from hyperpocket_anthropic import PocketAnthropic
 
 pocket = PocketAnthropic(tools=[
-    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-    *pk.curated_tools.LINEAR,
-    "https://github.com/my-org/some-awesome-tool"]
+    "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 anthropic compatible tool specs from pocket 
@@ -90,14 +86,15 @@ response_after_tool_call = llm.messages.create(
 import os
 
 from anthropic import Anthropic
+from hyperpocket.tool import from_git
 
-import hyperpocket as pk
-from pocket_anthropic import PocketAnthropic
+from hyperpocket_anthropic import PocketAnthropic
 
 pocket = PocketAnthropic(tools=[
-    *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-    *pk.curated_tools.LINEAR,
-    "https://github.com/my-org/some-awesome-tool"]
+    "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 anthropic compatible tool specs from pocket 
@@ -141,16 +138,16 @@ response_after_tool_call = llm.messages.create(
 import os
 
 from anthropic import Anthropic
+from hyperpocket.tool import from_git
 
-import hyperpocket as pk
-from pocket_anthropic import PocketAnthropic
+from hyperpocket_anthropic import PocketAnthropic
 
 client = Anthropic()
-pocket = PocketAnthropic(
-    tools=[
-        *pk.curated_tools.SLACK,  # SLACK = [slack_get_message, slack_post_message, ..]
-        *pk.curated_tools.LINEAR,
-        "https://github.com/my-org/some-awesome-tool"]
+pocket = PocketAnthropic(tools=[
+    "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"),
+]
 )
 
 tool_specs = pocket.get_anthropic_tool_specs()
@@ -191,4 +188,4 @@ while True:
 
         if response.stop_reason != "tool_use":
             break
-```
+```

hyperpocket_anthropic-0.1.9.dist-info/RECORD

@@ -0,0 +1,7 @@
+hyperpocket_anthropic/__init__.py,sha256=XZl2uODSPMjSOO1LwrJOVMUD1qmGCuykx9SDePN7gLo,98
+hyperpocket_anthropic/pocket_anthropic.py,sha256=UViv8vIWJ88eItWMAIGUBhCXZQWCn2PYc2EuQdgoAPs,3263
+hyperpocket_anthropic/util/__init__.py,sha256=U8x3PBHTFMpbtbAA5-pQVsggQ3Yph4oeC13wfyzK21I,123
+hyperpocket_anthropic/util/tool_to_anthropic_spec.py,sha256=Lnxs-RMdHkpswjG1aJeEXGuJe8vQfWsbZccsbdPb3_U,749
+hyperpocket_anthropic-0.1.9.dist-info/METADATA,sha256=W9Tc1Ltb1bNVCE3SWqk2MYWy8O-6x1xH0_46u3e9Jqw,4826
+hyperpocket_anthropic-0.1.9.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hyperpocket_anthropic-0.1.9.dist-info/RECORD,,

{hyperpocket_anthropic-0.1.7.dist-info → hyperpocket_anthropic-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_anthropic-0.1.7.dist-info/RECORD

@@ -1,7 +0,0 @@
-hyperpocket_anthropic/__init__.py,sha256=XZl2uODSPMjSOO1LwrJOVMUD1qmGCuykx9SDePN7gLo,98
-hyperpocket_anthropic/pocket_anthropic.py,sha256=gv7K6tfqDylYgn-SyBvySzwZTkz4G1mhJyQi5-6NHKM,2304
-hyperpocket_anthropic/util/__init__.py,sha256=U8x3PBHTFMpbtbAA5-pQVsggQ3Yph4oeC13wfyzK21I,123
-hyperpocket_anthropic/util/tool_to_anthropic_spec.py,sha256=lMq1QiWDPhPjJ3dPzQOJhJdF_F4Y-HwXiBC2dQf_ul4,678
-hyperpocket_anthropic-0.1.7.dist-info/METADATA,sha256=gBPSSGKj27mq8UfrTfF-tSJCwlh26569EfT3JmB5PE4,4710
-hyperpocket_anthropic-0.1.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-hyperpocket_anthropic-0.1.7.dist-info/RECORD,,