opencode-mem 2.7.6 → 2.8.0

package/README.md

@@ -28,7 +28,7 @@ Add to your OpenCode configuration at `~/.config/opencode/opencode.json`:
 
 ```jsonc
 {
-  "plugins": ["opencode-mem"],
+  "plugin": ["opencode-mem"],
 }
 ```
 

package/dist/services/ai/ai-provider-factory.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-provider-factory.d.ts","sourceRoot":"","sources":["../../../src/services/ai/ai-provider-factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,KAAK,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAKnF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAEjE,qBAAa,iBAAiB;IAC5B,MAAM,CAAC,cAAc,CAAC,YAAY,EAAE,cAAc,EAAE,MAAM,EAAE,cAAc,GAAG,cAAc;IAgB3F,MAAM,CAAC,qBAAqB,IAAI,cAAc,EAAE;IAIhD,MAAM,CAAC,sBAAsB,IAAI,MAAM;CAGxC"}
+{"version":3,"file":"ai-provider-factory.d.ts","sourceRoot":"","sources":["../../../src/services/ai/ai-provider-factory.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,KAAK,cAAc,EAAE,MAAM,8BAA8B,CAAC;AAMnF,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,4BAA4B,CAAC;AAEjE,qBAAa,iBAAiB;IAC5B,MAAM,CAAC,cAAc,CAAC,YAAY,EAAE,cAAc,EAAE,MAAM,EAAE,cAAc,GAAG,cAAc;IAmB3F,MAAM,CAAC,qBAAqB,IAAI,cAAc,EAAE;IAIhD,MAAM,CAAC,sBAAsB,IAAI,MAAM;CAGxC"}

package/dist/services/ai/ai-provider-factory.js

@@ -2,6 +2,7 @@ import { BaseAIProvider } from "./providers/base-provider.js";
 import { OpenAIChatCompletionProvider } from "./providers/openai-chat-completion.js";
 import { OpenAIResponsesProvider } from "./providers/openai-responses.js";
 import { AnthropicMessagesProvider } from "./providers/anthropic-messages.js";
+import { GoogleGeminiProvider } from "./providers/google-gemini.js";
 import { aiSessionManager } from "./session/ai-session-manager.js";
 export class AIProviderFactory {
     static createProvider(providerType, config) {
@@ -12,12 +13,14 @@ export class AIProviderFactory {
                 return new OpenAIResponsesProvider(config, aiSessionManager);
             case "anthropic":
                 return new AnthropicMessagesProvider(config, aiSessionManager);
+            case "google-gemini":
+                return new GoogleGeminiProvider(config, aiSessionManager);
             default:
                 throw new Error(`Unknown provider type: ${providerType}`);
         }
     }
     static getSupportedProviders() {
-        return ["openai-chat", "openai-responses", "anthropic"];
+        return ["openai-chat", "openai-responses", "anthropic", "google-gemini"];
     }
     static cleanupExpiredSessions() {
         return aiSessionManager.cleanupExpiredSessions();

package/dist/services/ai/providers/google-gemini.d.ts

@@ -0,0 +1,16 @@
+import { BaseAIProvider, type ToolCallResult } from "./base-provider.js";
+import { AISessionManager } from "../session/ai-session-manager.js";
+import type { ChatCompletionTool } from "../tools/tool-schema.js";
+/**
+ * Google Gemini Provider
+ * Supports Google's Gemini models (e.g. gemini-1.5-flash) via Google AI Studio API.
+ */
+export declare class GoogleGeminiProvider extends BaseAIProvider {
+    private aiSessionManager;
+    constructor(config: any, aiSessionManager: AISessionManager);
+    getProviderName(): string;
+    supportsSession(): boolean;
+    private addToolResponse;
+    executeToolCall(systemPrompt: string, userPrompt: string, toolSchema: ChatCompletionTool, sessionId: string): Promise<ToolCallResult>;
+}
+//# sourceMappingURL=google-gemini.d.ts.map

package/dist/services/ai/providers/google-gemini.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"google-gemini.d.ts","sourceRoot":"","sources":["../../../../src/services/ai/providers/google-gemini.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,KAAK,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACzE,OAAO,EAAE,gBAAgB,EAAE,MAAM,kCAAkC,CAAC;AACpE,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAIlE;;;GAGG;AACH,qBAAa,oBAAqB,SAAQ,cAAc;IACtD,OAAO,CAAC,gBAAgB,CAAmB;gBAE/B,MAAM,EAAE,GAAG,EAAE,gBAAgB,EAAE,gBAAgB;IAK3D,eAAe,IAAI,MAAM;IAIzB,eAAe,IAAI,OAAO;IAI1B,OAAO,CAAC,eAAe;IA4BjB,eAAe,CACnB,YAAY,EAAE,MAAM,EACpB,UAAU,EAAE,MAAM,EAClB,UAAU,EAAE,kBAAkB,EAC9B,SAAS,EAAE,MAAM,GAChB,OAAO,CAAC,cAAc,CAAC;CAyN3B"}

package/dist/services/ai/providers/google-gemini.js

@@ -0,0 +1,226 @@
+import { BaseAIProvider } from "./base-provider.js";
+import { AISessionManager } from "../session/ai-session-manager.js";
+import { log } from "../../logger.js";
+import { UserProfileValidator } from "../validators/user-profile-validator.js";
+/**
+ * Google Gemini Provider
+ * Supports Google's Gemini models (e.g. gemini-1.5-flash) via Google AI Studio API.
+ */
+export class GoogleGeminiProvider extends BaseAIProvider {
+    aiSessionManager;
+    constructor(config, aiSessionManager) {
+        super(config);
+        this.aiSessionManager = aiSessionManager;
+    }
+    getProviderName() {
+        return "google-gemini";
+    }
+    supportsSession() {
+        return true;
+    }
+    addToolResponse(sessionId, messages, toolCallId, content) {
+        const sequence = this.aiSessionManager.getLastSequence(sessionId) + 1;
+        this.aiSessionManager.addMessage({
+            aiSessionId: sessionId,
+            sequence,
+            role: "tool",
+            content,
+            toolCallId,
+        });
+        // Gemini tool response format
+        messages.push({
+            role: "function",
+            parts: [
+                {
+                    functionResponse: {
+                        name: toolCallId.split(":")[0], // Gemini expects the name of the function
+                        response: JSON.parse(content),
+                    },
+                },
+            ],
+        });
+    }
+    async executeToolCall(systemPrompt, userPrompt, toolSchema, sessionId) {
+        let session = this.aiSessionManager.getSession(sessionId, "google-gemini");
+        if (!session) {
+            session = this.aiSessionManager.createSession({
+                provider: "google-gemini",
+                sessionId,
+            });
+        }
+        const existingMessages = this.aiSessionManager.getMessages(session.id);
+        const contents = [];
+        // System instruction is separate in Gemini API
+        const geminiSystemInstruction = {
+            parts: [{ text: systemPrompt }],
+        };
+        // Convert existing messages to Gemini format
+        for (const msg of existingMessages) {
+            if (msg.role === "system")
+                continue; // Skip system as it's passed separately
+            const role = msg.role === "assistant" ? "model" : "user";
+            const parts = [];
+            if (msg.content) {
+                parts.push({ text: msg.content });
+            }
+            if (msg.toolCalls) {
+                for (const tc of msg.toolCalls) {
+                    parts.push({
+                        functionCall: {
+                            name: tc.function.name,
+                            args: JSON.parse(tc.function.arguments),
+                        },
+                    });
+                }
+            }
+            if (msg.role === "tool") {
+                contents.push({
+                    role: "function",
+                    parts: [
+                        {
+                            functionResponse: {
+                                name: (msg.toolCallId || "").split(":")[0],
+                                response: JSON.parse(msg.content),
+                            },
+                        },
+                    ],
+                });
+                continue;
+            }
+            contents.push({ role, parts });
+        }
+        if (contents.length === 0 || contents[contents.length - 1].role !== "user") {
+            const userSequence = this.aiSessionManager.getLastSequence(session.id) + 1;
+            this.aiSessionManager.addMessage({
+                aiSessionId: session.id,
+                sequence: userSequence,
+                role: "user",
+                content: userPrompt,
+            });
+            contents.push({ role: "user", parts: [{ text: userPrompt }] });
+        }
+        let iterations = 0;
+        const maxIterations = this.config.maxIterations ?? 5;
+        const iterationTimeout = this.config.iterationTimeout ?? 30000;
+        // Gemini API expects the tool name as a function declaration
+        const tools = [
+            {
+                functionDeclarations: [
+                    {
+                        name: toolSchema.function.name,
+                        description: toolSchema.function.description,
+                        parameters: toolSchema.function.parameters,
+                    },
+                ],
+            },
+        ];
+        while (iterations < maxIterations) {
+            iterations++;
+            const controller = new AbortController();
+            const timeout = setTimeout(() => controller.abort(), iterationTimeout);
+            try {
+                const baseUrl = this.config.apiUrl || "https://generativelanguage.googleapis.com/v1beta";
+                const url = `${baseUrl}/models/${this.config.model}:generateContent?key=${this.config.apiKey}`;
+                const requestBody = {
+                    contents,
+                    systemInstruction: geminiSystemInstruction,
+                    tools,
+                    toolConfig: {
+                        functionCallingConfig: {
+                            mode: "ANY", // Force function calling
+                            allowedFunctionNames: [toolSchema.function.name],
+                        },
+                    },
+                    generationConfig: {
+                        temperature: this.config.memoryTemperature ?? 0.3,
+                    },
+                };
+                const response = await fetch(url, {
+                    method: "POST",
+                    headers: { "Content-Type": "application/json" },
+                    body: JSON.stringify(requestBody),
+                    signal: controller.signal,
+                });
+                clearTimeout(timeout);
+                if (!response.ok) {
+                    const errorText = await response.text().catch(() => response.statusText);
+                    log("Gemini API error", {
+                        status: response.status,
+                        error: errorText,
+                        iteration: iterations,
+                    });
+                    return {
+                        success: false,
+                        error: `Gemini API error: ${response.status} - ${errorText}`,
+                        iterations,
+                    };
+                }
+                const data = (await response.json());
+                const candidate = data.candidates?.[0];
+                if (!candidate || !candidate.content) {
+                    return { success: false, error: "Invalid Gemini API response format", iterations };
+                }
+                const modelMsg = candidate.content;
+                const assistantSequence = this.aiSessionManager.getLastSequence(session.id) + 1;
+                // Map Gemini response back to our internal message format
+                const assistantMsg = {
+                    aiSessionId: session.id,
+                    sequence: assistantSequence,
+                    role: "assistant",
+                    content: "",
+                    toolCalls: [],
+                };
+                for (const part of modelMsg.parts) {
+                    if (part.text)
+                        assistantMsg.content += part.text;
+                    if (part.functionCall) {
+                        assistantMsg.toolCalls.push({
+                            id: `${part.functionCall.name}:${Date.now()}`,
+                            type: "function",
+                            function: {
+                                name: part.functionCall.name,
+                                arguments: JSON.stringify(part.functionCall.args),
+                            },
+                        });
+                    }
+                }
+                this.aiSessionManager.addMessage(assistantMsg);
+                contents.push(modelMsg);
+                if (assistantMsg.toolCalls.length > 0) {
+                    for (const toolCall of assistantMsg.toolCalls) {
+                        if (toolCall.function.name === toolSchema.function.name) {
+                            try {
+                                const parsed = JSON.parse(toolCall.function.arguments);
+                                const result = UserProfileValidator.validate(parsed);
+                                if (!result.valid)
+                                    throw new Error(result.errors.join(", "));
+                                this.addToolResponse(session.id, contents, toolCall.id, JSON.stringify({ success: true }));
+                                return { success: true, data: result.data, iterations };
+                            }
+                            catch (validationError) {
+                                const errorMessage = `Validation failed: ${String(validationError)}`;
+                                this.addToolResponse(session.id, contents, toolCall.id, JSON.stringify({ success: false, error: errorMessage }));
+                                return { success: false, error: errorMessage, iterations };
+                            }
+                        }
+                    }
+                }
+                // Retry if no tool call was made
+                const retryPrompt = "Please use the save_memories tool as instructed.";
+                const retrySequence = this.aiSessionManager.getLastSequence(session.id) + 1;
+                this.aiSessionManager.addMessage({
+                    aiSessionId: session.id,
+                    sequence: retrySequence,
+                    role: "user",
+                    content: retryPrompt,
+                });
+                contents.push({ role: "user", parts: [{ text: retryPrompt }] });
+            }
+            catch (error) {
+                clearTimeout(timeout);
+                return { success: false, error: String(error), iterations };
+            }
+        }
+        return { success: false, error: `Max iterations (${maxIterations}) reached`, iterations };
+    }
+}

package/dist/services/ai/session/session-types.d.ts

@@ -1,4 +1,4 @@
-export type AIProviderType = "openai-chat" | "openai-responses" | "anthropic";
+export type AIProviderType = "openai-chat" | "openai-responses" | "anthropic" | "google-gemini";
 export interface AIMessage {
     id?: number;
     aiSessionId: string;

package/dist/services/ai/session/session-types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"session-types.d.ts","sourceRoot":"","sources":["../../../../src/services/ai/session/session-types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,cAAc,GAAG,aAAa,GAAG,kBAAkB,GAAG,WAAW,CAAC;AAE9E,MAAM,WAAW,SAAS;IACxB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,KAAK,CAAC;QAChB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,UAAU,CAAC;QACjB,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC;KAC/C,CAAC,CAAC;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,KAAK,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;KACpB,CAAC,CAAC;IACH,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,cAAc,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,cAAc,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,mBAAmB;IAClC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC"}
+{"version":3,"file":"session-types.d.ts","sourceRoot":"","sources":["../../../../src/services/ai/session/session-types.ts"],"names":[],"mappings":"AAAA,MAAM,MAAM,cAAc,GAAG,aAAa,GAAG,kBAAkB,GAAG,WAAW,GAAG,eAAe,CAAC;AAEhG,MAAM,WAAW,SAAS;IACxB,EAAE,CAAC,EAAE,MAAM,CAAC;IACZ,WAAW,EAAE,MAAM,CAAC;IACpB,QAAQ,EAAE,MAAM,CAAC;IACjB,IAAI,EAAE,QAAQ,GAAG,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;IAC/C,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE,KAAK,CAAC;QAChB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,EAAE,UAAU,CAAC;QACjB,QAAQ,EAAE;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAA;SAAE,CAAC;KAC/C,CAAC,CAAC;IACH,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,aAAa,CAAC,EAAE,KAAK,CAAC;QACpB,IAAI,EAAE,MAAM,CAAC;QACb,CAAC,GAAG,EAAE,MAAM,GAAG,GAAG,CAAC;KACpB,CAAC,CAAC;IACH,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,SAAS;IACxB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,cAAc,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAC/B,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,mBAAmB;IAClC,QAAQ,EAAE,cAAc,CAAC;IACzB,SAAS,EAAE,MAAM,CAAC;IAClB,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,mBAAmB;IAClC,cAAc,CAAC,EAAE,MAAM,CAAC;IACxB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CAChC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "opencode-mem",
-  "version": "2.7.6",
+  "version": "2.8.0",
   "description": "OpenCode plugin that gives coding agents persistent memory using local vector database",
   "type": "module",
   "main": "dist/plugin.js",