example.py

# example_2.py - Simple MemoryOS Basic Demo

import os
from memoryos import Memoryos

# --- 基础配置 ---
USER_ID = "demo_user"
ASSISTANT_ID = "demo_assistant"
OPENAI_API_KEY = ""
OPENAI_BASE_URL = ""
DATA_STORAGE_PATH = "./simple_demo_data"
LLM_MODEL = "gpt-4o-mini"

def simple_demo():
 print("🚀 MemoryOS Simple Demo")
 
 # 1. 初始化 MemoryOS
 print("📦 Initializing MemoryOS...")
 try:
 memo = Memoryos(
 user_id=USER_ID,
 openai_api_key=OPENAI_API_KEY,
 openai_base_url=OPENAI_BASE_URL,
 data_storage_path=DATA_STORAGE_PATH,
 llm_model=LLM_MODEL,
 assistant_id=ASSISTANT_ID,
 short_term_capacity=7, 
 mid_term_heat_threshold=5, 
 )
 print("✅ MemoryOS initialized successfully!\n")
 except Exception as e:
 print(f"❌ Error: {e}")
 return

 # 2. 添加一些基础记忆
 print("💾 Adding some memories...")
 
 memo.add_memory(
 user_input="Hi! I'm Tom, I work as a data scientist in San Francisco.",
 agent_response="Hello Tom! Nice to meet you. Data science is such an exciting field. What kind of data do you work with?"
 )
 
 memo.add_memory(
 user_input="I mainly work with e-commerce data. I also love playing guitar in my free time.",
 agent_response="That's a great combination! E-commerce analytics must provide fascinating insights into consumer behavior. How long have you been playing guitar?"
 )
 
 memo.add_memory(
 user_input="I've been playing for about 5 years. I really enjoy blues and rock music.",
 agent_response="Five years is a solid foundation! Blues and rock are fantastic genres for guitar. Do you have a favorite artist or song you like to play?"
 )
 


 
 test_query = "What do you remember about my job and hobbies?"
 print(f" User: {test_query}")
 
 response = memo.get_response(
 query=test_query,
 )
 
 print(f"Assistant: {response}")

if __name__ == "__main__":
 simple_demo()