@better-agent/ollama 0.2.0-beta.1

package/dist/index.d.mts

@@ -0,0 +1,34 @@
+import { OllamaChatProviderOptions, OllamaChatSettings, OllamaEmbeddingProviderOptions as OllamaEmbeddingProviderOptions$1, OllamaEmbeddingSettings, OllamaProviderOptions as OllamaProviderOptions$1, OllamaProviderSettings } from "ai-sdk-ollama";
+import { AgentCapabilities, AgentModel, EmbeddingGenerationModel, TextGenerationModel } from "@better-agent/core";
+
+//#region src/model.d.ts
+type CreateOllamaOptions = OllamaProviderSettings;
+type OllamaProviderOptions = {
+  ollama?: OllamaProviderOptions$1;
+};
+type OllamaTextProviderOptions = {
+  ollama?: OllamaChatProviderOptions;
+};
+type OllamaEmbeddingProviderOptions = {
+  ollama?: OllamaEmbeddingProviderOptions$1;
+};
+type OllamaGenerationProviderOptions = OllamaProviderOptions | OllamaTextProviderOptions | OllamaEmbeddingProviderOptions;
+type OllamaChatModelSettings = OllamaChatSettings;
+type OllamaEmbeddingModelSettings = OllamaEmbeddingSettings;
+type OllamaModelId = string;
+type OllamaTextModelId = OllamaModelId;
+type OllamaEmbeddingModelId = string;
+type OllamaModelOptions = OllamaChatModelSettings;
+type OllamaModel<TModelId extends OllamaModelId = OllamaModelId> = AgentModel<AgentCapabilities, undefined, OllamaProviderOptions, TModelId>;
+type OllamaTextModel<TModelId extends OllamaTextModelId = OllamaTextModelId> = TextGenerationModel<OllamaTextProviderOptions, TModelId>;
+type OllamaEmbeddingModel<TModelId extends OllamaEmbeddingModelId = OllamaEmbeddingModelId> = EmbeddingGenerationModel<OllamaEmbeddingProviderOptions, TModelId>;
+type OllamaProvider = {
+  <TModelId extends OllamaModelId>(modelId: TModelId, options?: OllamaModelOptions): OllamaModel<TModelId>;
+  text<TModelId extends OllamaTextModelId>(modelId: TModelId, settings?: OllamaChatModelSettings): OllamaTextModel<TModelId>;
+  embedding<TModelId extends OllamaEmbeddingModelId>(modelId: TModelId, settings?: OllamaEmbeddingModelSettings): OllamaEmbeddingModel<TModelId>;
+};
+declare const createOllama: (options?: CreateOllamaOptions) => OllamaProvider;
+declare const ollama: OllamaProvider;
+//#endregion
+export { type CreateOllamaOptions, type OllamaChatModelSettings, type OllamaEmbeddingModel, type OllamaEmbeddingModelId, type OllamaEmbeddingModelSettings, type OllamaEmbeddingProviderOptions, type OllamaGenerationProviderOptions, type OllamaModel, type OllamaModelId, type OllamaModelOptions, type OllamaProvider, type OllamaProviderOptions, type OllamaTextModel, type OllamaTextModelId, type OllamaTextProviderOptions, createOllama, ollama };
+//# sourceMappingURL=index.d.mts.map

package/dist/index.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.mts","names":[],"sources":["../src/model.ts"],"mappings":";;;;KAoBY,mBAAA,GAAsB,sBAAA;AAAA,KAEtB,qBAAA;EACR,MAAA,GAAS,uBAAA;AAAA;AAAA,KAED,yBAAA;EACR,MAAA,GAAS,yBAAA;AAAA;AAAA,KAED,8BAAA;EACR,MAAA,GAAS,gCAAA;AAAA;AAAA,KAED,+BAAA,GACN,qBAAA,GACA,yBAAA,GACA,8BAAA;AAAA,KAEM,uBAAA,GAA0B,kBAAA;AAAA,KAC1B,4BAAA,GAA+B,uBAAA;AAAA,KAE/B,aAAA;AAAA,KACA,iBAAA,GAAoB,aAAA;AAAA,KACpB,sBAAA;AAAA,KAEA,kBAAA,GAAqB,uBAAA;AAAA,KAErB,WAAA,kBAA6B,aAAA,GAAgB,aAAA,IAAiB,UAAA,CACtE,iBAAA,aAEA,qBAAA,EACA,QAAA;AAAA,KAEQ,eAAA,kBAAiC,iBAAA,GAAoB,iBAAA,IAC7D,mBAAA,CAAoB,yBAAA,EAA2B,QAAA;AAAA,KACvC,oBAAA,kBAAsC,sBAAA,GAAyB,sBAAA,IACvE,wBAAA,CAAyB,8BAAA,EAAgC,QAAA;AAAA,KAEjD,cAAA;EAAA,kBACU,aAAA,EACd,OAAA,EAAS,QAAA,EACT,OAAA,GAAU,kBAAA,GACX,WAAA,CAAY,QAAA;EACf,IAAA,kBAAsB,iBAAA,EAClB,OAAA,EAAS,QAAA,EACT,QAAA,GAAW,uBAAA,GACZ,eAAA,CAAgB,QAAA;EACnB,SAAA,kBAA2B,sBAAA,EACvB,OAAA,EAAS,QAAA,EACT,QAAA,GAAW,4BAAA,GACZ,oBAAA,CAAqB,QAAA;AAAA;AAAA,cAsGf,YAAA,GAAY,OAAA,GAAa,mBAAA,KAA2B,cAAA;AAAA,cAIpD,MAAA,EAAM,cAAA"}

