aixtools 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

notebooks/example_faulty_mcp_server.ipynb

@@ -0,0 +1,74 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "57d36f62",
+   "metadata": {},
+   "source": [
+    "# Example \"Faulty MCP Server\"\n",
+    "\n",
+    "Run the server using the script:\n",
+    "\n",
+    "```bash\n",
+    "./scripts/run_faulty_mcp_server.sh\n",
+    "``` "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ebf6d915",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from aixtools.agents.agent import get_agent, run_agent\n",
+    "from pydantic_ai.mcp import MCPServerStreamableHTTP"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "97d1e607",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "server = MCPServerStreamableHTTP(\"http://localhost:9999/mcp/\")\n",
+    "agent = get_agent(mcp_servers=[server])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fc560d25",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "async with agent:\n",
+    "    # ret = await run_agent(agent, \"What is the add of 40123456789 and 2123456789?\", verbose=True, debug=True)\n",
+    "    # ret = await run_agent(agent, \"Invoke the always_error tool\", verbose=True, debug=True)\n",
+    "    ret = await run_agent(agent, \"Invoke the throw_404_exception tool\", verbose=True, debug=True)"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "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.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

notebooks/example_mcp_server_stdio.ipynb

@@ -0,0 +1,76 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "57d36f62",
+   "metadata": {},
+   "source": [
+    "# Example MCP Server STDIO"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "ebf6d915",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from aixtools.agents.agent import get_agent, run_agent\n",
+    "from aixtools.utils.config import PROJECT_DIR\n",
+    "from pydantic_ai.mcp import MCPServerStdio"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "97d1e607",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "mcp_path = PROJECT_DIR /'aixtools' / 'mcp' / 'example_server.py'\n",
+    "server = MCPServerStdio(command='fastmcp', args=['run', str(mcp_path)])\n",
+    "agent = get_agent(mcp_servers=[server])"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "fc560d25",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "async with agent:\n",
+    "    ret = await run_agent(agent, \"What is the add of 40123456789 and 2123456789?\", verbose=True, debug=True)"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cad32e85",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "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.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

notebooks/example_raw_mcp_client.ipynb

@@ -0,0 +1,84 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "57d36f62",
+   "metadata": {},
+   "source": [
+    "# Example \"Raw MCP Client\"\n",
+    "\n",
+    "Run the server using the script:\n",
+    "\n",
+    "```bash\n",
+    "./scripts/run_faulty_mcp_server.sh\n",
+    "``` "
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "cad32e85",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from fastmcp import Client\n",
+    "import rich\n",
+    "\n",
+    "client = Client(\"http://localhost:9999/mcp/\")"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e372fc6a",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "async with client:\n",
+    "    # Basic server interaction\n",
+    "    await client.ping()\n",
+    "    \n",
+    "    # # List available operations\n",
+    "    # tools = await client.list_tools()\n",
+    "    # rich.print(tools)\n",
+    "    # \n",
+    "    # # Execute 'add' tool\n",
+    "    # result = await client.call_tool(\"add\", {\"a\": 7, \"b\": 3})\n",
+    "    # rich.print(result)\n",
+    "\n",
+    "    # Execute 'add' tool\n",
+    "    result = await client.call_tool(\"throw_404_exception\", {})\n",
+    "    rich.print(result)\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "a8a743e2",
+   "metadata": {},
+   "outputs": [],
+   "source": []
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "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.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

notebooks/example_tool_doctor.ipynb

@@ -0,0 +1,65 @@
+{
+ "cells": [
+  {
+   "cell_type": "markdown",
+   "id": "dab47237",
+   "metadata": {},
+   "source": [
+    "# Tool doctor example"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "9735e38c",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Example 1: A reasonably well-defined tool\n",
+    "from aixtools.tools.doctor import tool_doctor\n",
+    "\n",
+    "\n",
+    "def add(a: int, b: int) -> int:\n",
+    "    \"\"\" Add two numbers \"\"\"\n",
+    "    return a + b\n",
+    "\n",
+    "# Example 2: A poorly defined tool\n",
+    "def z(a: int, b: int) -> int:\n",
+    "    \"\"\" performs some arithmentic with two parameters \"\"\"\n",
+    "    return a + b\n"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "e50d48d6",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "# Call tool doctor\n",
+    "ret = await tool_doctor([add, z])"
+   ]
+  }
+ ],
+ "metadata": {
+  "kernelspec": {
+   "display_name": ".venv",
+   "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.12.2"
+  }
+ },
+ "nbformat": 4,
+ "nbformat_minor": 5
+}

scripts/config.sh

@@ -0,0 +1,28 @@
+#-----------------------------------------------------------------------------
+#
+# This template script sets up the environment for the project by defining the
+# project directory and activating the virtual environment.
+#
+#-----------------------------------------------------------------------------
+
+export SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+
+# Project directory
+export PROJECT_DIR="$( cd $SCRIPTS_DIR/.. && pwd -P )"
+export PROJECT_NAME="$(basename $PROJECT_DIR)"
+
+# Data directories
+export DATA_DIR="$PROJECT_DIR/data"
+export LOGS_DIR="$PROJECT_DIR/logs"
+export PG_DATA_DIR="$DATA_DIR/data/db/postgres"
+
+# Server configuration
+export PORT=8081
+
+# Activate virtual environment
+if [ "${OS-}" == "Windows_NT" ]; then
+    source .venv/Scripts/activate
+else
+    source .venv/bin/activate
+fi
+

scripts/lint.sh

@@ -0,0 +1,32 @@
+#!/bin/bash -eu
+set -o pipefail
+
+#-----------------------------------------------------------------------------
+#
+# Template script for running the linter
+#
+#-----------------------------------------------------------------------------
+
+
+# Get script directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source "$SCRIPTS_DIR/config.sh"
+
+# Check if the fix parameter is provided
+FIX_MODE=false
+if [[ $# -gt 0 && "$1" == "--fix" ]]; then
+    FIX_MODE=true
+fi
+
+# Run linter
+if [ "$FIX_MODE" = true ]; then
+    echo "Running linters in fix mode..."
+    ruff format .
+    ruff check --fix .
+else
+    echo "Running linters in check mode..."
+    ruff format --check .
+    ruff check .
+fi
+echo "Running pylint..."
+pylint aixtools/

scripts/log_view.sh

@@ -0,0 +1,18 @@
+#!/bin/bash -eu
+set -o pipefail
+
+#-----------------------------------------------------------------------------
+#
+# Template script for running the Chainlit server
+#
+#-----------------------------------------------------------------------------
+
+# Get script directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source "$SCRIPTS_DIR/config.sh"
+
+# Activate virtual environment (with special handling for Windows)
+cd "$PROJECT_DIR"
+
+# Run the server
+streamlit run aixtools/log_view/app.py $*

scripts/run_example_mcp_server.sh

@@ -0,0 +1,14 @@
+#!/bin/bash -eu
+set -o pipefail
+
+# Get script directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source "$SCRIPTS_DIR/config.sh"
+
+# Activate virtual environment (with special handling for Windows)
+cd "$PROJECT_DIR"
+
+# Run the example MCP server
+fastmcp run \
+    --transport http \
+    "$PROJECT_DIR/aixtools/mcp/example_server.py"

scripts/run_faulty_mcp_server.sh

@@ -0,0 +1,13 @@
+#!/bin/bash -eu
+set -o pipefail
+
+# Get script directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source "$SCRIPTS_DIR/config.sh"
+
+cd "$PROJECT_DIR"
+
+while true; do
+    uv run aixtools/mcp/faulty_mcp.py "$@" || true
+    echo "MCP server terminated. Restarting..."
+done

scripts/run_server.sh

@@ -0,0 +1,29 @@
+#!/bin/bash -eu
+set -o pipefail
+
+#-----------------------------------------------------------------------------
+#
+# Template script for running the Chainlit server
+#
+#-----------------------------------------------------------------------------
+
+# Get script directory
+SCRIPTS_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+source "$SCRIPTS_DIR/config.sh"
+
+# Activate virtual environment (with special handling for Windows)
+cd "$PROJECT_DIR"
+
+# Run linter before starting the server
+"$SCRIPTS_DIR/lint.sh"
+
+# Go into the sub-dir (most projects don't need this)
+cd "$PROJECT_NAME"
+
+# Run the server
+chainlit run \
+    --host 0.0.0.0 \
+    --port $PORT \
+    --root-path /$PROJECT_NAME \
+    --watch \
+    app.py

aixtools-0.1.0.dist-info/top_level.txt

@@ -1 +0,0 @@
-aixtools