@agentpolicyspecification/ollama 0.1.0

package/LICENSE

@@ -0,0 +1,187 @@
+                                 Apache License
+                           Version 2.0, January 2004
+                        http://www.apache.org/licenses/
+
+   TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
+
+   1. Definitions.
+
+      "License" shall mean the terms and conditions for use, reproduction,
+      and distribution as defined by Sections 1 through 9 of this document.
+
+      "Licensor" shall mean the copyright owner or entity authorized by
+      the copyright owner that is granting the License.
+
+      "Legal Entity" shall mean the union of the acting entity and all
+      other entities that control, are controlled by, or are under common
+      control with that entity. For the purposes of this definition,
+      "control" means (i) the power, direct or indirect, to cause the
+      direction or management of such entity, whether by contract or
+      otherwise, or (ii) ownership of fifty percent (50%) or more of the
+      outstanding shares, or (iii) beneficial ownership of such entity.
+
+      "You" (or "Your") shall mean an individual or Legal Entity
+      exercising permissions granted by this License.
+
+      "Source" form shall mean the preferred form for making modifications,
+      including but not limited to software source code, documentation
+      source, and configuration files.
+
+      "Object" form shall mean any form resulting from mechanical
+      transformation or translation of a Source form, including but
+      not limited to compiled object code, generated documentation,
+      and conversions to other media types.
+
+      "Work" shall mean the work of authorship made available under
+      the License, as indicated by a copyright notice that is included in
+      or attached to the work (an example is provided in the Appendix below).
+
+      "Derivative Works" shall mean any work, whether in Source or Object
+      form, that is based on (or derived from) the Work and for which the
+      editorial revisions, annotations, elaborations, or other transformations
+      represent, as a whole, an original work of authorship. For the purposes
+      of this License, Derivative Works shall not include works that remain
+      separable from, or merely link (or bind by name) to the interfaces of,
+      the Work and Derivative Works thereof.
+
+      "Contribution" shall mean, as submitted to the Licensor for inclusion
+      in the Work by the copyright owner or by an individual or Legal Entity
+      authorized to submit on behalf of the copyright owner. For the purposes
+      of this definition, "submitted" means any form of electronic, verbal,
+      or written communication sent to the Licensor or its representatives,
+      including but not limited to communication on electronic mailing lists,
+      source code control systems, and issue tracking systems that are managed
+      by, or on behalf of, the Licensor for the purpose of discussing and
+      improving the Work, but excluding communication that is conspicuously
+      marked or otherwise designated in writing by the copyright owner as
+      "Not a Contribution."
+
+      "Contributor" shall mean Licensor and any Legal Entity on behalf of
+      whom a Contribution has been received by the Licensor and included
+      within the Work.
+
+   2. Grant of Copyright License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      copyright license to reproduce, prepare Derivative Works of,
+      publicly display, publicly perform, sublicense, and distribute the
+      Work and such Derivative Works in Source or Object form.
+
+   3. Grant of Patent License. Subject to the terms and conditions of
+      this License, each Contributor hereby grants to You a perpetual,
+      worldwide, non-exclusive, no-charge, royalty-free, irrevocable
+      (except as stated in this section) patent license to make, have made,
+      use, offer to sell, sell, import, and otherwise transfer the Work,
+      where such license applies only to those patent claims licensable
+      by such Contributor that are necessarily infringed by their
+      Contribution(s) alone or by the combination of their Contribution(s)
+      with the Work to which such Contribution(s) was submitted. If You
+      institute patent litigation against any entity (including a cross-claim
+      or counterclaim in a lawsuit) alleging that the Work or any Contribution
+      embodied within the Work constitutes direct or contributory patent
+      infringement, then any patent licenses granted to You under this License
+      for that Work shall terminate as of the date such litigation is filed.
+
+   4. Redistribution. You may reproduce and distribute copies of the
+      Work or Derivative Works thereof in any medium, with or without
+      modifications, and in Source or Object form, provided that You
+      meet the following conditions:
+
+      (a) You must give any other recipients of the Work or Derivative
+          Works a copy of this License; and
+
+      (b) You must cause any modified files to carry prominent notices
+          stating that You changed the files; and
+
+      (c) You must retain, in the Source form of any Derivative Works
+          that You distribute, all copyright, patent, trademark, and
+          attribution notices from the Source form of the Work,
+          excluding those notices that do not pertain to any part of
+          the Derivative Works; and
+
+      (d) If the Work includes a "NOTICE" text file as part of its
+          distribution, You must include a readable copy of the
+          attribution notices contained within such NOTICE file, in
+          at least one of the following places: within a NOTICE text
+          file distributed as part of the Derivative Works; within
+          the Source form or documentation, if provided along with the
+          Derivative Works; or, within a display generated by the
+          Derivative Works, if and wherever such third-party notices
+          normally appear. The contents of the NOTICE file are for
+          informational purposes only and do not modify the License.
+          You may add Your own attribution notices within Derivative
+          Works that You distribute, alongside or in addition to the
+          NOTICE text from the Work, provided that such additional
+          attribution notices cannot be construed as modifying the License.
+
+      You may add Your own license statement for Your modifications and
+      may provide additional grant of rights to use, copy, modify, merge,
+      publish, distribute, sublicense, and/or sell copies of the
+      Contribution, either on its own or as part of the Work.
+
+   5. Submission of Contributions. Unless You explicitly state otherwise,
+      any Contribution intentionally submitted for inclusion in the Work
+      by You to the Licensor shall be under the terms and conditions of
+      this License, without any additional terms or conditions.
+      Notwithstanding the above, nothing herein shall supersede or modify
+      the terms of any separate license agreement you may have executed
+      with Licensor regarding such Contributions.
+
+   6. Trademarks. This License does not grant permission to use the trade
+      names, trademarks, service marks, or product names of the Licensor,
+      except as required for reasonable and customary use in describing the
+      origin of the Work and reproducing the content of the NOTICE file.
+
+   7. Disclaimer of Warranty. Unless required by applicable law or
+      agreed to in writing, Licensor provides the Work (and each
+      Contributor provides its Contributions) on an "AS IS" BASIS,
+      WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+      implied, including, without limitation, any warranties or conditions
+      of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
+      PARTICULAR PURPOSE. You are solely responsible for determining the
+      appropriateness of using or reproducing the Work and assume any
+      risks associated with Your exercise of permissions under this License.
+
+   8. Limitation of Liability. In no event and under no legal theory,
+      whether in tort (including negligence), contract, or otherwise,
+      unless required by applicable law (such as deliberate and grossly
+      negligent acts) or agreed to in writing, shall any Contributor be
+      liable to You for damages, including any direct, indirect, special,
+      incidental, or exemplary damages of any character arising as a
+      result of this License or out of the use or inability to use the
+      Work (including but not limited to damages for loss of goodwill,
+      work stoppage, computer failure or malfunction, or all other
+      commercial damages or losses), even if such Contributor has been
+      advised of the possibility of such damages.
+
+   9. Accepting Warranty or Liability. While redistributing the Work or
+      Derivative Works thereof, You may choose to offer, and charge a fee
+      for, acceptance of support, warranty, indemnity, or other liability
+      obligations and/or rights consistent with this License. However, in
+      accepting such obligations, You may offer only on behalf with
+      conditions fully consistent with this License.
+
+   END OF TERMS AND CONDITIONS
+
+   APPENDIX: How to apply the Apache License to your work.
+
+      To apply the Apache License to your work, attach the following
+      boilerplate notice, with the fields enclosed by brackets "[]"
+      replaced with your own identifying information. (Don't include
+      the brackets!)  The text should be enclosed in the appropriate
+      comment syntax for the format in use. Please also include a
+      "NOTICE" file to clarify that the license applies to your work.
+
+   Copyright [yyyy] [name of copyright owner]
+
+   Licensed under the Apache License, Version 2.0 (the "License");
+   you may not use this file except in compliance with the License.
+   You may obtain a copy of the License at
+
+       http://www.apache.org/licenses/LICENSE-2.0
+
+   Unless required by applicable law or agreed to in writing, software
+   distributed under the License is distributed on an "AS IS" BASIS,
+   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+   See the License for the specific language governing permissions and
+   limitations under the License.