package/dist/index.mjs

@@ -0,0 +1,66 @@
+import { aiSdkEmbeddingModel, aiSdkModel, aiSdkTextModel } from "@better-agent/ai-sdk";
+import { createOllama as createOllama$1, ollama as ollama$1 } from "ai-sdk-ollama";
+
+//#region src/model.ts
+const ollamaCapabilities = {
+	identity: { provider: "ollama" },
+	transport: { streaming: true },
+	output: {
+		structuredOutput: true,
+		supportedMimeTypes: ["text/plain", "application/json"]
+	},
+	multimodal: { input: { image: true } },
+	tools: {
+		supported: true,
+		parallelCalls: false
+	},
+	reasoning: { supported: true }
+};
+const toAiSdkLanguageModel = (model) => {
+	return model;
+};
+const toAiSdkEmbeddingModel = (model) => {
+	return model;
+};
+const createAgentModel = (provider, modelId, options) => {
+	return aiSdkModel({
+		model: toAiSdkLanguageModel(provider(modelId, options)),
+		providerId: "ollama",
+		modelId,
+		capabilities: ollamaCapabilities
+	});
+};
+const createTextModel = (provider, modelId, settings) => {
+	return aiSdkTextModel({
+		model: toAiSdkLanguageModel(provider(modelId, settings)),
+		providerId: "ollama",
+		modelId
+	});
+};
+const createEmbeddingModel = (provider, modelId, settings) => {
+	return aiSdkEmbeddingModel({
+		model: toAiSdkEmbeddingModel(provider.embedding(modelId, settings)),
+		providerId: "ollama",
+		modelId
+	});
+};
+const createProvider = (provider) => {
+	return Object.assign((modelId, options) => {
+		return createAgentModel(provider, modelId, options);
+	}, {
+		text(modelId, settings) {
+			return createTextModel(provider, modelId, settings);
+		},
+		embedding(modelId, settings) {
+			return createEmbeddingModel(provider, modelId, settings);
+		}
+	});
+};
+const createOllama = (options = {}) => {
+	return createProvider(createOllama$1(options));
+};
+const ollama = createProvider(ollama$1);
+
+//#endregion
+export { createOllama, ollama };
+//# sourceMappingURL=index.mjs.map

