A client facing API for interacting with the Weco AI's AI function platform. It empowers you to go from zero to AI in just a few seconds!

Use this API to build complex AI features fast. We lower the barrier of entry to AI features by providing an interface to prototype solutions quickly, in just a few lines of code and in natural language.

What We Offer

• Structured Output (outputs are Python dictionaries that always follow your AI functions JSON schema)
• Multimodal (language & vision)
• Grounding (Access to the web)
• Interpretable (observe reasoning behind outputs)
• Batched Inputs (have inputs be processed in concurrently)
• Sync-Async Duality (functions can be both synchronous & asynchronous)

Getting Started

Install the aifn package:

pip install aifn

When using the Weco API, you will need to set the API key: You can find/create your API key here. Once you have your API key, you can pass it directly to our core functions and classes using the api_key argument or set it as an environment variable as shown:

export WECO_API_KEY=<YOUR_WECO_API_KEY>

Example

We created a function on our platform for the following task:

"Analyze a business idea and provide a well reasoned evaluation. Return 'viability_score' (0-100), 'strengths' (list), 'weaknesses' (list), and 'next_steps' (list)."

Here's how you can use this function anywhere in your code!

from aifn import AIFunction
idea_evaluator = AIFunction("BusinessIdeaAnalyzer-XYZ123") # Replace with your actual function name
response = idea_evaluator("A subscription service for personalized, AI-generated bedtime stories for children.").output

• The aifn.build function enables quick and easy prototyping of new AI functions that use foundation models as their core. We encourage users to do this through our platform for maximum control and ease of use, however, you can also do this through our API as shown here.
• aifn.AIFunction allows you to retrieve an AI function you've already created.
• Finally, as shown in the example above, AIFunction objects are akin to any function that we are used to...the only difference is that they have a large language model to power them!

To learn how to get the most your of your AI functions, check out our cookbook.

Contributing

We value your contributions! If you believe you can help to improve our package enabling people to build AI with AI, please contribute!

Use the following steps as a guideline to help you make contributions:

1. Download and install package from source:

   git clone https://github.com/WecoAI/aifn-python.git
   cd aifn-python
   pip install -e ".[dev,docs]"
   

2. Create a new branch for your feature or bugfix:

   git checkout -b feature/your-feature-name
   

3. Make your changes and run tests to ensure everything is working:

   Tests can be expensive to run as they make LLM requests with the API key being used so it is the developers best interests to write small and simple tests that adds coverage for a large portion of the package.

   pytest -n auto tests
   

   If you're just making changes to the docs, feel free to skip this step.

4. Commit and push your changes, then open a PR for us to view 😁

Please ensure your code follows our style guidelines (Numpy docstrings) and includes appropriate tests. We appreciate your contributions!