package/README.md

@@ -0,0 +1,13 @@
+# @agentpolicyspecification/ollama
+
+Ollama integration for the Agent Policy Specification (APS)
+
+## Installation
+
+```bash
+npm install @agentpolicyspecification/ollama
+```
+
+## License
+
+Apache-2.0

package/dist/index.d.ts

@@ -0,0 +1,17 @@
+import type { Ollama, ChatRequest, ChatResponse } from 'ollama';
+import type { ApsEngine, Metadata } from '@agentpolicyspecification/core';
+export interface ApsOllamaOptions {
+    engine: ApsEngine;
+    metadata?: Partial<Metadata>;
+}
+export type ApsOllamaClient = {
+    chat(request: ChatRequest & {
+        stream?: false;
+    }): Promise<ChatResponse>;
+    chat(request: ChatRequest & {
+        stream: true;
+    }): Promise<AsyncIterable<ChatResponse>>;
+    chat(request: ChatRequest): Promise<ChatResponse | AsyncIterable<ChatResponse>>;
+};
+export declare function withAps(ollama: Ollama, options: ApsOllamaOptions): ApsOllamaClient;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,EAAE,WAAW,EAAE,YAAY,EAA4B,MAAM,QAAQ,CAAC;AAC1F,OAAO,KAAK,EAAE,SAAS,EAAyB,QAAQ,EAAkC,MAAM,gCAAgC,CAAC;AAEjI,MAAM,WAAW,gBAAgB;IAC/B,MAAM,EAAE,SAAS,CAAC;IAClB,QAAQ,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;CAC9B;AAED,MAAM,MAAM,eAAe,GAAG;IAC5B,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG;QAAE,MAAM,CAAC,EAAE,KAAK,CAAA;KAAE,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IACvE,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG;QAAE,MAAM,EAAE,IAAI,CAAA;KAAE,GAAG,OAAO,CAAC,aAAa,CAAC,YAAY,CAAC,CAAC,CAAC;IACpF,IAAI,CAAC,OAAO,EAAE,WAAW,GAAG,OAAO,CAAC,YAAY,GAAG,aAAa,CAAC,YAAY,CAAC,CAAC,CAAC;CACjF,CAAC;AAEF,wBAAgB,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,gBAAgB,GAAG,eAAe,CAsBlF"}