package/dist/index.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.mjs","names":["createAiSdkOllama","defaultAiSdkOllama"],"sources":["../src/model.ts"],"sourcesContent":["import { aiSdkEmbeddingModel, aiSdkModel, aiSdkTextModel } from \"@better-agent/ai-sdk\";\nimport type {\n    AgentCapabilities,\n    AgentModel,\n    EmbeddingGenerationModel,\n    TextGenerationModel,\n} from \"@better-agent/core\";\nimport type { EmbeddingModel, LanguageModel } from \"ai\";\nimport {\n    type OllamaChatProviderOptions as AiSdkOllamaChatProviderOptions,\n    type OllamaEmbeddingProviderOptions as AiSdkOllamaEmbeddingProviderOptions,\n    type OllamaProvider as AiSdkOllamaProvider,\n    type OllamaProviderOptions as AiSdkOllamaProviderOptions,\n    type OllamaChatSettings,\n    type OllamaEmbeddingSettings,\n    type OllamaProviderSettings,\n    createOllama as createAiSdkOllama,\n    ollama as defaultAiSdkOllama,\n} from \"ai-sdk-ollama\";\n\nexport type CreateOllamaOptions = OllamaProviderSettings;\n\nexport type OllamaProviderOptions = {\n    ollama?: AiSdkOllamaProviderOptions;\n};\nexport type OllamaTextProviderOptions = {\n    ollama?: AiSdkOllamaChatProviderOptions;\n};\nexport type OllamaEmbeddingProviderOptions = {\n    ollama?: AiSdkOllamaEmbeddingProviderOptions;\n};\nexport type OllamaGenerationProviderOptions =\n    | OllamaProviderOptions\n    | OllamaTextProviderOptions\n    | OllamaEmbeddingProviderOptions;\n\nexport type OllamaChatModelSettings = OllamaChatSettings;\nexport type OllamaEmbeddingModelSettings = OllamaEmbeddingSettings;\n\nexport type OllamaModelId = string;\nexport type OllamaTextModelId = OllamaModelId;\nexport type OllamaEmbeddingModelId = string;\n\nexport type OllamaModelOptions = OllamaChatModelSettings;\n\nexport type OllamaModel<TModelId extends OllamaModelId = OllamaModelId> = AgentModel<\n    AgentCapabilities,\n    undefined,\n    OllamaProviderOptions,\n    TModelId\n>;\nexport type OllamaTextModel<TModelId extends OllamaTextModelId = OllamaTextModelId> =\n    TextGenerationModel<OllamaTextProviderOptions, TModelId>;\nexport type OllamaEmbeddingModel<TModelId extends OllamaEmbeddingModelId = OllamaEmbeddingModelId> =\n    EmbeddingGenerationModel<OllamaEmbeddingProviderOptions, TModelId>;\n\nexport type OllamaProvider = {\n    <TModelId extends OllamaModelId>(\n        modelId: TModelId,\n        options?: OllamaModelOptions,\n    ): OllamaModel<TModelId>;\n    text<TModelId extends OllamaTextModelId>(\n        modelId: TModelId,\n        settings?: OllamaChatModelSettings,\n    ): OllamaTextModel<TModelId>;\n    embedding<TModelId extends OllamaEmbeddingModelId>(\n        modelId: TModelId,\n        settings?: OllamaEmbeddingModelSettings,\n    ): OllamaEmbeddingModel<TModelId>;\n};\n\nconst ollamaCapabilities = {\n    identity: {\n        provider: \"ollama\",\n    },\n    transport: {\n        streaming: true,\n    },\n    output: {\n        structuredOutput: true,\n        supportedMimeTypes: [\"text/plain\", \"application/json\"],\n    },\n    multimodal: {\n        input: {\n            image: true,\n        },\n    },\n    tools: {\n        supported: true,\n        parallelCalls: false,\n    },\n    reasoning: {\n        supported: true,\n    },\n} as const satisfies AgentCapabilities;\n\nconst toAiSdkLanguageModel = (model: ReturnType<AiSdkOllamaProvider>): LanguageModel => {\n    return model;\n};\n\nconst toAiSdkEmbeddingModel = (\n    model: ReturnType<AiSdkOllamaProvider[\"embedding\"]>,\n): EmbeddingModel => {\n    return model;\n};\n\nconst createAgentModel = <TModelId extends OllamaModelId>(\n    provider: AiSdkOllamaProvider,\n    modelId: TModelId,\n    options: OllamaModelOptions | undefined,\n): OllamaModel<TModelId> => {\n    return aiSdkModel<AgentCapabilities, OllamaProviderOptions, TModelId>({\n        model: toAiSdkLanguageModel(provider(modelId, options)),\n        providerId: \"ollama\",\n        modelId,\n        capabilities: ollamaCapabilities,\n    });\n};\n\nconst createTextModel = <TModelId extends OllamaTextModelId>(\n    provider: AiSdkOllamaProvider,\n    modelId: TModelId,\n    settings?: OllamaChatModelSettings,\n): OllamaTextModel<TModelId> => {\n    return aiSdkTextModel<OllamaTextProviderOptions, TModelId>({\n        model: toAiSdkLanguageModel(provider(modelId, settings)),\n        providerId: \"ollama\",\n        modelId,\n    });\n};\n\nconst createEmbeddingModel = <TModelId extends OllamaEmbeddingModelId>(\n    provider: AiSdkOllamaProvider,\n    modelId: TModelId,\n    settings?: OllamaEmbeddingModelSettings,\n): OllamaEmbeddingModel<TModelId> => {\n    return aiSdkEmbeddingModel<OllamaEmbeddingProviderOptions, TModelId>({\n        model: toAiSdkEmbeddingModel(provider.embedding(modelId, settings)),\n        providerId: \"ollama\",\n        modelId,\n    });\n};\n\nconst createProvider = (provider: AiSdkOllamaProvider): OllamaProvider => {\n    const createOllamaModel: OllamaProvider = Object.assign(\n        <TModelId extends OllamaModelId>(\n            modelId: TModelId,\n            options?: OllamaModelOptions,\n        ): OllamaModel<TModelId> => {\n            return createAgentModel(provider, modelId, options);\n        },\n        {\n            text<TModelId extends OllamaTextModelId>(\n                modelId: TModelId,\n                settings?: OllamaChatModelSettings,\n            ): OllamaTextModel<TModelId> {\n                return createTextModel(provider, modelId, settings);\n            },\n            embedding<TModelId extends OllamaEmbeddingModelId>(\n                modelId: TModelId,\n                settings?: OllamaEmbeddingModelSettings,\n            ): OllamaEmbeddingModel<TModelId> {\n                return createEmbeddingModel(provider, modelId, settings);\n            },\n        },\n    );\n\n    return createOllamaModel;\n};\n\nexport const createOllama = (options: CreateOllamaOptions = {}): OllamaProvider => {\n    return createProvider(createAiSdkOllama(options));\n};\n\nexport const ollama = createProvider(defaultAiSdkOllama);\n"],"mappings":";;;;AAuEA,MAAM,qBAAqB;CACvB,UAAU,EACN,UAAU,UACb;CACD,WAAW,EACP,WAAW,MACd;CACD,QAAQ;EACJ,kBAAkB;EAClB,oBAAoB,CAAC,cAAc,mBAAmB;EACzD;CACD,YAAY,EACR,OAAO,EACH,OAAO,MACV,EACJ;CACD,OAAO;EACH,WAAW;EACX,eAAe;EAClB;CACD,WAAW,EACP,WAAW,MACd;CACJ;AAED,MAAM,wBAAwB,UAA0D;AACpF,QAAO;;AAGX,MAAM,yBACF,UACiB;AACjB,QAAO;;AAGX,MAAM,oBACF,UACA,SACA,YACwB;AACxB,QAAO,WAA+D;EAClE,OAAO,qBAAqB,SAAS,SAAS,QAAQ,CAAC;EACvD,YAAY;EACZ;EACA,cAAc;EACjB,CAAC;;AAGN,MAAM,mBACF,UACA,SACA,aAC4B;AAC5B,QAAO,eAAoD;EACvD,OAAO,qBAAqB,SAAS,SAAS,SAAS,CAAC;EACxD,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,wBACF,UACA,SACA,aACiC;AACjC,QAAO,oBAA8D;EACjE,OAAO,sBAAsB,SAAS,UAAU,SAAS,SAAS,CAAC;EACnE,YAAY;EACZ;EACH,CAAC;;AAGN,MAAM,kBAAkB,aAAkD;AAwBtE,QAvB0C,OAAO,QAEzC,SACA,YACwB;AACxB,SAAO,iBAAiB,UAAU,SAAS,QAAQ;IAEvD;EACI,KACI,SACA,UACyB;AACzB,UAAO,gBAAgB,UAAU,SAAS,SAAS;;EAEvD,UACI,SACA,UAC8B;AAC9B,UAAO,qBAAqB,UAAU,SAAS,SAAS;;EAE/D,CACJ;;AAKL,MAAa,gBAAgB,UAA+B,EAAE,KAAqB;AAC/E,QAAO,eAAeA,eAAkB,QAAQ,CAAC;;AAGrD,MAAa,SAAS,eAAeC,SAAmB"}

