weco 0.1.3__tar.gz → 0.1.5__tar.gz

{weco-0.1.3 → weco-0.1.5}/.github/workflows/release.yml

@@ -1,9 +1,9 @@
-name: Publish Python 🐍 distribution 📦 to PyPI
+name: Publish Python Package
 
 on:
   push:
     branches:
-      - main
+      - 'main'
 
   release:
     types: [published]
@@ -32,7 +32,7 @@ jobs:
 
   publish-to-pypi:
     name: >-
-      Publish Python 🐍 distribution 📦 to PyPI
+      Publish Python 🐍 distribution PyPI
     needs:
     - build
     runs-on: ubuntu-latest
@@ -43,17 +43,17 @@ jobs:
       id-token: write
 
     steps:
-    - name: Download all the dists
+    - name: Download the dists
       uses: actions/download-artifact@v3
       with:
         name: python-package-distributions
         path: dist/
-    - name: Publish distribution 📦 to PyPI
+    - name: Publish to PyPI
       uses: pypa/gh-action-pypi-publish@release/v1
 
   github-release:
     name: >-
-      Sign Python 🐍 distribution 📦 and create GitHub Release
+      Create GitHub Release
     needs:
     - publish-to-pypi
     runs-on: ubuntu-latest
@@ -63,12 +63,12 @@ jobs:
       id-token: write
 
     steps:
-    - name: Download all the dists
+    - name: Download dists
       uses: actions/download-artifact@v3
       with:
         name: python-package-distributions
         path: dist/
-    - name: Sign the dists with Sigstore
+    - name: Sign dists with Sigstore
       uses: sigstore/gh-action-sigstore-python@v2.1.1
       with:
         inputs: >-
@@ -78,9 +78,8 @@ jobs:
       env:
         GITHUB_TOKEN: ${{ github.token }}
       run: >-
-        LATEST_TAG=$(git describe --tags --abbrev=0)
         gh release create
-        "$LATEST_TAG"
+        '${{ github.event.release.tag_name }}'
         --repo '${{ github.repository }}'
         --notes ""
     - name: Upload artifact signatures to GitHub Release
@@ -91,5 +90,5 @@ jobs:
       # sigstore-produced signatures and certificates.
       run: >-
         gh release upload
