feat(integration): support claude

Author: idiotWu

examples/integration/claude/main.py

@@ -0,0 +1,62 @@
+import asyncio
+
+import anthropic
+
+from npiai import FunctionTool, function
+from npiai.utils.test_utils import DebugContext
+from npiai.integration.claude import get_claude_tools
+
+
+class MyTool(FunctionTool):
+    name = "Fibonacci"
+    description = "My first NPi tool"
+
+    def __init__(self):
+        super().__init__()
+
+    @function
+    def fibonacci(self, n: int) -> int:
+        """
+        Get the nth Fibonacci number.
+
+        Args:
+            n: The index of the Fibonacci number in the sequence.
+        """
+        if n == 0:
+            return 0
+        if n == 1:
+            return 1
+        return self.fibonacci(n - 1) + self.fibonacci(n - 2)
+
+
+async def main():
+    async with MyTool() as tool:
+        ctx = DebugContext()
+        client = anthropic.Anthropic()
+        messages = [
+            {
+                "role": "user",
+                "content": "What's the 10-th fibonacci number?",
+            }
+        ]
+
+        response = client.messages.create(
+            model="claude-3-5-sonnet-20241022",
+            max_tokens=1024,
+            messages=messages,
+            tools=get_claude_tools(tool),
+        )
+
+        for content in response.content:
+            match content.type:
+                case "text":
+                    print(content.text)
+                case "tool_use":
+                    result = await tool.exec(ctx, content.name, content.input)
+                    print(
+                        f"Calling {content.name} with {content.input} returned {result}"
+                    )
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

npiai/integration/claude/__init__.py

@@ -0,0 +1,3 @@
+from .get_claude_tools import get_claude_tools
+
+__all__ = ["get_claude_tools"]

npiai/integration/claude/get_claude_tools.py

@@ -0,0 +1,23 @@
+from npiai.core import BaseTool
+from npiai.utils import sanitize_schema
+
+__EMPTY_INPUT_SCHEMA__ = {"type": "object", "properties": {}, "required": []}
+
+
+def get_claude_tools(tool: BaseTool):
+    tools = []
+
+    for fn_reg in tool.unpack_functions():
+        tool = {
+            "name": fn_reg.name,
+            "description": fn_reg.description,
+            "input_schema": (
+                sanitize_schema(fn_reg.model, strict=False)
+                if fn_reg.model
+                else __EMPTY_INPUT_SCHEMA__
+            ),
+        }
+
+        tools.append(tool)
+
+    return tools

poetry.lock

@@ -54,6 +54,7 @@ pytest-asyncio = "^0.24.0"
 pytest-dotenv = "^0.5.2"
 pytest-repeat = "^0.9.3"
 csv-diff = "^1.2"
+anthropic = "^0.42.0"
 
 [build-system]
 requires = ["poetry-core"]