package/package.json

@@ -0,0 +1,55 @@
+{
+    "name": "@better-agent/ollama",
+    "version": "0.2.0-beta.1",
+    "description": "Better Agent Ollama provider",
+    "license": "MIT",
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/better-agent/better-agent.git",
+        "directory": "packages/ollama"
+    },
+    "homepage": "https://github.com/better-agent/better-agent/tree/main/packages/ollama",
+    "bugs": {
+        "url": "https://github.com/better-agent/better-agent/issues"
+    },
+    "engines": {
+        "bun": ">=1.3.3",
+        "node": ">=20"
+    },
+    "type": "module",
+    "types": "./dist/index.d.mts",
+    "sideEffects": false,
+    "exports": {
+        ".": {
+            "types": "./dist/index.d.mts",
+            "import": "./dist/index.mjs"
+        }
+    },
+    "scripts": {
+        "build": "bunx tsdown",
+        "dev": "bunx tsdown --watch --no-clean",
+        "lint:package": "bunx publint run --strict",
+        "lint:types": "npm_config_cache=../../.npm-cache bunx attw --profile esm-only --pack .",
+        "lint": "bunx biome check .",
+        "typecheck": "bunx tsc -p tsconfig.json --noEmit",
+        "format": "bunx biome format --write .",
+        "clean": "bunx rimraf dist"
+    },
+    "files": ["dist"],
+    "publishConfig": {
+        "access": "public"
+    },
+    "dependencies": {
+        "@better-agent/ai-sdk": "0.2.0-beta.1",
+        "ai": "^6.0.168",
+        "ai-sdk-ollama": "^0.5.5"
+    },
+    "devDependencies": {
+        "@better-agent/core": "0.2.0-beta.1",
+        "@types/bun": "^1.3.12",
+        "@types/node": "^22.19.17"
+    },
+    "peerDependencies": {
+        "@better-agent/core": "0.2.0-beta.1"
+    }
+}