import os
from memoryos import Memory
from openai import OpenAI
BASE_PROMPT = "You are a helpful product assistant."
def answer_user(external_user_id: str, user_message: str) -> str:
memory = Memory(api_key=os.environ["MEMORYOS_API_KEY"])
openai = OpenAI(api_key=os.environ["OPENAI_API_KEY"])
memory.add(
messages=[{"role": "user", "content": user_message}],
external_user_id=external_user_id,
metadata={"source": "chat"},
)
memories = memory.get(query=user_message, external_user_id=external_user_id, limit=5, context_max_tokens=500)
system_prompt = BASE_PROMPT
if memories.has_context:
system_prompt = f"{BASE_PROMPT}\n\n{memories.system_prompt_addition}"
response = openai.responses.create(
model="gpt-4.1-mini",
input=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": user_message},
],
)
memory.close()
return response.output_text