package/dist/index.js

@@ -0,0 +1,80 @@
+export function withAps(ollama, options) {
+    return {
+        async chat(request) {
+            const messages = request.messages ?? [];
+            await options.engine.evaluateInput(toInputContext(messages, options));
+            if (request.stream === true) {
+                const stream = await ollama.chat({ ...request, stream: true });
+                return streamWithEval(stream, options);
+            }
+            const response = await ollama.chat({ ...request, stream: false });
+            await evaluateToolCalls(response, options);
+            if (response.message.content) {
+                await options.engine.evaluateOutput(toOutputContext(response.message.content, options));
+            }
+            return response;
+        },
+    };
+}
+async function evaluateToolCalls(response, options) {
+    const toolCalls = response.message.tool_calls ?? [];
+    console.log('evaluate tool calls:', toolCalls);
+    for (const toolCall of toolCalls) {
+        await options.engine.evaluateToolCall(toToolCallContext(toolCall, response.message.content, options));
+    }
+}
+async function* streamWithEval(stream, options) {
+    const textChunks = [];
+    let lastChunk;
+    for await (const chunk of stream) {
+        if (chunk.message.content) {
+            textChunks.push(chunk.message.content);
+        }
+        lastChunk = chunk;
+        yield chunk;
+    }
+    // Tool calls arrive in the final chunk
+    if (lastChunk) {
+        await evaluateToolCalls(lastChunk, options);
+    }
+    const text = textChunks.join('');
+    if (text) {
+        await options.engine.evaluateOutput(toOutputContext(text, options));
+    }
+}
+function toInputContext(messages, options) {
+    return {
+        messages: messages.map(toApsMessage),
+        metadata: getMetadata(options),
+    };
+}
+function toOutputContext(text, options) {
+    return {
+        response: { role: 'assistant', content: text },
+        metadata: getMetadata(options),
+    };
+}
+function toToolCallContext(toolCall, assistantContent, options) {
+    return {
+        tool_name: toolCall.function.name,
+        arguments: toolCall.function.arguments,
+        calling_message: { role: 'assistant', content: assistantContent },
+        metadata: getMetadata(options),
+    };
+}
+function toApsMessage(m) {
+    const role = m.role === 'user' || m.role === 'assistant' || m.role === 'system'
+        ? m.role
+        : 'user';
+    const content = typeof m.content === 'string' ? m.content : '';
+    return { role, content };
+}
+function getMetadata(options) {
+    return {
+        agent_id: options.metadata?.agent_id ?? 'unknown',
+        session_id: options.metadata?.session_id ?? 'unknown',
+        timestamp: new Date().toISOString(),
+        ...options.metadata,
+    };
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAcA,MAAM,UAAU,OAAO,CAAC,MAAc,EAAE,OAAyB;IAC/D,OAAO;QACL,KAAK,CAAC,IAAI,CAAC,OAAoB;YAC7B,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YACxC,MAAM,OAAO,CAAC,MAAM,CAAC,aAAa,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC,CAAC;YAEtE,IAAI,OAAO,CAAC,MAAM,KAAK,IAAI,EAAE,CAAC;gBAC5B,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC/D,OAAO,cAAc,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YACzC,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,MAAM,CAAC,IAAI,CAAC,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;YAElE,MAAM,iBAAiB,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YAE3C,IAAI,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC7B,MAAM,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;YAC1F,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;KACiB,CAAC;AACvB,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,QAAsB,EAAE,OAAyB;IAChF,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,UAAU,IAAI,EAAE,CAAC;IACpD,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,SAAS,CAAC,CAAC;IAC/C,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;QACjC,MAAM,OAAO,CAAC,MAAM,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,QAAQ,EAAE,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC;IACxG,CAAC;AACH,CAAC;AAED,KAAK,SAAS,CAAC,CAAC,cAAc,CAC5B,MAAmC,EACnC,OAAyB;IAEzB,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,IAAI,SAAmC,CAAC;IAExC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QACjC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC1B,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QACD,SAAS,GAAG,KAAK,CAAC;QAClB,MAAM,KAAK,CAAC;IACd,CAAC;IAED,uCAAuC;IACvC,IAAI,SAAS,EAAE,CAAC;QACd,MAAM,iBAAiB,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACjC,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,OAAO,CAAC,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC;IACtE,CAAC;AACH,CAAC;AAED,SAAS,cAAc,CAAC,QAAyB,EAAE,OAAyB;IAC1E,OAAO;QACL,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,YAAY,CAAC;QACpC,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC;KAC/B,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,IAAY,EAAE,OAAyB;IAC9D,OAAO;QACL,QAAQ,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,IAAI,EAAE;QAC9C,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC;KAC/B,CAAC;AACJ,CAAC;AAED,SAAS,iBAAiB,CACxB,QAA4E,EAC5E,gBAAwB,EACxB,OAAyB;IAEzB,OAAO;QACL,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;QACjC,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;QACtC,eAAe,EAAE,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,gBAAgB,EAAE;QACjE,QAAQ,EAAE,WAAW,CAAC,OAAO,CAAC;KAC/B,CAAC;AACJ,CAAC;AAED,SAAS,YAAY,CAAC,CAAgB;IACpC,MAAM,IAAI,GAAG,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,KAAK,WAAW,IAAI,CAAC,CAAC,IAAI,KAAK,QAAQ;QAC7E,CAAC,CAAC,CAAC,CAAC,IAAI;QACR,CAAC,CAAC,MAAM,CAAC;IACX,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/D,OAAO,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC;AAC3B,CAAC;AAED,SAAS,WAAW,CAAC,OAAyB;IAC5C,OAAO;QACL,QAAQ,EAAE,OAAO,CAAC,QAAQ,EAAE,QAAQ,IAAI,SAAS;QACjD,UAAU,EAAE,OAAO,CAAC,QAAQ,EAAE,UAAU,IAAI,SAAS;QACrD,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;QACnC,GAAG,OAAO,CAAC,QAAQ;KACpB,CAAC;AACJ,CAAC"}

package/jest.config.js

@@ -0,0 +1,16 @@
+export default {
+  preset: "ts-jest/presets/default-esm",
+  testEnvironment: "node",
+  testMatch: ["**/test/**/*.test.ts"],
+  transform: {
+    "^.+\\.tsx?$": [
+      "ts-jest",
+      {
+        useESM: true,
+      },
+    ],
+  },
+  moduleNameMapper: {
+    "^(\\.{1,2}/.*)\\.js$": "$1",
+  },
+};

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@agentpolicyspecification/ollama",
+  "version": "0.1.0",
+  "type": "module",
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "dependencies": {
+    "@agentpolicyspecification/core": "0.1.0"
+  },
+  "peerDependencies": {
+    "ollama": ">=0.5.0"
+  },
+  "devDependencies": {
+    "@jest/globals": "^29.5.0",
+    "@types/jest": "^29.5.0",
+    "@types/node": "^20.0.0",
+    "jest": "^29.5.0",
+    "ollama": "^0.5.0",
+    "ts-jest": "^29.1.0",
+    "typescript": "^5.4.0"
+  },
+  "author": "Pascal Wilbrink",
+  "license": "Apache-2.0",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/agentpolicyspecification/aps-typescript.git",
+    "directory": "integrations/ollama"
+  },
+  "bugs": {
+    "url": "https://github.com/agentpolicyspecification/aps-typescript/issues"
+  },
+  "homepage": "https://github.com/agentpolicyspecification/aps-typescript/tree/main/integrations/ollama#readme",
+  "publishConfig": {
+    "access": "public"
+  },
+  "scripts": {
+    "build": "tsc",
+    "test": "NODE_OPTIONS=--experimental-vm-modules jest --passWithNoTests"
+  }
+}

