aixtools 0.1.10__py3-none-any.whl → 0.1.11__py3-none-any.whl

aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.10'
-__version_tuple__ = version_tuple = (0, 1, 10)
+__version__ = version = '0.1.11'
+__version_tuple__ = version_tuple = (0, 1, 11)
 
 __commit_id__ = commit_id = None

aixtools/mcp/client.py

@@ -1,12 +1,18 @@
 """MCP server utilities with caching and robust error handling."""
 
 import asyncio
-from typing import Any
+from contextlib import asynccontextmanager
+from datetime import timedelta
+from typing import Any, AsyncGenerator
 
 import anyio
+import httpx
+from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
 from cachebox import TTLCache
 from mcp import types as mcp_types
+from mcp.client import streamable_http
 from mcp.shared.exceptions import McpError
+from mcp.shared.message import SessionMessage
 from pydantic_ai import RunContext, exceptions
 from pydantic_ai.mcp import MCPServerStreamableHTTP, ToolResult
 from pydantic_ai.toolsets.abstract import ToolsetTool
@@ -16,6 +22,7 @@ from aixtools.logging.logging_config import get_logger
 
 MCP_TOOL_CACHE_TTL = 300  # 5 minutes
 DEFAULT_MCP_CONNECTION_TIMEOUT = 30
+DEFAULT_MCP_READ_TIMEOUT = float(60 * 5)  # 5 minutes
 CACHE_KEY = "TOOL_LIST"
 
 logger = get_logger(__name__)
@@ -145,6 +152,23 @@ class CachedMCPServerStreamableHTTP(MCPServerStreamableHTTP):
                 logger.warning("MCP %s: %s exception %s: %s", self.url, func.__name__, type(exc), exc)
             return fallback(exc)
 
+    @property
+    def _transport_client(self):
+        """Override base transport client with wrapper logging and suppressing exceptions"""
+        return patched_streamablehttp_client
+
+    @asynccontextmanager
+    async def client_streams(self):
+        """Override base client_streams with wrapper logging and suppressing exceptions"""
+        try:
+            async with super().client_streams() as streams:  # pylint: disable=contextmanager-generator-missing-cleanup
+                try:
+                    yield streams
+                except Exception as exc:  # pylint: disable=broad-except
+                    logger.error("MCP %s: client_streams; %s: %s", self.url, type(exc).__name__, exc)
+        except Exception as exc:  # pylint: disable=broad-except
+            logger.error("MCP %s: client_streams: %s: %s", self.url, type(exc).__name__, exc)
+
     async def __aenter__(self):
         """Enter the context of the cached MCP server with complete cancellation isolation."""
         async with self._isolation_lock:
@@ -272,3 +296,80 @@ class CachedMCPServerStreamableHTTP(MCPServerStreamableHTTP):
             raise exceptions.ModelRetry(text)
 
         return content[0] if len(content) == 1 else content