-        "$LATEST_TAG" dist/**
+        '${{ github.event.release.tag_name }}' dist/**
         --repo '${{ github.repository }}'

{weco-0.1.3 → weco-0.1.5}/.gitignore

@@ -165,4 +165,7 @@ cython_debug/
 .DS_Store
 
 # Case Studies
-case_studies/
+case_studies/
+
+# Testing files
+test.py

{weco-0.1.3 → weco-0.1.5}/PKG-INFO

@@ -1,11 +1,11 @@
 Metadata-Version: 2.1
 Name: weco
-Version: 0.1.3
+Version: 0.1.5
 Summary: A client facing API for interacting with the WeCo AI function builder service.
 Author-email: WeCo AI Team <dhruv@weco.ai>
 License: MIT
 Project-URL: Homepage, https://github.com/WecoAI/weco-python
-Keywords: AI,LLM,machine learning,data science,function builder
+Keywords: AI,LLM,machine learning,data science,function builder,AI function
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent
 Classifier: License :: OSI Approved :: MIT License
@@ -14,21 +14,31 @@ Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: asyncio
 Requires-Dist: httpx[http2]
+Requires-Dist: pillow
 Provides-Extra: dev
-Requires-Dist: build; extra == "dev"
-Requires-Dist: setuptools_scm; extra == "dev"
 Requires-Dist: flake8; extra == "dev"
 Requires-Dist: flake8-pyproject; extra == "dev"
 Requires-Dist: black; extra == "dev"
 Requires-Dist: isort; extra == "dev"
+Requires-Dist: pytest-asyncio; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
+Requires-Dist: build; extra == "dev"
+Requires-Dist: setuptools_scm; extra == "dev"
 
 <div align="center" style="display: flex; align-items: center; justify-content: center;">
   <img src="assets/weco.svg" alt="WeCo AI" style="height: 50px; margin-right: 10px;">
   <a href="https://git.io/typing-svg"><img src="https://readme-typing-svg.demolab.com?font=Georgia&size=32&duration=4000&pause=400&color=FD4578&vCenter=true&multiline=false&width=200&height=50&lines=WeCo+Client" alt="Typing SVG" /></a>
 </div>
 
+![Python](https://img.shields.io/badge/Python-3.10.14-blue)
+[![License](https://img.shields.io/badge/License-MIT-green.svg)](https://opensource.org/licenses/MIT)
+
+<!-- TODO: Update examples -->
 # $f$(👷‍♂️)
 
+<a href="https://colab.research.google.com/github/WecoAI/weco-python/blob/main/examples/cookbook.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" width=110 height=20/></a>
+<a target="_blank" href="https://lightning.ai/new?repo_url=https%3A%2F%2Fgithub.com%2FWecoAI%2Fweco-python%2Fblob%2Fmain%2Fexamples%2Fcookbook.ipynb"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open in Studio" width=100 height=20/></a>
+
 A client facing API for interacting with the [WeCo AI](https://www.weco.ai/) function builder [service](https://weco-app.vercel.app/function)!
 
 
@@ -47,6 +57,7 @@ pip install weco
 - The **query** function allows you to test and use the newly created function in your own code.
 - We offer asynchronous versions of the above clients.
 - We provide a **batch_query** functions that allows users to batch functions for various inputs as well as multiple inputs for the same function in a query. This is helpful to make a large number of queries more efficiently.
+- We also offer multimodality capabilities. You can now query our client with both **language** AND **vision** inputs!
 
 We provide both services in two ways:
 - `weco.WecoAI` client to be used when you want to maintain the same client service across a portion of code. This is better for dense service usage.
@@ -63,18 +74,48 @@ export WECO_API_KEY=<YOUR_WECO_API_KEY>
 ## Example
 
 We create a function on the [web console](https://weco-app.vercel.app/function) for the following task:
-> "I want to evaluate the feasibility of a machine learning task. Give me a json object with three keys - 'feasibility', 'justification', and 'suggestions'."
+> "Analyze a business idea and provide a structured evaluation. Output a JSON with 'viability_score' (0-100), 'strengths' (list), 'weaknesses' (list), and 'next_steps' (list)."
 
 Now, you're ready to query this function anywhere in your code!
 
 ```python
 from weco import query
 response = query(
-    fn_name=fn_name,
-    fn_input="I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.",
+    fn_name="BusinessIdeaAnalyzer-XYZ123",  # Replace with your actual function name
+    text_input="A subscription service for personalized, AI-generated bedtime stories for children."
 )
 ```
 
 For more examples and an advanced user guide, check out our function builder [cookbook](examples/cookbook.ipynb).
 
 ## Happy building $f$(👷‍♂️)!
+
+## 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:
+   ```bash
+   git clone https://github.com/WecoAI/weco-python.git
+   cd weco-python
+   pip install -e ".[dev]"
+   ```
+
+2. Create a new branch for your feature or bugfix:
+   ```bash
+   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.**
+   
+   ```bash
+   pytest -n auto tests
+   ```
+
+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!

{weco-0.1.3 → weco-0.1.5}/README.md

@@ -3,8 +3,15 @@
   <a href="https://git.io/typing-svg"><img src="https://readme-typing-svg.demolab.com?font=Georgia&size=32&duration=4000&pause=400&color=FD4578&vCenter=true&multiline=false&width=200&height=50&lines=WeCo+Client" alt="Typing SVG" /></a>
 </div>
 
+![Python](https://img.shields.io/badge/Python-3.10.14-blue)
+[![License](https://img.shields.io/badge/License-MIT-green.svg)](https://opensource.org/licenses/MIT)
+
+<!-- TODO: Update examples -->
 # $f$(👷‍♂️)
 
+<a href="https://colab.research.google.com/github/WecoAI/weco-python/blob/main/examples/cookbook.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab" width=110 height=20/></a>
+<a target="_blank" href="https://lightning.ai/new?repo_url=https%3A%2F%2Fgithub.com%2FWecoAI%2Fweco-python%2Fblob%2Fmain%2Fexamples%2Fcookbook.ipynb"><img src="https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg" alt="Open in Studio" width=100 height=20/></a>
+
 A client facing API for interacting with the [WeCo AI](https://www.weco.ai/) function builder [service](https://weco-app.vercel.app/function)!
 
 
@@ -23,6 +30,7 @@ pip install weco
 - The **query** function allows you to test and use the newly created function in your own code.
 - We offer asynchronous versions of the above clients.
 - We provide a **batch_query** functions that allows users to batch functions for various inputs as well as multiple inputs for the same function in a query. This is helpful to make a large number of queries more efficiently.
+- We also offer multimodality capabilities. You can now query our client with both **language** AND **vision** inputs!
 
 We provide both services in two ways:
 - `weco.WecoAI` client to be used when you want to maintain the same client service across a portion of code. This is better for dense service usage.
@@ -39,18 +47,48 @@ export WECO_API_KEY=<YOUR_WECO_API_KEY>
 ## Example
 
 We create a function on the [web console](https://weco-app.vercel.app/function) for the following task:
-> "I want to evaluate the feasibility of a machine learning task. Give me a json object with three keys - 'feasibility', 'justification', and 'suggestions'."
+> "Analyze a business idea and provide a structured evaluation. Output a JSON with 'viability_score' (0-100), 'strengths' (list), 'weaknesses' (list), and 'next_steps' (list)."
 
 Now, you're ready to query this function anywhere in your code!
 
 ```python
 from weco import query
 response = query(
-    fn_name=fn_name,
-    fn_input="I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.",
+    fn_name="BusinessIdeaAnalyzer-XYZ123",  # Replace with your actual function name
+    text_input="A subscription service for personalized, AI-generated bedtime stories for children."
 )
 ```
 
 For more examples and an advanced user guide, check out our function builder [cookbook](examples/cookbook.ipynb).
 
-## Happy building $f$(👷‍♂️)!
+## Happy building $f$(👷‍♂️)!
+
+## 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:
+   ```bash
+   git clone https://github.com/WecoAI/weco-python.git
+   cd weco-python
+   pip install -e ".[dev]"
+   ```
+
+2. Create a new branch for your feature or bugfix:
+   ```bash
+   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.**
+   
+   ```bash
+   pytest -n auto tests
+   ```
+
+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!

weco-0.1.5/examples/cookbook.ipynb

@@ -0,0 +1,358 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "<div align=\"center\" style=\"display: flex; align-items: center; justify-content: left;\">\n",
+    "  <img src=\"../assets/weco.svg\" alt=\"WeCo AI\" style=\"height: 50px; margin-right: 10px;\">\n",
+    "  <a href=\"https://git.io/typing-svg\"><img src=\"https://readme-typing-svg.demolab.com?font=Georgia&size=32&duration=4000&pause=400&color=FD4578&vCenter=true&multiline=false&width=750&height=100&lines=AI+Function+Builder;\" alt=\"Typing SVG\" /></a>\n",
+    "</div>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Getting Started"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "`weco` is a client facing API for interacting with the [WeCo AI](https://www.weco.ai/) function builder [service](https://weco-app.vercel.app/function). Use this API to build *complex* systems *fast* $f$(👷‍♂️)!\n",
+    "\n",
+    "Here's a short introduction to how you can use our client. Feel free to follow along:\n",
+    "\n",
+    "<a href=\"https://colab.research.google.com/github/WecoAI/weco-python/blob/main/examples/cookbook.ipynb\" target=\"_parent\"><img src=\"https://colab.research.google.com/assets/colab-badge.svg\" alt=\"Open In Colab\"/></a>\n",
+    "<a target=\"_blank\" href=\"https://lightning.ai/new?repo_url=https%3A%2F%2Fgithub.com%2FWecoAI%2Fweco-python%2Fblob%2Fmain%2Fexamples%2Fcookbook.ipynb\"><img src=\"https://pl-bolts-doc-images.s3.us-east-2.amazonaws.com/app-2/studio-badge.svg\" alt=\"Open in Studio\" width=100 height=20/></a>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Install the `weco` package in your terminal of choice:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%pip install weco"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Export your API key which can be found [here](https://weco-app.vercel.app/account)."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "%env WECO_API_KEY=<YOUR_WECO_API_KEY>"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You can build powerful AI functions for complex tasks quickly and without friction. For example, you can create a function in the [web console](https://weco-app.vercel.app/function) with this description:\n",
+    "\n",
+    "> \"Analyze a business idea and provide a structured evaluation. Output a JSON with 'viability_score' (0-100), 'strengths' (list), 'weaknesses' (list), and 'next_steps' (list).\"\n",
+    "\n",
+    "Once created, you can query this function anywhere in your code with just a few lines:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import query\n",
+    "\n",
+    "response = query(\n",
+    "    fn_name=\"BusinessIdeaAnalyzer-XYZ123\",  # Replace with your actual function name\n",
+    "    text_input=\"A subscription service for personalized, AI-generated bedtime stories for children.\"\n",
+    ")\n",
+    "\n",
+    "print(response)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Multimodality"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Our AI functions can interpret complex visual information, follow instructions in natural language and provide practical insights. Let's explore how we can all have a chef give us personalized advice! They can look at the food we have and offer recipe suggestions even providing nutritional information.\n",
+    "As shown in the example above, you can provde the image input in various ways such as:\n",
+    "1. Base64 encoding\n",
+    "3. Public URL\n",
+    "4. Local Path"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import build, query\n",
+    "import base64\n",
+    "\n",
+    "task_description = \"\"\"\n",
+    "Create a Smart Home Energy Analyzer that can process images of smart meters, home exteriors, \n",
+    "and indoor spaces to provide energy efficiency insights. The analyzer should:\n",
+    "    1. Interpret smart meter readings\n",
+    "    2. Assess home features relevant to energy consumption\n",
+    "    3. Analyze thermostat settings\n",
+    "    4. Provide energy-saving recommendations\n",
+    "    5. Evaluate renewable energy potential\n",
+    "\n",
+    "The output should include:\n",
+    "    - 'energy_consumption': current usage and comparison to average\n",
+    "    - 'home_analysis': visible energy features and potential issues\n",
+    "    - 'thermostat_settings': current settings and recommendations\n",
+    "    - 'energy_saving_recommendations': actionable suggestions with estimated savings\n",
+    "    - 'renewable_energy_potential': assessment of current and potential renewable energy use\n",
+    "    - 'estimated_carbon_footprint': current footprint and potential reduction\n",
+    "\"\"\"\n",
+    "\n",
+    "fn_name, _ = build(task_description=task_description)\n",
+    "\n",
+    "request = \"\"\"\n",
+    "Analyze these images of my home and smart meter to provide energy efficiency insights \n",
+    "and recommendations for reducing my electricity consumption.\n",
+    "\"\"\"\n",
+    "\n",
+    "# Base64 encoded image\n",
+    "with open(\"/path/to/home_exterior.jpeg\", \"rb\") as img_file:\n",
+    "    my_home_exterior = base64.b64encode(img_file.read()).decode('utf-8')\n",
+    "\n",
+    "response = query(\n",
+    "    fn_name=fn_name,\n",
+    "    text_input=request,\n",
+    "    images_input=[\n",
+    "        \"https://example.com/my_smart_meter_reading.png\",  # Public URL\n",
+    "        f\"data:image/jpeg;base64,{my_home_exterior}\",      # Base64 encoding\n",
+    "        \"/path/to/living_room_thermostat.jpg\"              # Local image path\n",
+    "    ]\n",
+    ")\n",
+    "\n",
+    "print(response)"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Running Example"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Consider the previous example of:\n",
+    "> \"I want to evaluate the feasibility of a machine learning task. Help me understand this through - 'feasibility', 'justification', and 'suggestions'.\"\n",
+    "\n",
+    "Here's how you can take advantage of our API to best suit your needs."
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "task_description = \"I want to evaluate the feasibility of a machine learning task. Give me a json object with three keys - 'feasibility', 'justification', and 'suggestions'.\""
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Dense vs. Sparse Usage"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Though we recommend building functions in the [web console](https://weco-app.vercel.app/function) for maximum control over the function I/O, you can also do this programmatically:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import build, query\n",
+    "\n",
+    "# Describe the task you want the function to perform\n",
+    "fn_name, fn_desc = build(task_description=task_description)\n",
+    "print(f\"AI Function {fn_name} built. This does the following - \\n{fn_desc}.\")\n",
+    "\n",
+    "# Query the function with a specific input\n",
+    "query_response = query(\n",
+    "    fn_name=fn_name,\n",
+    "    text_input=\"I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.\"\n",
+    ")\n",
+    "for key, value in query_response.items(): print(f\"{key}: {value}\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We recommend to use the `weco.build` and `weco.query` when you want to build or query LLM functions sparsely, i.e., you **don't** call `weco.build` or `weco.query` in many places within your code. However, for more dense usage, we've found users prefer our `weco.WecoAI` client instance. Its easy to switch between the two as shown below:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import WecoAI\n",
+    "\n",
+    "# Connect to our service, using our client\n",
+    "client = WecoAI()\n",
+    "\n",
+    "# Make the same query as before\n",
+    "query_response = client.query(\n",
+    "    fn_name=fn_name,\n",
+    "    text_input=\"I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.\"\n",
+    ")\n",
+    "for key, value in query_response.items(): print(f\"{key}: {value}\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Batching"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "We understand that sometimes, independent of how many times in your code you call `weco` functions, you want to submit a large batch of requests for the same or different LLM functions you've created. This can be done in the following ways:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import batch_query\n",
+    "\n",
+    "# Query the same function with multiple inputs by batching them for maximum efficiency\n",
+    "input_1 = {\"text_input\": \"I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.\"}\n",
+    "input_2 = {\n",
+    "    \"text_input\": \"I want to train a model to classify digits using the MNIST dataset hosted on Kaggle using a Google Colab notebook. Attached is an example of what some of the digits would look like.\",\n",
+    "    \"images_input\": [\"https://machinelearningmastery.com/wp-content/uploads/2019/02/Plot-of-a-Subset-of-Images-from-the-MNIST-Dataset-1024x768.png\"]\n",
+    "}\n",
+    "query_responses = batch_query(\n",
+    "    fn_names=fn_name,\n",
+    "    batch_inputs=[input_1, input_2]\n",
+    ")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "You can do the same using the `weco.WecoAI` client. If you wanted to batch different functions, you can pass a list of function names to the `batch_query()` function. Note that the names of functions would need to be ordered the same as the function inputs provided.\n",
+    "\n",
+    "In addition, `weco.batch_query` takes the input batch as an array of individual inputs formatted in the following way -\n",
+    "```json\n",
+    "{\n",
+    "    \"text_input\": \"Your text input\",\n",
+    "    \"images_input\": [\"image1\", \"image2\"]\n",
+    "}\n",
+    "```"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Async Calls"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "Until now you've been making synchronous calls to our client by we also support asynchronous programmers. This is actually how we implement batching! You can also make asynchronous calls to our service using our `weco.WecoAI` client or as shown below for the same example as before:"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from weco import abuild, aquery\n",
+    "\n",
+    "# Describe the task you want the function to perform\n",
+    "fn_name, fn_desc = await abuild(task_description=task_description)\n",
+    "print(f\"AI Function {fn_name} built. This does the following - \\n{fn_desc}.\")\n",
+    "\n",
+    "# Query the function with a specific input\n",
+    "query_response = await aquery(\n",
+    "    fn_name=fn_name,\n",
+    "    text_input=\"I want to train a model to predict house prices using the Boston Housing dataset hosted on Kaggle.\"\n",
+    ")\n",
+    "for key, value in query_response.items(): print(f\"{key}: {value}\")"
+   ]
+  },
+  {
+   "cell_type": "markdown",
+   "metadata": {},
+   "source": [
+    "## Happy Building $f$(👷‍♂️)!"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": "weco",
+   "language": "python",
+   "name": "python3"
+  },
+  "language_info": {
+   "codemirror_mode": {
+    "name": "ipython",
+    "version": 3
+   },
+   "file_extension": ".py",
+   "mimetype": "text/x-python",
+   "name": "python",
+   "nbconvert_exporter": "python",
+   "pygments_lexer": "ipython3",
+   "version": "3.10.14"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 2
+}

{weco-0.1.3 → weco-0.1.5}/pyproject.toml

@@ -10,20 +10,11 @@ authors = [
 ]
 description = "A client facing API for interacting with the WeCo AI function builder service."
 readme = "README.md"
-version = "0.1.3"
+version = "0.1.5"
 license = {text = "MIT"}
 requires-python = ">=3.8"
-dependencies = [
-  "asyncio",
-  "httpx[http2]"
-]
-keywords = [
-    "AI",
-    "LLM",
-    "machine learning",
-    "data science",
-    "function builder"
-]
+dependencies = ["asyncio", "httpx[http2]", "pillow"]
+keywords = ["AI", "LLM", "machine learning", "data science", "function builder", "AI function"]
 classifiers = [
     "Programming Language :: Python :: 3",
     "Operating System :: OS Independent",
@@ -34,7 +25,7 @@ classifiers = [
 Homepage = "https://github.com/WecoAI/weco-python"
 
 [project.optional-dependencies]
-dev = ["build", "setuptools_scm", "flake8", "flake8-pyproject", "black", "isort"]
+dev = ["flake8", "flake8-pyproject", "black", "isort", "pytest-asyncio", "pytest-xdist", "build", "setuptools_scm"]
 
 [tool.setuptools]
 packages = ["weco"]