package/src/index.ts

@@ -0,0 +1,115 @@
+import type { Ollama, ChatRequest, ChatResponse, Message as OllamaMessage } from 'ollama';
+import type { ApsEngine, InputContext, Message, Metadata, OutputContext, ToolCallContext } from '@agentpolicyspecification/core';
+
+export interface ApsOllamaOptions {
+  engine: ApsEngine;
+  metadata?: Partial<Metadata>;
+}
+
+export type ApsOllamaClient = {
+  chat(request: ChatRequest & { stream?: false }): Promise<ChatResponse>;
+  chat(request: ChatRequest & { stream: true }): Promise<AsyncIterable<ChatResponse>>;
+  chat(request: ChatRequest): Promise<ChatResponse | AsyncIterable<ChatResponse>>;
+};
+
+export function withAps(ollama: Ollama, options: ApsOllamaOptions): ApsOllamaClient {
+  return {
+    async chat(request: ChatRequest): Promise<ChatResponse | AsyncIterable<ChatResponse>> {
+      const messages = request.messages ?? [];
+      await options.engine.evaluateInput(toInputContext(messages, options));
+
+      if (request.stream === true) {
+        const stream = await ollama.chat({ ...request, stream: true });
+        return streamWithEval(stream, options);
+      }
+
+      const response = await ollama.chat({ ...request, stream: false });
+
+      await evaluateToolCalls(response, options);
+
+      if (response.message.content) {
+        await options.engine.evaluateOutput(toOutputContext(response.message.content, options));
+      }
+
+      return response;
+    },
+  } as ApsOllamaClient;
+}
+
+async function evaluateToolCalls(response: ChatResponse, options: ApsOllamaOptions): Promise<void> {
+  const toolCalls = response.message.tool_calls ?? [];
+  console.log('evaluate tool calls:', toolCalls);
+  for (const toolCall of toolCalls) {
+    await options.engine.evaluateToolCall(toToolCallContext(toolCall, response.message.content, options));
+  }
+}
+
+async function* streamWithEval(
+  stream: AsyncIterable<ChatResponse>,
+  options: ApsOllamaOptions,
+): AsyncGenerator<ChatResponse> {
+  const textChunks: string[] = [];
+  let lastChunk: ChatResponse | undefined;
+
+  for await (const chunk of stream) {
+    if (chunk.message.content) {
+      textChunks.push(chunk.message.content);
+    }
+    lastChunk = chunk;
+    yield chunk;
+  }
+
+  // Tool calls arrive in the final chunk
+  if (lastChunk) {
+    await evaluateToolCalls(lastChunk, options);
+  }
+
+  const text = textChunks.join('');
+  if (text) {
+    await options.engine.evaluateOutput(toOutputContext(text, options));
+  }
+}
+
+function toInputContext(messages: OllamaMessage[], options: ApsOllamaOptions): InputContext {
+  return {
+    messages: messages.map(toApsMessage),
+    metadata: getMetadata(options),
+  };
+}
+
+function toOutputContext(text: string, options: ApsOllamaOptions): OutputContext {
+  return {
+    response: { role: 'assistant', content: text },
+    metadata: getMetadata(options),
+  };
+}
+
+function toToolCallContext(
+  toolCall: { function: { name: string; arguments: Record<string, unknown> } },
+  assistantContent: string,
+  options: ApsOllamaOptions,
+): ToolCallContext {
+  return {
+    tool_name: toolCall.function.name,
+    arguments: toolCall.function.arguments,
+    calling_message: { role: 'assistant', content: assistantContent },
+    metadata: getMetadata(options),
+  };
+}
+
+function toApsMessage(m: OllamaMessage): Message {
+  const role = m.role === 'user' || m.role === 'assistant' || m.role === 'system'
+    ? m.role
+    : 'user';
+  const content = typeof m.content === 'string' ? m.content : '';
+  return { role, content };
+}
+
+function getMetadata(options: ApsOllamaOptions): Metadata {
+  return {
+    agent_id: options.metadata?.agent_id ?? 'unknown',
+    session_id: options.metadata?.session_id ?? 'unknown',
+    timestamp: new Date().toISOString(),
+    ...options.metadata,
+  };
+}