+
+
+class PatchedStreamableHTTPTransport(streamable_http.StreamableHTTPTransport):
+    """Patched StreamableHTTPTransport with exception suppression for _handle_post_request."""
+
+    async def _handle_post_request(self, ctx: streamable_http.RequestContext) -> None:
+        """Patched _handle_post_request with proper error handling."""
+        try:
+            await super()._handle_post_request(ctx)
+        except Exception as exc:  # pylint: disable=broad-except
+            logger.error("MCP %s: _handle_post_request %s: %s", self.url, type(exc).__name__, exc)
+
+
+@asynccontextmanager
+async def patched_streamablehttp_client(  # noqa: PLR0913, pylint: disable=too-many-arguments,too-many-positional-arguments,too-many-locals
+    url: str,
+    headers: dict[str, str] | None = None,
+    timeout: float | timedelta = 30,
+    sse_read_timeout: float | timedelta = DEFAULT_MCP_READ_TIMEOUT,
+    terminate_on_close: bool = True,
+    httpx_client_factory: streamable_http.McpHttpClientFactory = streamable_http.create_mcp_http_client,
+    auth: httpx.Auth | None = None,
+) -> AsyncGenerator[
+    tuple[
+        MemoryObjectReceiveStream[SessionMessage | Exception],
+        MemoryObjectSendStream[SessionMessage],
+        streamable_http.GetSessionIdCallback,
+    ],
+    None,
+]:
+    """Patched version of `streamablehttp_client` with exception suppression."""
+    try:
+        transport = PatchedStreamableHTTPTransport(url, headers, timeout, sse_read_timeout, auth)
+
+        read_stream_writer, read_stream = anyio.create_memory_object_stream[SessionMessage | Exception](0)
+        write_stream, write_stream_reader = anyio.create_memory_object_stream[SessionMessage](0)
+        async with anyio.create_task_group() as tg:
+            try:
+                async with httpx_client_factory(
+                    headers=transport.request_headers,
+                    timeout=httpx.Timeout(transport.timeout, read=transport.sse_read_timeout),
+                    auth=transport.auth,
+                ) as client:
+                    # Define callbacks that need access to tg
+                    def start_get_stream() -> None:
+                        tg.start_soon(transport.handle_get_stream, client, read_stream_writer)
+
+                    tg.start_soon(
+                        transport.post_writer,
+                        client,
+                        write_stream_reader,
+                        read_stream_writer,
+                        write_stream,
+                        start_get_stream,
+                        tg,
+                    )
+
+                    try:
+                        yield (
+                            read_stream,
+                            write_stream,
+                            transport.get_session_id,
+                        )
+                    except GeneratorExit:
+                        logger.warning("patched_streamablehttp_client: GeneratorExit caught, closing streams.")
+                    finally:
+                        if transport.session_id and terminate_on_close:
+                            await transport.terminate_session(client)
+                        tg.cancel_scope.cancel()
+            finally:
+                await read_stream_writer.aclose()
+                await write_stream.aclose()
+    except Exception as exc:  # pylint: disable=broad-except
+        if str(exc) == "Attempted to exit cancel scope in a different task than it was entered in":
+            logger.warning("MCP %s: patched_streamablehttp_client: enter/exit cancel scope task mismatch.", url)
+        else:
+            logger.error("MCP %s: patched_streamablehttp_client: %s: %s", url, type(exc).__name__, exc)

aixtools/testing/aix_test_model.py

@@ -108,6 +108,8 @@ class AixTestModel(Model):
         messages: list[ModelMessage],
         model_settings: ModelSettings | None,
         model_request_parameters: ModelRequestParameters,
+        *args,  # pylint: disable=unused-argument # Accept additional arguments for compatibility with pydantic-ai 1.0.9
+        **kwargs,  # pylint: disable=unused-argument
     ) -> AsyncIterator[StreamedResponse]:
         model_response = await self._request(messages, model_settings, model_request_parameters)
         yield TestStreamedResponse(_model_name=self.model_name, _structured_response=model_response, _messages=messages)

