@pikku/addon-ollama 0.1.0

package/dist/src/functions/embeddings/embed-many.function.d.ts

@@ -0,0 +1,37 @@
+import { z } from 'zod';
+export declare const OllamaEmbedManyInput: z.ZodObject<{
+    model: z.ZodString;
+    values: z.ZodArray<z.ZodString>;
+}, z.core.$strip>;
+export declare const OllamaEmbedManyOutput: z.ZodObject<{
+    embeddings: z.ZodArray<z.ZodArray<z.ZodNumber>>;
+    usage: z.ZodObject<{
+        tokens: z.ZodNumber;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export declare const ollamaEmbedMany: import("#pikku").PikkuFunctionConfig<{
+    model: string;
+    values: string[];
+}, {
+    embeddings: number[][];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").PikkuFunctionSessionless<{
+    model: string;
+    values: string[];
+}, {
+    embeddings: number[][];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services> | import("#pikku").PikkuFunction<{
+    model: string;
+    values: string[];
+}, {
+    embeddings: number[][];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services>, undefined, undefined>;
+//# sourceMappingURL=embed-many.function.d.ts.map

package/dist/src/functions/embeddings/embed-many.function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed-many.function.d.ts","sourceRoot":"","sources":["../../../../src/functions/embeddings/embed-many.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAIvB,eAAO,MAAM,oBAAoB;;;iBAG/B,CAAA;AAMF,eAAO,MAAM,qBAAqB;;;;;iBAGhC,CAAA;AAIF,eAAO,MAAM,eAAe;;;;;;;;;;;;;;;;;;;;;;;;uEAe1B,CAAA"}

package/dist/src/functions/embeddings/embed-many.function.js

@@ -0,0 +1,31 @@
+import { z } from 'zod';
+import { embedMany } from 'ai';
+import { pikkuSessionlessFunc } from '#pikku';
+export const OllamaEmbedManyInput = z.object({
+    model: z.string().describe('The Ollama embedding model to use (e.g. "nomic-embed-text", "mxbai-embed-large")'),
+    values: z.array(z.string()).describe('The texts to embed'),
+});
+const UsageSchema = z.object({
+    tokens: z.number().describe('Number of tokens used'),
+});
+export const OllamaEmbedManyOutput = z.object({
+    embeddings: z.array(z.array(z.number())).describe('The embedding vectors'),
+    usage: UsageSchema.describe('Token usage statistics'),
+});
+export const ollamaEmbedMany = pikkuSessionlessFunc({
+    description: 'Create embedding vectors from multiple texts using an Ollama model',
+    node: { displayName: 'Embed Multiple Texts', category: 'Embeddings', type: 'action' },
+    input: OllamaEmbedManyInput,
+    output: OllamaEmbedManyOutput,
+    func: async ({ ollama }, data) => {
+        const result = await embedMany({
+            model: ollama.embedding(data.model),
+            values: data.values,
+        });
+        return {
+            embeddings: result.embeddings,
+            usage: result.usage,
+        };
+    },
+});
+//# sourceMappingURL=embed-many.function.js.map

package/dist/src/functions/embeddings/embed-many.function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed-many.function.js","sourceRoot":"","sources":["../../../../src/functions/embeddings/embed-many.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,SAAS,EAAE,MAAM,IAAI,CAAA;AAC9B,OAAO,EAAE,oBAAoB,EAAE,MAAM,QAAQ,CAAA;AAE7C,MAAM,CAAC,MAAM,oBAAoB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kFAAkF,CAAC;IAC9G,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC;CAC3D,CAAC,CAAA;AAEF,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CACrD,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,qBAAqB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC5C,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;IAC1E,KAAK,EAAE,WAAW,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CACtD,CAAC,CAAA;AAIF,MAAM,CAAC,MAAM,eAAe,GAAG,oBAAoB,CAAC;IAClD,WAAW,EAAE,oEAAoE;IACjF,IAAI,EAAE,EAAE,WAAW,EAAE,sBAAsB,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE;IACrF,KAAK,EAAE,oBAAoB;IAC3B,MAAM,EAAE,qBAAqB;IAC7B,IAAI,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,MAAM,SAAS,CAAC;YAC7B,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;YACnC,MAAM,EAAE,IAAI,CAAC,MAAM;SACpB,CAAC,CAAA;QACF,OAAO;YACL,UAAU,EAAE,MAAM,CAAC,UAAU;YAC7B,KAAK,EAAE,MAAM,CAAC,KAAK;SACV,CAAA;IACb,CAAC;CACF,CAAC,CAAA"}

package/dist/src/functions/embeddings/embed.function.d.ts

@@ -0,0 +1,37 @@
+import { z } from 'zod';
+export declare const OllamaEmbedInput: z.ZodObject<{
+    model: z.ZodString;
+    value: z.ZodString;
+}, z.core.$strip>;
+export declare const OllamaEmbedOutput: z.ZodObject<{
+    embedding: z.ZodArray<z.ZodNumber>;
+    usage: z.ZodObject<{
+        tokens: z.ZodNumber;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export declare const ollamaEmbed: import("#pikku").PikkuFunctionConfig<{
+    model: string;
+    value: string;
+}, {
+    embedding: number[];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").PikkuFunctionSessionless<{
+    model: string;
+    value: string;
+}, {
+    embedding: number[];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services> | import("#pikku").PikkuFunction<{
+    model: string;
+    value: string;
+}, {
+    embedding: number[];
+    usage: {
+        tokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services>, undefined, undefined>;
+//# sourceMappingURL=embed.function.d.ts.map

package/dist/src/functions/embeddings/embed.function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.function.d.ts","sourceRoot":"","sources":["../../../../src/functions/embeddings/embed.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAIvB,eAAO,MAAM,gBAAgB;;;iBAG3B,CAAA;AAMF,eAAO,MAAM,iBAAiB;;;;;iBAG5B,CAAA;AAIF,eAAO,MAAM,WAAW;;;;;;;;;;;;;;;;;;;;;;;;uEAetB,CAAA"}

package/dist/src/functions/embeddings/embed.function.js

@@ -0,0 +1,31 @@
+import { z } from 'zod';
+import { embed } from 'ai';
+import { pikkuSessionlessFunc } from '#pikku';
+export const OllamaEmbedInput = z.object({
+    model: z.string().describe('The Ollama embedding model to use (e.g. "nomic-embed-text", "mxbai-embed-large")'),
+    value: z.string().describe('The text to embed'),
+});
+const UsageSchema = z.object({
+    tokens: z.number().describe('Number of tokens used'),
+});
+export const OllamaEmbedOutput = z.object({
+    embedding: z.array(z.number()).describe('The embedding vector'),
+    usage: UsageSchema.describe('Token usage statistics'),
+});
+export const ollamaEmbed = pikkuSessionlessFunc({
+    description: 'Create an embedding vector from text using an Ollama model',
+    node: { displayName: 'Embed Text', category: 'Embeddings', type: 'action' },
+    input: OllamaEmbedInput,
+    output: OllamaEmbedOutput,
+    func: async ({ ollama }, data) => {
+        const result = await embed({
+            model: ollama.embedding(data.model),
+            value: data.value,
+        });
+        return {
+            embedding: result.embedding,
+            usage: result.usage,
+        };
+    },
+});
+//# sourceMappingURL=embed.function.js.map

package/dist/src/functions/embeddings/embed.function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embed.function.js","sourceRoot":"","sources":["../../../../src/functions/embeddings/embed.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,KAAK,EAAE,MAAM,IAAI,CAAA;AAC1B,OAAO,EAAE,oBAAoB,EAAE,MAAM,QAAQ,CAAA;AAE7C,MAAM,CAAC,MAAM,gBAAgB,GAAG,CAAC,CAAC,MAAM,CAAC;IACvC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kFAAkF,CAAC;IAC9G,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;CAChD,CAAC,CAAA;AAEF,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uBAAuB,CAAC;CACrD,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,iBAAiB,GAAG,CAAC,CAAC,MAAM,CAAC;IACxC,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,sBAAsB,CAAC;IAC/D,KAAK,EAAE,WAAW,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CACtD,CAAC,CAAA;AAIF,MAAM,CAAC,MAAM,WAAW,GAAG,oBAAoB,CAAC;IAC9C,WAAW,EAAE,4DAA4D;IACzE,IAAI,EAAE,EAAE,WAAW,EAAE,YAAY,EAAE,QAAQ,EAAE,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE;IAC3E,KAAK,EAAE,gBAAgB;IACvB,MAAM,EAAE,iBAAiB;IACzB,IAAI,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,MAAM,KAAK,CAAC;YACzB,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,KAAK,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,KAAK;SAClB,CAAC,CAAA;QACF,OAAO;YACL,SAAS,EAAE,MAAM,CAAC,SAAS;YAC3B,KAAK,EAAE,MAAM,CAAC,KAAK;SACV,CAAA;IACb,CAAC;CACF,CAAC,CAAA"}

package/dist/src/functions/text/generate-object.function.d.ts

@@ -0,0 +1,101 @@
+import { z } from 'zod';
+export declare const OllamaGenerateObjectInput: z.ZodObject<{
+    model: z.ZodString;
+    prompt: z.ZodOptional<z.ZodString>;
+    messages: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        role: z.ZodEnum<{
+            system: "system";
+            user: "user";
+            assistant: "assistant";
+        }>;
+        content: z.ZodString;
+    }, z.core.$strip>>>;
+    system: z.ZodOptional<z.ZodString>;
+    schema: z.ZodRecord<z.ZodString, z.ZodAny>;
+    schemaName: z.ZodOptional<z.ZodString>;
+    schemaDescription: z.ZodOptional<z.ZodString>;
+    mode: z.ZodOptional<z.ZodEnum<{
+        tool: "tool";
+        auto: "auto";
+        json: "json";
+    }>>;
+    maxTokens: z.ZodOptional<z.ZodNumber>;
+    temperature: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+export declare const OllamaGenerateObjectOutput: z.ZodObject<{
+    object: z.ZodAny;
+    finishReason: z.ZodString;
+    usage: z.ZodObject<{
+        promptTokens: z.ZodNumber;
+        completionTokens: z.ZodNumber;
+        totalTokens: z.ZodNumber;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export declare const ollamaGenerateObject: import("#pikku").PikkuFunctionConfig<{
+    model: string;
+    schema: Record<string, any>;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    schemaName?: string | undefined;
+    schemaDescription?: string | undefined;
+    mode?: "tool" | "auto" | "json" | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+}, {
+    object: any;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").PikkuFunctionSessionless<{
+    model: string;
+    schema: Record<string, any>;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    schemaName?: string | undefined;
+    schemaDescription?: string | undefined;
+    mode?: "tool" | "auto" | "json" | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+}, {
+    object: any;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services> | import("#pikku").PikkuFunction<{
+    model: string;
+    schema: Record<string, any>;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    schemaName?: string | undefined;
+    schemaDescription?: string | undefined;
+    mode?: "tool" | "auto" | "json" | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+}, {
+    object: any;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services>, undefined, undefined>;
+//# sourceMappingURL=generate-object.function.d.ts.map

package/dist/src/functions/text/generate-object.function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-object.function.d.ts","sourceRoot":"","sources":["../../../../src/functions/text/generate-object.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAUvB,eAAO,MAAM,yBAAyB;;;;;;;;;;;;;;;;;;;;;;iBAWpC,CAAA;AAQF,eAAO,MAAM,0BAA0B;;;;;;;;iBAIrC,CAAA;AAIF,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uEAwB/B,CAAA"}

package/dist/src/functions/text/generate-object.function.js

@@ -0,0 +1,56 @@
+import { z } from 'zod';
+import { generateObject } from 'ai';
+import { jsonSchema } from 'ai';
+import { pikkuSessionlessFunc } from '#pikku';
+const MessageSchema = z.object({
+    role: z.enum(['system', 'user', 'assistant']).describe('The role of the message author'),
+    content: z.string().describe('The contents of the message'),
+});
+export const OllamaGenerateObjectInput = z.object({
+    model: z.string().describe('The Ollama model to use (e.g. "llama3.2", "phi3:mini")'),
+    prompt: z.string().optional().describe('The prompt to generate a structured object from'),
+    messages: z.array(MessageSchema).optional().describe('A list of messages comprising the conversation so far'),
+    system: z.string().optional().describe('System message to set the behavior of the model'),
+    schema: z.record(z.string(), z.any()).describe('JSON Schema describing the expected output object'),
+    schemaName: z.string().optional().describe('Optional name for the schema'),
+    schemaDescription: z.string().optional().describe('Optional description of what the schema represents'),
+    mode: z.enum(['auto', 'tool', 'json']).optional().describe('Mode for object generation. Use "json" for models that do not support tools.'),
+    maxTokens: z.number().optional().describe('Maximum number of tokens to generate'),
+    temperature: z.number().min(0).max(2).optional().describe('Sampling temperature between 0 and 2'),
+});
+const UsageSchema = z.object({
+    promptTokens: z.number().describe('Number of tokens in the prompt'),
+    completionTokens: z.number().describe('Number of tokens in the completion'),
+    totalTokens: z.number().describe('Total number of tokens used'),
+});
+export const OllamaGenerateObjectOutput = z.object({
+    object: z.any().describe('The generated structured object'),
+    finishReason: z.string().describe('The reason the model stopped generating'),
+    usage: UsageSchema.describe('Token usage statistics'),
+});
+export const ollamaGenerateObject = pikkuSessionlessFunc({
+    description: 'Generate a structured JSON object using an Ollama model',
+    node: { displayName: 'Generate Object', category: 'Text', type: 'action' },
+    input: OllamaGenerateObjectInput,
+    output: OllamaGenerateObjectOutput,
+    func: async ({ ollama }, data) => {
+        const result = await generateObject({
+            model: ollama(data.model),
+            prompt: data.prompt,
+            messages: data.messages,
+            system: data.system,
+            schema: jsonSchema(data.schema),
+            schemaName: data.schemaName,
+            schemaDescription: data.schemaDescription,
+            mode: data.mode,
+            maxTokens: data.maxTokens,
+            temperature: data.temperature,
+        });
+        return {
+            object: result.object,
+            finishReason: result.finishReason,
+            usage: result.usage,
+        };
+    },
+});
+//# sourceMappingURL=generate-object.function.js.map

package/dist/src/functions/text/generate-object.function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate-object.function.js","sourceRoot":"","sources":["../../../../src/functions/text/generate-object.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,cAAc,EAAE,MAAM,IAAI,CAAA;AACnC,OAAO,EAAE,UAAU,EAAE,MAAM,IAAI,CAAA;AAC/B,OAAO,EAAE,oBAAoB,EAAE,MAAM,QAAQ,CAAA;AAE7C,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7B,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACxF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;CAC5D,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC,MAAM,CAAC;IAChD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wDAAwD,CAAC;IACpF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IACzF,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uDAAuD,CAAC;IAC7G,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IACzF,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,CAAC,QAAQ,CAAC,mDAAmD,CAAC;IACnG,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;IAC1E,iBAAiB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,oDAAoD,CAAC;IACvG,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8EAA8E,CAAC;IAC1I,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;IACjF,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;CAClG,CAAC,CAAA;AAEF,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACnE,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;IAC3E,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;CAChE,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAAG,CAAC,CAAC,MAAM,CAAC;IACjD,MAAM,EAAE,CAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,CAAC,iCAAiC,CAAC;IAC3D,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IAC5E,KAAK,EAAE,WAAW,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CACtD,CAAC,CAAA;AAIF,MAAM,CAAC,MAAM,oBAAoB,GAAG,oBAAoB,CAAC;IACvD,WAAW,EAAE,yDAAyD;IACtE,IAAI,EAAE,EAAE,WAAW,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;IAC1E,KAAK,EAAE,yBAAyB;IAChC,MAAM,EAAE,0BAA0B;IAClC,IAAI,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC;YAClC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,MAAM,EAAE,UAAU,CAAC,IAAI,CAAC,MAAM,CAAC;YAC/B,UAAU,EAAE,IAAI,CAAC,UAAU;YAC3B,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;YACzC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;SAC9B,CAAC,CAAA;QACF,OAAO;YACL,MAAM,EAAE,MAAM,CAAC,MAAM;YACrB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,KAAK,EAAE,MAAM,CAAC,KAAK;SACV,CAAA;IACb,CAAC;CACF,CAAC,CAAA"}

package/dist/src/functions/text/generate.function.d.ts

@@ -0,0 +1,89 @@
+import { z } from 'zod';
+export declare const OllamaGenerateTextInput: z.ZodObject<{
+    model: z.ZodString;
+    prompt: z.ZodOptional<z.ZodString>;
+    messages: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        role: z.ZodEnum<{
+            system: "system";
+            user: "user";
+            assistant: "assistant";
+        }>;
+        content: z.ZodString;
+    }, z.core.$strip>>>;
+    system: z.ZodOptional<z.ZodString>;
+    maxTokens: z.ZodOptional<z.ZodNumber>;
+    temperature: z.ZodOptional<z.ZodNumber>;
+    topP: z.ZodOptional<z.ZodNumber>;
+    seed: z.ZodOptional<z.ZodNumber>;
+}, z.core.$strip>;
+export declare const OllamaGenerateTextOutput: z.ZodObject<{
+    text: z.ZodString;
+    finishReason: z.ZodString;
+    usage: z.ZodObject<{
+        promptTokens: z.ZodNumber;
+        completionTokens: z.ZodNumber;
+        totalTokens: z.ZodNumber;
+    }, z.core.$strip>;
+}, z.core.$strip>;
+export declare const ollamaGenerateText: import("#pikku").PikkuFunctionConfig<{
+    model: string;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    seed?: number | undefined;
+}, {
+    text: string;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").PikkuFunctionSessionless<{
+    model: string;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    seed?: number | undefined;
+}, {
+    text: string;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services> | import("#pikku").PikkuFunction<{
+    model: string;
+    prompt?: string | undefined;
+    messages?: {
+        role: "system" | "user" | "assistant";
+        content: string;
+    }[] | undefined;
+    system?: string | undefined;
+    maxTokens?: number | undefined;
+    temperature?: number | undefined;
+    topP?: number | undefined;
+    seed?: number | undefined;
+}, {
+    text: string;
+    finishReason: string;
+    usage: {
+        promptTokens: number;
+        completionTokens: number;
+        totalTokens: number;
+    };
+}, "rpc" | "session", import("#pikku").Services>, undefined, undefined>;
+//# sourceMappingURL=generate.function.d.ts.map

package/dist/src/functions/text/generate.function.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.function.d.ts","sourceRoot":"","sources":["../../../../src/functions/text/generate.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AASvB,eAAO,MAAM,uBAAuB;;;;;;;;;;;;;;;;iBASlC,CAAA;AAQF,eAAO,MAAM,wBAAwB;;;;;;;;iBAInC,CAAA;AAKF,eAAO,MAAM,kBAAkB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;uEAsB7B,CAAA"}

package/dist/src/functions/text/generate.function.js

@@ -0,0 +1,51 @@
+import { z } from 'zod';
+import { generateText } from 'ai';
+import { pikkuSessionlessFunc } from '#pikku';
+const MessageSchema = z.object({
+    role: z.enum(['system', 'user', 'assistant']).describe('The role of the message author'),
+    content: z.string().describe('The contents of the message'),
+});
+export const OllamaGenerateTextInput = z.object({
+    model: z.string().describe('The Ollama model to use (e.g. "llama3.2", "phi3:mini")'),
+    prompt: z.string().optional().describe('The prompt to generate text from'),
+    messages: z.array(MessageSchema).optional().describe('A list of messages comprising the conversation so far'),
+    system: z.string().optional().describe('System message to set the behavior of the model'),
+    maxTokens: z.number().optional().describe('Maximum number of tokens to generate'),
+    temperature: z.number().min(0).max(2).optional().describe('Sampling temperature between 0 and 2'),
+    topP: z.number().optional().describe('Nucleus sampling parameter'),
+    seed: z.number().optional().describe('Random seed for reproducible results'),
+});
+const UsageSchema = z.object({
+    promptTokens: z.number().describe('Number of tokens in the prompt'),
+    completionTokens: z.number().describe('Number of tokens in the completion'),
+    totalTokens: z.number().describe('Total number of tokens used'),
+});
+export const OllamaGenerateTextOutput = z.object({
+    text: z.string().describe('The generated text'),
+    finishReason: z.string().describe('The reason the model stopped generating'),
+    usage: UsageSchema.describe('Token usage statistics'),
+});
+export const ollamaGenerateText = pikkuSessionlessFunc({
+    description: 'Generate text using an Ollama model',
+    node: { displayName: 'Generate Text', category: 'Text', type: 'action' },
+    input: OllamaGenerateTextInput,
+    output: OllamaGenerateTextOutput,
+    func: async ({ ollama }, data) => {
+        const result = await generateText({
+            model: ollama(data.model),
+            prompt: data.prompt,
+            messages: data.messages,
+            system: data.system,
+            maxTokens: data.maxTokens,
+            temperature: data.temperature,
+            topP: data.topP,
+            seed: data.seed,
+        });
+        return {
+            text: result.text,
+            finishReason: result.finishReason,
+            usage: result.usage,
+        };
+    },
+});
+//# sourceMappingURL=generate.function.js.map

package/dist/src/functions/text/generate.function.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"generate.function.js","sourceRoot":"","sources":["../../../../src/functions/text/generate.function.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,YAAY,EAAE,MAAM,IAAI,CAAA;AACjC,OAAO,EAAE,oBAAoB,EAAE,MAAM,QAAQ,CAAA;AAE7C,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7B,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,QAAQ,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACxF,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;CAC5D,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,uBAAuB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC9C,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wDAAwD,CAAC;IACpF,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;IAC1E,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,uDAAuD,CAAC;IAC7G,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,iDAAiD,CAAC;IACzF,SAAS,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;IACjF,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;IACjG,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,4BAA4B,CAAC;IAClE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,sCAAsC,CAAC;CAC7E,CAAC,CAAA;AAEF,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC;IAC3B,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;IACnE,gBAAgB,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oCAAoC,CAAC;IAC3E,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;CAChE,CAAC,CAAA;AAEF,MAAM,CAAC,MAAM,wBAAwB,GAAG,CAAC,CAAC,MAAM,CAAC;IAC/C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;IAC/C,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yCAAyC,CAAC;IAC5E,KAAK,EAAE,WAAW,CAAC,QAAQ,CAAC,wBAAwB,CAAC;CACtD,CAAC,CAAA;AAKF,MAAM,CAAC,MAAM,kBAAkB,GAAG,oBAAoB,CAAC;IACrD,WAAW,EAAE,qCAAqC;IAClD,IAAI,EAAE,EAAE,WAAW,EAAE,eAAe,EAAE,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE;IACxE,KAAK,EAAE,uBAAuB;IAC9B,MAAM,EAAE,wBAAwB;IAChC,IAAI,EAAE,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,IAAI,EAAE,EAAE;QAC/B,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC;YAChC,KAAK,EAAE,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC;YACzB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,SAAS,EAAE,IAAI,CAAC,SAAS;YACzB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,IAAI,EAAE,IAAI,CAAC,IAAI;SAChB,CAAC,CAAA;QACF,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,YAAY,EAAE,MAAM,CAAC,YAAY;YACjC,KAAK,EAAE,MAAM,CAAC,KAAK;SACV,CAAA;IACb,CAAC;CACF,CAAC,CAAA"}

package/dist/src/index.d.ts

@@ -0,0 +1,5 @@
+export { ollamaGenerateText } from './functions/text/generate.function.js';
+export { ollamaGenerateObject } from './functions/text/generate-object.function.js';
+export { ollamaEmbed } from './functions/embeddings/embed.function.js';
+export { ollamaEmbedMany } from './functions/embeddings/embed-many.function.js';
+//# sourceMappingURL=index.d.ts.map

package/dist/src/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAA;AAC1E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAA;AAGnF,OAAO,EAAE,WAAW,EAAE,MAAM,0CAA0C,CAAA;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,+CAA+C,CAAA"}

package/dist/src/index.js

@@ -0,0 +1,7 @@
+// Ollama functions - Text
+export { ollamaGenerateText } from './functions/text/generate.function.js';
+export { ollamaGenerateObject } from './functions/text/generate-object.function.js';
+// Ollama functions - Embeddings
+export { ollamaEmbed } from './functions/embeddings/embed.function.js';
+export { ollamaEmbedMany } from './functions/embeddings/embed-many.function.js';
+//# sourceMappingURL=index.js.map

package/dist/src/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,0BAA0B;AAC1B,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAA;AAC1E,OAAO,EAAE,oBAAoB,EAAE,MAAM,8CAA8C,CAAA;AAEnF,gCAAgC;AAChC,OAAO,EAAE,WAAW,EAAE,MAAM,0CAA0C,CAAA;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,+CAA+C,CAAA"}

package/dist/src/ollama.variable.d.ts

@@ -0,0 +1,3 @@
+import { z } from 'zod';
+export declare const ollamaBaseUrlSchema: z.ZodDefault<z.ZodString>;
+//# sourceMappingURL=ollama.variable.d.ts.map

package/dist/src/ollama.variable.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.variable.d.ts","sourceRoot":"","sources":["../../src/ollama.variable.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AAGvB,eAAO,MAAM,mBAAmB,2BAA0F,CAAA"}

package/dist/src/ollama.variable.js

@@ -0,0 +1,11 @@
+import { z } from 'zod';
+import { wireVariable } from '@pikku/core/variable';
+export const ollamaBaseUrlSchema = z.string().default('http://localhost:11434/v1').describe('Base URL for the Ollama API');
+wireVariable({
+    name: 'ollama_base_url',
+    displayName: 'Ollama Base URL',
+    description: 'Base URL for the Ollama API',
+    variableId: 'OLLAMA_BASE_URL',
+    schema: ollamaBaseUrlSchema,
+});
+//# sourceMappingURL=ollama.variable.js.map

package/dist/src/ollama.variable.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.variable.js","sourceRoot":"","sources":["../../src/ollama.variable.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAA;AACvB,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAA;AAEnD,MAAM,CAAC,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,OAAO,CAAC,2BAA2B,CAAC,CAAC,QAAQ,CAAC,6BAA6B,CAAC,CAAA;AAE1H,YAAY,CAAC;IACX,IAAI,EAAE,iBAAiB;IACvB,WAAW,EAAE,iBAAiB;IAC9B,WAAW,EAAE,6BAA6B;IAC1C,UAAU,EAAE,iBAAiB;IAC7B,MAAM,EAAE,mBAAmB;CAC5B,CAAC,CAAA"}

package/dist/src/services.d.ts

@@ -0,0 +1,2 @@
+export declare const createSingletonServices: (config: import("../types/application-types.js").Config, existingServices?: Partial<import("#pikku").SingletonServices>) => Promise<import("../.pikku/pikku-services.gen.js").RequiredSingletonServices>;
+//# sourceMappingURL=services.d.ts.map

package/dist/src/services.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"services.d.ts","sourceRoot":"","sources":["../../src/services.ts"],"names":[],"mappings":"AAGA,eAAO,MAAM,uBAAuB,0MASlC,CAAA"}

package/dist/src/services.js

@@ -0,0 +1,11 @@
+import { createOpenAI } from '@ai-sdk/openai';
+import { pikkuAddonServices } from '#pikku';
+export const createSingletonServices = pikkuAddonServices(async (_config, { variables }) => {
+    const baseURL = (await variables.get('OLLAMA_BASE_URL')) ?? 'http://localhost:11434/v1';
+    const ollama = createOpenAI({
+        baseURL,
+        apiKey: 'ollama',
+    });
+    return { ollama };
+});
+//# sourceMappingURL=services.js.map

package/dist/src/services.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"services.js","sourceRoot":"","sources":["../../src/services.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAA;AAC7C,OAAO,EAAE,kBAAkB,EAAE,MAAM,QAAQ,CAAA;AAE3C,MAAM,CAAC,MAAM,uBAAuB,GAAG,kBAAkB,CAAC,KAAK,EAAE,OAAO,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE;IACzF,MAAM,OAAO,GAAG,CAAC,MAAM,SAAS,CAAC,GAAG,CAAC,iBAAiB,CAAC,CAAC,IAAI,2BAA2B,CAAA;IAEvF,MAAM,MAAM,GAAG,YAAY,CAAC;QAC1B,OAAO;QACP,MAAM,EAAE,QAAQ;KACjB,CAAC,CAAA;IAEF,OAAO,EAAE,MAAM,EAAE,CAAA;AACnB,CAAC,CAAC,CAAA"}

package/package.json

@@ -0,0 +1,44 @@
+{
+  "name": "@pikku/addon-ollama",
+  "version": "0.1.0",
+  "type": "module",
+  "imports": {
+    "#pikku": "./.pikku/pikku-types.gen.ts"
+  },
+  "exports": {
+    ".": {
+      "types": "./dist/src/index.d.ts",
+      "import": "./dist/src/index.js"
+    },
+    "./.pikku/*": "./.pikku/*",
+    "./.pikku/pikku-metadata.gen.json": "./.pikku/pikku-metadata.gen.json",
+    "./.pikku/rpc/pikku-rpc-wirings-map.internal.gen.js": {
+      "types": "./.pikku/rpc/pikku-rpc-wirings-map.internal.gen.d.ts"
+    }
+  },
+  "files": [
+    "dist",
+    ".pikku"
+  ],
+  "scripts": {
+    "prebuild": "pikku all",
+    "build": "tsc && cp -r .pikku dist/",
+    "pikku": "pikku all",
+    "ncu": "npx npm-check-updates",
+    "test": "cd test && yarn test"
+  },
+  "peerDependencies": {
+    "@pikku/core": "*",
+    "zod": "^4"
+  },
+  "devDependencies": {
+    "@pikku/cli": "*",
+    "@pikku/core": "*",
+    "typescript": "^5.7.2",
+    "zod": "^4"
+  },
+  "dependencies": {
+    "@ai-sdk/openai": "^1",
+    "ai": "^4"
+  }
+}