package/test/index.test.ts

@@ -0,0 +1,193 @@
+import { jest, describe, it, expect } from '@jest/globals';
+import type { Ollama, ChatResponse } from 'ollama';
+import { withAps } from '../src/index.js';
+import { ApsEngine, PolicyDenialError } from '@agentpolicyspecification/core';
+import type { InputContext, OutputContext, ToolCallContext, PolicyDecision, PolicySet } from '@agentpolicyspecification/core';
+
+// ─── Helpers ─────────────────────────────────────────────────────────────────
+
+const ALLOW_DECISION: PolicyDecision = { decision: 'allow' };
+const DENY_DECISION: PolicyDecision = { decision: 'deny', reason: 'blocked by policy' };
+
+function makeMockOllama(reply: string, toolCalls?: ChatResponse['message']['tool_calls']): { ollama: Ollama; chatSpy: ReturnType<typeof jest.fn> } {
+  const chatSpy = jest.fn<() => Promise<ChatResponse>>().mockResolvedValue({
+    message: { role: 'assistant', content: reply, tool_calls: toolCalls },
+  } as ChatResponse);
+  return { ollama: { chat: chatSpy } as unknown as Ollama, chatSpy };
+}
+
+function makeStreamingMockOllama(chunks: string[]): { ollama: Ollama; chatSpy: ReturnType<typeof jest.fn> } {
+  async function* gen(): AsyncGenerator<ChatResponse> {
+    for (const chunk of chunks) {
+      yield { message: { role: 'assistant', content: chunk } } as ChatResponse;
+    }
+  }
+  const chatSpy = jest.fn<() => Promise<AsyncGenerator<ChatResponse>>>().mockResolvedValue(gen());
+  return { ollama: { chat: chatSpy } as unknown as Ollama, chatSpy };
+}
+
+function makeEngine(policySet: PolicySet): ApsEngine {
+  return new ApsEngine({ policySet });
+}
+
+// ─── Tests ────────────────────────────────────────────────────────────────────
+
+describe('withAps – non-streaming', () => {
+  it('passes through when all policies allow', async () => {
+    const { ollama } = makeMockOllama('Madrid');
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      output: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+    });
+
+    const client = withAps(ollama, { engine, metadata: { agent_id: 'test', session_id: 's1' } });
+    const response = await client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'Capital of Spain?' }] });
+
+    expect(response.message.content).toBe('Madrid');
+  });
+
+  it('throws PolicyDenialError when input policy denies', async () => {
+    const { ollama, chatSpy } = makeMockOllama('Madrid');
+    const engine = makeEngine({
+      input: [{ id: 'no-pii', evaluate: () => DENY_DECISION }],
+    });
+
+    const client = withAps(ollama, { engine });
+    await expect(
+      client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'My SSN is 123-45-6789' }] }),
+    ).rejects.toBeInstanceOf(PolicyDenialError);
+
+    expect(chatSpy).not.toHaveBeenCalled();
+  });
+
+  it('throws PolicyDenialError when output policy denies', async () => {
+    const { ollama } = makeMockOllama('secret information');
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      output: [{ id: 'no-secrets', evaluate: () => DENY_DECISION }],
+    });
+
+    const client = withAps(ollama, { engine });
+    await expect(
+      client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'Tell me something' }] }),
+    ).rejects.toBeInstanceOf(PolicyDenialError);
+  });
+
+  it('forwards metadata to engine context', async () => {
+    const inputEvaluate = jest.fn<(ctx: InputContext) => PolicyDecision>().mockReturnValue(ALLOW_DECISION);
+    const { ollama } = makeMockOllama('ok');
+    const engine = makeEngine({
+      input: [{ id: 'check-meta', evaluate: inputEvaluate as (ctx: InputContext) => PolicyDecision }],
+    });
+
+    const client = withAps(ollama, { engine, metadata: { agent_id: 'agent-1', session_id: 'sess-1' } });
+    await client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'hi' }] });
+
+    const ctx = inputEvaluate.mock.calls[0]?.[0] as InputContext;
+    expect(ctx.metadata?.agent_id).toBe('agent-1');
+    expect(ctx.metadata?.session_id).toBe('sess-1');
+  });
+});
+
+describe('withAps – streaming', () => {
+  it('yields all chunks and evaluates output after stream ends', async () => {
+    const chunks = ['Hello', ', ', 'world!'];
+    const { ollama } = makeStreamingMockOllama(chunks);
+    const outputEvaluate = jest.fn<(ctx: OutputContext) => PolicyDecision>().mockReturnValue(ALLOW_DECISION);
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      output: [{ id: 'check-output', evaluate: outputEvaluate as (ctx: OutputContext) => PolicyDecision }],
+    });
+
+    const client = withAps(ollama, { engine });
+    const stream = await client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'hi' }], stream: true });
+
+    const received: string[] = [];
+    for await (const chunk of stream as AsyncIterable<ChatResponse>) {
+      received.push(chunk.message.content);
+    }
+
+    expect(received).toEqual(chunks);
+
+    const outputCtx = outputEvaluate.mock.calls[0]?.[0] as OutputContext;
+    expect(outputCtx.response?.content).toBe('Hello, world!');
+  });
+
+  it('throws PolicyDenialError when input policy denies (streaming)', async () => {
+    const { ollama, chatSpy } = makeStreamingMockOllama(['hi']);
+    const engine = makeEngine({
+      input: [{ id: 'deny-all', evaluate: () => DENY_DECISION }],
+    });
+
+    const client = withAps(ollama, { engine });
+    await expect(
+      client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'hi' }], stream: true }),
+    ).rejects.toBeInstanceOf(PolicyDenialError);
+
+    expect(chatSpy).not.toHaveBeenCalled();
+  });
+});
+
+describe('withAps – tool calls', () => {
+  it('evaluates each tool call returned by the model', async () => {
+    const toolCalls = [
+      { function: { name: 'get_weather', arguments: { city: 'Madrid' } } },
+      { function: { name: 'search_web', arguments: { query: 'news' } } },
+    ];
+    const { ollama } = makeMockOllama('', toolCalls);
+    const toolEvaluate = jest.fn<(ctx: ToolCallContext) => PolicyDecision>().mockReturnValue(ALLOW_DECISION);
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      tool_call: [{ id: 'check-tools', evaluate: toolEvaluate as (ctx: ToolCallContext) => PolicyDecision }],
+    });
+
+    const client = withAps(ollama, { engine });
+    await client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'What is the weather?' }] });
+
+    expect(toolEvaluate).toHaveBeenCalledTimes(2);
+    const ctx0 = toolEvaluate.mock.calls[0]?.[0] as ToolCallContext;
+    expect(ctx0.tool_name).toBe('get_weather');
+    expect(ctx0.arguments).toEqual({ city: 'Madrid' });
+    const ctx1 = toolEvaluate.mock.calls[1]?.[0] as ToolCallContext;
+    expect(ctx1.tool_name).toBe('search_web');
+  });
+
+  it('throws PolicyDenialError when a tool call is denied', async () => {
+    const toolCalls = [{ function: { name: 'delete_file', arguments: { path: '/etc/passwd' } } }];
+    const { ollama } = makeMockOllama('', toolCalls);
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      tool_call: [{ id: 'no-dangerous-tools', evaluate: () => DENY_DECISION }],
+    });
+
+    const client = withAps(ollama, { engine });
+    await expect(
+      client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'Delete the file' }] }),
+    ).rejects.toBeInstanceOf(PolicyDenialError);
+  });
+
+  it('evaluates tool calls from the final chunk in streaming mode', async () => {
+    const toolCalls = [{ function: { name: 'get_weather', arguments: { city: 'Paris' } } }];
+    async function* gen(): AsyncGenerator<ChatResponse> {
+      yield { message: { role: 'assistant', content: 'Checking...' } } as ChatResponse;
+      yield { message: { role: 'assistant', content: '', tool_calls: toolCalls } } as unknown as ChatResponse;
+    }
+    const chatSpy = jest.fn<() => Promise<AsyncGenerator<ChatResponse>>>().mockResolvedValue(gen());
+    const ollama = { chat: chatSpy } as unknown as Ollama;
+
+    const toolEvaluate = jest.fn<(ctx: ToolCallContext) => PolicyDecision>().mockReturnValue(ALLOW_DECISION);
+    const engine = makeEngine({
+      input: [{ id: 'allow-all', evaluate: () => ALLOW_DECISION }],
+      tool_call: [{ id: 'check-tools', evaluate: toolEvaluate as (ctx: ToolCallContext) => PolicyDecision }],
+    });
+
+    const client = withAps(ollama, { engine });
+    const stream = await client.chat({ model: 'llama3', messages: [{ role: 'user', content: 'Weather?' }], stream: true });
+    for await (const _ of stream as AsyncIterable<ChatResponse>) { /* consume */ }
+
+    expect(toolEvaluate).toHaveBeenCalledTimes(1);
+    const ctx = toolEvaluate.mock.calls[0]?.[0] as ToolCallContext;
+    expect(ctx.tool_name).toBe('get_weather');
+    expect(ctx.arguments).toEqual({ city: 'Paris' });
+  });
+});

package/tsconfig.json

@@ -0,0 +1,18 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "module": "NodeNext",
+    "moduleResolution": "NodeNext",
+    "lib": ["ES2022"],
+    "outDir": "./dist",
+    "rootDir": "./src",
+    "declaration": true,
+    "declarationMap": true,
+    "sourceMap": true,
+    "strict": true,
+    "noUncheckedIndexedAccess": true,
+    "esModuleInterop": true,
+    "skipLibCheck": true
+  },
+  "include": ["src/**/*"]
+}