mem0-go

Go client SDK for mem0 - the AI memory layer for agents and apps.

Installation

go get github.com/plexusone/mem0-go

Quick Start

package main

import (
    "context"
    "fmt"
    "log"

    "github.com/plexusone/mem0-go"
)

func main() {
    // Create a client (API key from MEM0_API_KEY env var or WithAPIKey option)
    client, err := mem0.NewClient(mem0.WithAPIKey("your-api-key"))
    if err != nil {
        log.Fatal(err)
    }

    ctx := context.Background()

    // Add a memory
    messages := []mem0.Message{
        {Role: mem0.RoleUser, Content: "I prefer dark mode in all my applications"},
    }

    result, err := client.Memory().Add(ctx, messages, mem0.WithUserID("user-123"))
    if err != nil {
        log.Fatal(err)
    }
    fmt.Printf("Added memory: %v\n", result)

    // Search memories
    results, err := client.Memory().Search(ctx, "dark mode preferences",
        mem0.WithFilters(mem0.Filters{UserID: "user-123"}),
        mem0.WithTopK(5),
    )
    if err != nil {
        log.Fatal(err)
    }

    for _, r := range results {
        fmt.Printf("Memory: %s (score: %.2f)\n", r.Memory, r.Score)
    }
}

Configuration

Environment Variables

Variable	Description
MEM0_API_KEY	API key for authentication
MEM0_BASE_URL	Custom base URL (default: https://api.mem0.ai)

Client Options

client, err := mem0.NewClient(
    mem0.WithAPIKey("your-api-key"),
    mem0.WithBaseURL("https://custom.example.com"),
    mem0.WithHTTPClient(customHTTPClient),
    mem0.WithBackend(mem0.BackendHosted),
)

API Reference

Memory Interface

type Memory interface {
    Add(ctx context.Context, messages []Message, opts ...AddOption) (*AddResponse, error)
    Get(ctx context.Context, memoryID string) (*MemoryItem, error)
    GetAll(ctx context.Context, opts ...GetAllOption) (*GetAllResult, error)
    Search(ctx context.Context, query string, opts ...SearchOption) ([]SearchResult, error)
    Update(ctx context.Context, memoryID string, text string) (*MemoryItem, error)
    Delete(ctx context.Context, memoryID string) error
    DeleteAll(ctx context.Context, opts ...DeleteAllOption) (*DeleteAllResult, error)
    History(ctx context.Context, memoryID string) ([]HistoryItem, error)
    GetEventStatus(ctx context.Context, eventID string) (*EventStatus, error)
}

Add Options

mem0.WithUserID("user-123")
mem0.WithAgentID("agent-456")
mem0.WithAppID("app-789")
mem0.WithRunID("run-012")
mem0.WithMetadata(map[string]interface{}{"key": "value"})
mem0.WithInfer(true)

Search Options

mem0.WithFilters(mem0.Filters{UserID: "user-123"})
mem0.WithTopK(10)
mem0.WithRerank(true)
mem0.WithThreshold(0.5)

GetAll Options

mem0.WithGetAllFilters(mem0.Filters{UserID: "user-123"})
mem0.WithPage(1)
mem0.WithPageSize(50)

DeleteAll Options

mem0.WithDeleteUserID("user-123")
mem0.WithDeleteAgentID("agent-456")

Error Handling

_, err := client.Memory().Get(ctx, "non-existent-id")
if err != nil {
    if mem0.IsNotFoundError(err) {
        fmt.Println("Memory not found")
    } else if mem0.IsUnauthorizedError(err) {
        fmt.Println("Authentication failed")
    } else if mem0.IsBadRequestError(err) {
        fmt.Println("Invalid request")
    } else if mem0.IsRateLimitedError(err) {
        fmt.Println("Rate limit exceeded")
    }
}

Code Generation

This SDK uses ogen for OpenAPI code generation.

To regenerate the client:

./generate.sh

License

MIT