{aixtools-0.1.10.dist-info → aixtools-0.1.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.1.10
+Version: 0.1.11
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown
@@ -16,6 +16,7 @@ Requires-Dist: langchain-chroma>=0.2.3
 Requires-Dist: langchain-ollama>=0.3.2
 Requires-Dist: langchain-openai>=0.3.14
 Requires-Dist: mcp>=1.11.0
+Requires-Dist: mypy>=1.18.2
 Requires-Dist: pandas>=2.2.3
 Requires-Dist: pydantic-ai>=0.4.10
 Requires-Dist: pylint>=3.3.7

{aixtools-0.1.10.dist-info → aixtools-0.1.11.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=uf7mpKSLRNNF3RxSXHssYzKadEgmCS7IlRw4lFPAcUg,706
+aixtools/_version.py,sha256=0-Ruc52ECccw_8Ef0d7jMkzrb8fkobUkZLqGGvcm1ik,706
 aixtools/app.py,sha256=JzQ0nrv_bjDQokllIlGHOV0HEb-V8N6k_nGQH-TEsVU,5227
 aixtools/chainlit.md,sha256=yC37Ly57vjKyiIvK4oUvf4DYxZCwH7iocTlx7bLeGLU,761
 aixtools/context.py,sha256=I_MD40ZnvRm5WPKAKqBUAdXIf8YaurkYUUHSVVy-QvU,598
@@ -52,7 +52,7 @@ aixtools/logging/mcp_logger.py,sha256=d2I5l4t0d6rQH17w23FpE1IUD8Ax-mSaKfByCH86q4
 aixtools/logging/model_patch_logging.py,sha256=MY2EvR7ZSctC4hJxNMe8iACeVayUJ2V5In2GAnKdgOo,2880
 aixtools/logging/open_telemetry.py,sha256=fJjF1ou_8GyfNfbyWDQPGK6JAUrUaPwURYPHhXEtDBE,1121
 aixtools/mcp/__init__.py,sha256=tLo2KZ1Ojo-rgEEJBGtZfUw-iOoopWoHDnYQTq3IzfE,163
-aixtools/mcp/client.py,sha256=yUcurbNQ8bRaPc8-gK_ESpcPb26d5ckDOEDQrG_YW0c,11823
+aixtools/mcp/client.py,sha256=zN5Na3Vgub2-BoJeldtFrpciBW-TrLjZUpjcQm-biqU,16661
 aixtools/mcp/example_client.py,sha256=QCFGP3NCNJMOKWjUOnFwjnbJhUSb879IA1ZYmwjRnmc,889
 aixtools/mcp/example_server.py,sha256=1SWCyrLWsAnOa81HC4QbPJo_lBVu0b3SZBWI-qDh1vQ,458
 aixtools/mcp/fast_mcp_log.py,sha256=XYOS406dVjn5YTHyGRsRvVNQ0SKlRObfrKj6EeLFjHg,1057
@@ -64,7 +64,7 @@ aixtools/server/path.py,sha256=SaIJxvmhJy3kzx5zJ6d4cKP6kKu2wFFciQkOLGTA4gg,3056
 aixtools/server/utils.py,sha256=tZWITIx6M-luV9yve4j3rPtYGSSA6zWS0JWEAySne_M,2276
 aixtools/server/workspace_privacy.py,sha256=grcj82eHSd7gFbb5f_w9nv4TWp50QyU952l0iIPoChM,2375
 aixtools/testing/__init__.py,sha256=mlmaAR2gmS4SbsYNCxnIprmFpFp-syjgVUkpUszo3mE,166
-aixtools/testing/aix_test_model.py,sha256=dlI3sdyvmu4fUs_K4-oazs_a7cE6V-gnI6RQ0_fPVxg,5925
+aixtools/testing/aix_test_model.py,sha256=KDSPwvOOzBVFym14hkXw7BWSLqkFH9nu5OVdfNZoUeQ,6099
 aixtools/testing/mock_tool.py,sha256=4I0LxxSkLhGIKM2YxCP3cnYI8IYJjdKhfwGZ3dioXsM,2465
 aixtools/testing/model_patch_cache.py,sha256=238gKC_gSpR3BkeejhetObOkpOR1l2Iz3A6B_eUTRNc,10158
 aixtools/tools/doctor/__init__.py,sha256=FPwYzC1eJyw8IH0-BP0wgxSprLy6Y_4yXCek7496f2k,64
@@ -81,45 +81,8 @@ aixtools/utils/chainlit/cl_agent_show.py,sha256=vaRuowp4BRvhxEr5hw0zHEJ7iaSF_5bo
 aixtools/utils/chainlit/cl_utils.py,sha256=fxaxdkcZg6uHdM8uztxdPowg3a2f7VR7B26VPY4t-3c,5738
 aixtools/vault/__init__.py,sha256=fsr_NuX3GZ9WZ7dGfe0gp_5-z3URxAfwVRXw7Xyc0dU,141
 aixtools/vault/vault.py,sha256=9dZLWdZQk9qN_Q9Djkofw9LUKnJqnrX5H0fGusVLBhA,6037
-docker/mcp-base/Dockerfile,sha256=uislVoTEgRF--AAiyX24sBxlDdfA1ZU5rDM94XYFqvI,1388
-docker/mcp-base/zscaler.crt,sha256=fCUNiOfJlWTA7R4zV1Xyb-XC1_nMHPoYTFGvBj1oz6s,1732
-notebooks/example_faulty_mcp_server.ipynb,sha256=b2Cy3GXfj-gOBZ7SoUzj25F1rxp5u-32EWPHWQ-sxn8,1729
-notebooks/example_mcp_server_stdio.ipynb,sha256=ya4dRKNFU2vQxob-uIhKHGAzINXGQ6MehgKVmSCpHLk,1634
-notebooks/example_raw_mcp_client.ipynb,sha256=uchaG-LuuwEpE2oIkmhZ2s1EDb19AgT1fUv2Jxtjgu8,1795
-notebooks/example_tool_doctor.ipynb,sha256=bWTlPNI1ZQStwMfr-KSkTGYckJuJmRG_e112Gr4KZ0I,1339
-scripts/config.sh,sha256=xnA_S4p2w8fuIEPB4MiTWZdyIlCh5m4XuHY_fhE68kg,820
-scripts/lint.sh,sha256=YmPcjfFVe2s-xSaddgSxOsSm9dmHnsmbiLZnuaPgXmY,744
-scripts/log_view.sh,sha256=bp8oXFRRbbHpyvHAN85wfDHTVK7vMJOYsBx_-bgECQc,511
-scripts/run_example_mcp_server.sh,sha256=f7m7h7O_wo6-nAsYlOXVWIASCOh3Qbuu0XWizlxMhl8,355
-scripts/run_faulty_mcp_server.sh,sha256=u_-8NbPDnJQt6IinNSjh8tc2ed-_MjGyipJXrUXaGR8,291
-scripts/run_server.sh,sha256=5iiB9bB5M2MuOgxVQqu7Oa_tBVtJpt0uB4z9uLu2J50,720
-scripts/test.sh,sha256=KxXWkVqctFRNP8hItJr8K27nDHEkfwNWb1UFhpBQDOk,865
-tests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/a2a/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/a2a/google_sdk/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/a2a/google_sdk/test_card.py,sha256=g8OUIX9BCtApM9y8l_nL1Q7sm3yezxu5yBxrpy76mo4,4359
-tests/unit/a2a/google_sdk/test_remote_agent_connection.py,sha256=nIY8eg32w96BAddaQ25mT-lr0ozPb6UrG-_Vpqx5RMY,17492
-tests/unit/a2a/google_sdk/test_utils.py,sha256=-eHmIk2GJH57W2bAdTzfRrUUb5jnd9Pf-QSXJogN3g8,8312
-tests/unit/a2a/google_sdk/pydantic_ai_adapter/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/a2a/google_sdk/pydantic_ai_adapter/test_agent_executor.py,sha256=PcyCw0N3y-txu2KJzufzbCjs7ZfoBBCVjpZuRBqTmOw,7722
-tests/unit/a2a/google_sdk/pydantic_ai_adapter/test_storage.py,sha256=tb67pFfvyWSaDfKaiPDNBQfl6-o17WtCMZh3lQHrYxY,5468
-tests/unit/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/agents/test_prompt.py,sha256=YWFZdH_F774hxw79gsWoTWBPVs8UjOAtJOgNXJ8N9gs,15384
-tests/unit/compliance/test_private_data.py,sha256=GjH7NCp54Bz1S-CmH_mUe53lb53kllOOJEm448OniRI,13693
-tests/unit/google/__init__.py,sha256=eRYHldBi5cFWL7oo2_t5TErI8ESmIjNvBZIcp-w8hSA,45
-tests/unit/google/test_client.py,sha256=fXR4Cozea7bdL2prM-1s9IqUQ9AheklQnHpN-4YM3gg,11005
-tests/unit/mcp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/mcp/test_client.py,sha256=n9sZvmzNzJfozvxoHweAg4M5ZLNhEizq16IjcZHGdj0,8838
-tests/unit/server/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/server/test_path.py,sha256=1QKiKLLRga9GNxmaUEt_wEZ9U14yzB-7PIhAOgB4wwo,9523
-tests/unit/server/test_utils.py,sha256=kvhzdgNfsJl5tqcRBWg2yTR5GPpyrFCOmEIOuHb3904,14848
-tests/unit/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/utils/test_files.py,sha256=AKFmXQqXstyKd2PreE4EmQyhQYeqOmu1Sp80MwHrf_Q,5782
-tests/unit/vault/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-tests/unit/vault/test_vault.py,sha256=T9V2Opxl3N5sJPftw0Q4lnVOs6urGpAmffe0cz6PUfw,10445
-aixtools-0.1.10.dist-info/METADATA,sha256=BvV1AzDpgiy7rjt6AFyUdAiDOwt_mN8CmmEo0JkM4C4,18570
-aixtools-0.1.10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.1.10.dist-info/entry_points.txt,sha256=dHoutULEZx7xXSqJrZdViSVjfInJibfLibi2nRXL3SE,56
-aixtools-0.1.10.dist-info/top_level.txt,sha256=ee4eF-0pqu45zCUVml0mWIhnXQgqMQper2-49BBVHLY,40
-aixtools-0.1.10.dist-info/RECORD,,
+aixtools-0.1.11.dist-info/METADATA,sha256=4mR141qz8BCJbuQy5iTwN0yikrPEKeUb8XPInS5CC6k,18598
+aixtools-0.1.11.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.1.11.dist-info/entry_points.txt,sha256=dHoutULEZx7xXSqJrZdViSVjfInJibfLibi2nRXL3SE,56
+aixtools-0.1.11.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
+aixtools-0.1.11.dist-info/RECORD,,

aixtools-0.1.11.dist-info/top_level.txt

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

aixtools-0.1.10.dist-info/top_level.txt

@@ -1,5 +0,0 @@
-aixtools
-docker
-notebooks
-scripts
-tests

docker/mcp-base/Dockerfile

@@ -1,33 +0,0 @@
-FROM ubuntu:22.04
-
-RUN apt-get -y update && \
-    apt-get -y install ca-certificates curl gcc git libcap2-bin sudo
-RUN mv /usr/bin/sudo /usr/sbin
-
-# Add Zscaler CA certificate
-COPY ./zscaler.crt /usr/local/share/ca-certificates/
-RUN update-ca-certificates
-ENV SSL_CERT_FILE=/etc/ssl/certs/ca-certificates.crt
-ENV CURL_CA_BUNDLE=/etc/ssl/certs/ca-certificates.crt
-
-# Install `uv` Python package manager
-RUN bash -o pipefail -c "curl -LsSf https://astral.sh/uv/install.sh | env UV_INSTALL_DIR=/usr/local/bin sh"
-
-# Add a user matching the sandbox user so that the files and folders created by the MCP server
-# are writable by the user in the sandbox containers (UID=1000 must match!)
-ENV USER=mcp_user
-RUN useradd -m -s /bin/bash -u 1000 ${USER} && \
-    echo "${USER} ALL=(ALL) NOPASSWD:ALL" > /etc/sudoers.d/${USER} && chmod 0440 /etc/sudoers.d/${USER}
-USER ${USER}
-
-# Build argument: GITHUB_PAT (optional)
-# If provided, this GitHub Personal Access Token will be used to authenticate git
-# operations against github.com during the build. This is useful for accessing private repositories such as `aixtools`.
-# If not set, git will use unauthenticated access (public repositories only).
-ARG GITHUB_PAT
-RUN if [ -n "$GITHUB_PAT" ]; then \
-    git config --global url."https://x-access-token:${GITHUB_PAT}@github.com/".insteadOf "https://github.com/"; \
-    fi
-
-WORKDIR /app
-RUN mkdir data

docker/mcp-base/zscaler.crt

@@ -1,28 +0,0 @@
------BEGIN CERTIFICATE-----
-MIIE0zCCA7ugAwIBAgIJANu+mC2Jt3uTMA0GCSqGSIb3DQEBCwUAMIGhMQswCQYD
-VQQGEwJVUzETMBEGA1UECBMKQ2FsaWZvcm5pYTERMA8GA1UEBxMIU2FuIEpvc2Ux
-FTATBgNVBAoTDFpzY2FsZXIgSW5jLjEVMBMGA1UECxMMWnNjYWxlciBJbmMuMRgw
-FgYDVQQDEw9ac2NhbGVyIFJvb3QgQ0ExIjAgBgkqhkiG9w0BCQEWE3N1cHBvcnRA
-enNjYWxlci5jb20wHhcNMTQxMjE5MDAyNzU1WhcNNDIwNTA2MDAyNzU1WjCBoTEL
-MAkGA1UEBhMCVVMxEzARBgNVBAgTCkNhbGlmb3JuaWExETAPBgNVBAcTCFNhbiBK
-b3NlMRUwEwYDVQQKEwxac2NhbGVyIEluYy4xFTATBgNVBAsTDFpzY2FsZXIgSW5j
-LjEYMBYGA1UEAxMPWnNjYWxlciBSb290IENBMSIwIAYJKoZIhvcNAQkBFhNzdXBw
-b3J0QHpzY2FsZXIuY29tMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA
-qT7STSxZRTgEFFf6doHajSc1vk5jmzmM6BWuOo044EsaTc9eVEV/HjH/1DWzZtcr
-fTj+ni205apMTlKBW3UYR+lyLHQ9FoZiDXYXK8poKSV5+Tm0Vls/5Kb8mkhVVqv7
-LgYEmvEY7HPY+i1nEGZCa46ZXCOohJ0mBEtB9JVlpDIO+nN0hUMAYYdZ1KZWCMNf
-5J/aTZiShsorN2A38iSOhdd+mcRM4iNL3gsLu99XhKnRqKoHeH83lVdfu1XBeoQz
-z5V6gA3kbRvhDwoIlTBeMa5l4yRdJAfdpkbFzqiwSgNdhbxTHnYYorDzKfr2rEFM
-dsMU0DHdeAZf711+1CunuQIDAQABo4IBCjCCAQYwHQYDVR0OBBYEFLm33UrNww4M
-hp1d3+wcBGnFTpjfMIHWBgNVHSMEgc4wgcuAFLm33UrNww4Mhp1d3+wcBGnFTpjf
-oYGnpIGkMIGhMQswCQYDVQQGEwJVUzETMBEGA1UECBMKQ2FsaWZvcm5pYTERMA8G
-A1UEBxMIU2FuIEpvc2UxFTATBgNVBAoTDFpzY2FsZXIgSW5jLjEVMBMGA1UECxMM
-WnNjYWxlciBJbmMuMRgwFgYDVQQDEw9ac2NhbGVyIFJvb3QgQ0ExIjAgBgkqhkiG
-9w0BCQEWE3N1cHBvcnRAenNjYWxlci5jb22CCQDbvpgtibd7kzAMBgNVHRMEBTAD
-AQH/MA0GCSqGSIb3DQEBCwUAA4IBAQAw0NdJh8w3NsJu4KHuVZUrmZgIohnTm0j+
-RTmYQ9IKA/pvxAcA6K1i/LO+Bt+tCX+C0yxqB8qzuo+4vAzoY5JEBhyhBhf1uK+P
-/WVWFZN/+hTgpSbZgzUEnWQG2gOVd24msex+0Sr7hyr9vn6OueH+jj+vCMiAm5+u
-kd7lLvJsBu3AO3jGWVLyPkS3i6Gf+rwAp1OsRrv3WnbkYcFf9xjuaf4z0hRCrLN2
-xFNjavxrHmsH8jPHVvgc1VD0Opja0l/BRVauTrUaoW6tE+wFG5rEcPGS80jjHK4S
-pB5iDj2mUZH1T8lzYtuZy0ZPirxmtsk3135+CKNa2OCAhhFjE0xd
------END CERTIFICATE-----

notebooks/example_faulty_mcp_server.ipynb

@@ -1,74 +0,0 @@
-{
- "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

@@ -1,76 +0,0 @@
-{
- "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

@@ -1,84 +0,0 @@
-{
- "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

@@ -1,65 +0,0 @@
-{
- "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

@@ -1,28 +0,0 @@
-#-----------------------------------------------------------------------------
-#
-# 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
-