hyperpocket-anthropic 0.1.8__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.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.8.dist-info → hyperpocket_anthropic-0.1.9.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: hyperpocket-anthropic
-Version: 0.1.8
-Author-email: moon <moon@vessl.ai>
+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

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.8.dist-info/RECORD

@@ -1,7 +0,0 @@
-hyperpocket_anthropic/__init__.py,sha256=XZl2uODSPMjSOO1LwrJOVMUD1qmGCuykx9SDePN7gLo,98
-hyperpocket_anthropic/pocket_anthropic.py,sha256=XBSxxSeHjzvWx-jiqozf64e8CtBzz3I-dDQ4pBBHnNU,2309
-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.8.dist-info/METADATA,sha256=Bf6m9nu8oSigeDXx-ND1VGnngCmE0SeauiK7kr0gt0s,4807
-hyperpocket_anthropic-0.1.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hyperpocket_anthropic-0.1.8.dist-info/RECORD,,