from mem0 import MemoryClient

client = MemoryClient()

client.update_profile_settings(
    enabled=True,
    schema={
        "type": "object",
        "properties": {
            "communication_style": {
                "type": "string",
                "description": "How the user prefers to be addressed: terse, detailed, formal, casual",
            },
            "expertise_areas": {
                "type": "array",
                "items": {"type": "string"},
                "description": "Subjects the user demonstrates working knowledge of",
            },
            "current_goals": {
                "type": "array",
                "items": {"type": "string"},
                "description": "What the user is actively trying to accomplish",
            },
        },
    },
    custom_instructions="Prefer durable traits over one-off remarks.",
)

import MemoryClient from "mem0ai";

const client = new MemoryClient({ apiKey: "your-api-key" });

await client.updateProfileSettings({
  enabled: true,
  schema: {
    type: "object",
    properties: {
      communication_style: {
        type: "string",
        description:
          "How the user prefers to be addressed: terse, detailed, formal, casual",
      },
      expertise_areas: {
        type: "array",
        items: { type: "string" },
        description: "Subjects the user demonstrates working knowledge of",
      },
      current_goals: {
        type: "array",
        items: { type: "string" },
        description: "What the user is actively trying to accomplish",
      },
    },
  },
  customInstructions: "Prefer durable traits over one-off remarks.",
});

result = client.get_profile("alice")

if result["status"] == "succeeded":
    print(result["profile"])
else:
    print("not ready:", result["status"])

const result = await client.getProfile({ entityId: "alice" });

if (result.status === "succeeded") {
  console.log(result.profile);
} else {
  console.log("not ready:", result.status);
}

client.generate_profile("alice")

await client.generateProfile({ entityId: "alice" });

import time

job = client.sample_profiles(limit=5)

# Poll until the sample job reaches a terminal state (job status is UPPERCASE).
TERMINAL = {"SUCCEEDED", "PARTIALLY_SUCCEEDED", "FAILED", "CANCELLED"}
deadline = time.time() + 120
while True:
    status = client.get_profile_job(job["status_url"])["job"]
    if status["status"] in TERMINAL:
        break
    if time.time() > deadline:
        raise TimeoutError("Sample job did not finish in time")
    time.sleep(3)

print(status["status"], status["succeeded"], "of", status["total"])

# The create response lists the sampled entities; read each one's saved profile.
for entity_id in job.get("entity_ids", []):
    print(client.get_profile(entity_id))

const job = await client.sampleProfiles({ limit: 5 });

// Poll until the sample job reaches a terminal state (job status is UPPERCASE).
const TERMINAL = ["SUCCEEDED", "PARTIALLY_SUCCEEDED", "FAILED", "CANCELLED"];
const deadline = Date.now() + 120_000;
let status;
while (true) {
  status = (await client.getProfileJob(job.statusUrl)).job;
  if (TERMINAL.includes(status.status)) break;
  if (Date.now() > deadline)
    throw new Error("Sample job did not finish in time");
  await new Promise((resolve) => setTimeout(resolve, 3000));
}

console.log(status.status, status.succeeded, "of", status.total);

// The create response lists the sampled entities; read each one's saved profile.
for (const entityId of job.entityIds ?? []) {
  console.log(await client.getProfile({ entityId }));
}