npi-ai / npi

Action library for AI Agent
https://www.npi.ai
Apache License 2.0
184 stars 7 forks source link
agent artificial-intelligence autogpt autonomous-agent browser-automation chatgpt function-calling gpt-4 intergration large-language-models llm openai prompt-engineering workflow

NPI

[!WARNING] NPi is currently under active development and the APIs are subject to change in the future release. It is recommended to use the command line tool to try it out.

NPi is an open-source platform providing Tool-use APIs to empower AI agents with the ability to take action in virtual world!

๐Ÿ› ๏ธTry NPi Online: Try NPi on online Playground

๐Ÿ‘€ NPi Example: Highly recommended to check this first - See what you can build with NPi.

๐Ÿ”ฅ Introducing NPi: Why we build NPi?

๐Ÿ“š NPi Documentation: How to use NPi?

๐Ÿ“ข Join our community on Discord: Let's build NPi together ๐Ÿ‘ป !

NPi (Natural-language Programming Interface), pronounced as "N ฯ€", is an open-source platform providing Tool-use APIs to empower AI agents with the ability to operate and interact with a diverse array of software tools and applications.

Installation

pip install npiai

One-Minute Quick Start

Let's create a new tool to compute the nth Fibonacci number. Start by crafting a new Python file titled main.py and insert the following snippet:

import os
import json
import asyncio

from openai import OpenAI
from npiai import FunctionTool, function

class MyTool(FunctionTool):
    def __init__(self):
        super().__init__(
            name='Fibonacci',
            description='My first NPi tool',
        )

    @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:
        print(f'The schema of the tool is\n\n {json.dumps(tool.tools, indent=2)}')
        client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
        messages = [
            {
                "role": "user",
                "content": "What's the 10-th fibonacci number?",
            }
        ]
        response = client.chat.completions.create(
            model="gpt-4o",
            messages=messages,
            tools=tool.tools,  # use tool as functions package
            tool_choice="auto",
            max_tokens=4096,
        )
        response_message = response.choices[0].message
        if response_message.tool_calls:
            result = await tool.call(tool_calls=response_message.tool_calls)
            print(f'The result of function\n\n {json.dumps(result, indent=2)}')

if __name__ == "__main__":
    asyncio.run(main())

Now, run the tool:

python main.py

You will see the function result in OpenAI function calling format:

[
  {
    "role": "tool",
    "name": "fibonacci",
    "tool_call_id": "call_4KItpriZmoGxXgDloI5WOtHm",
    "content": 55
  }
]

content: 55 is the result of function calling, and the schema๏ผš

[
  {
    "type": "function",
    "function": {
      "name": "fibonacci",
      "description": "Get the nth Fibonacci number.",
      "parameters": {
        "properties": {
          "n": {
            "description": "The index of the Fibonacci number in the sequence.",
            "type": "integer"
          }
        },
        "required": [
          "n"
        ],
        "type": "object"
      }
    }
  }
]

That's it! You've successfully created and run your first NPi tool. ๐ŸŽ‰

Next Steps

License

Apache License 2.0